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