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