From 2821d2928a7377dd805efce3876db61b99ff7f01 Mon Sep 17 00:00:00 2001
From: Daniel Bauer <daniel.j.bauer@fau.de>
Date: Wed, 29 May 2024 11:14:23 +0200
Subject: [PATCH] fix inverse diagonal kernel
---
hog/operator_generation/kernel_types.py | 3 ++-
1 file changed, 2 insertions(+), 1 deletion(-)
diff --git a/hog/operator_generation/kernel_types.py b/hog/operator_generation/kernel_types.py
index c2c781a..778077c 100644
--- a/hog/operator_generation/kernel_types.py
+++ b/hog/operator_generation/kernel_types.py
@@ -396,6 +396,7 @@ class AssembleDiagonal(KernelType):
f' this->timingTree_->stop( "pre-communication" );\n'
f"\n"
f"{indent(macro_loop(3), 2 * INDENT)}\n"
+ f"{indent(self.dst.invert_elementwise(3), 2 * INDENT)}\n"
f" }}\n"
f" else\n"
f" {{\n"
@@ -404,9 +405,9 @@ class AssembleDiagonal(KernelType):
f' this->timingTree_->stop( "pre-communication" );\n'
f"\n"
f"{indent(macro_loop(2), 2 * INDENT)}\n"
+ f"{indent(self.dst.invert_elementwise(2), 2 * INDENT)}\n"
f" }}\n"
f"\n"
- f"{indent(self.dst.invert_elementwise(dim=0), INDENT)}\n"
f"}}\n"
f"\n"
f'this->stopTiming( "{self.name}" );'
--
GitLab