Skip to content
Snippets Groups Projects
Commit 2821d292 authored by Daniel Bauer's avatar Daniel Bauer :speech_balloon:
Browse files

fix inverse diagonal kernel

parent be239ad0
Branches
No related tags found
1 merge request!2Vector operators
Pipeline #66314 skipped
Hide whitespace changes
Inline Side-by-side
hog/operator_generation/kernel_types.py
+ 2
− 1
View file @ 2821d292
...@@ -396,6 +396,7 @@ class AssembleDiagonal(KernelType): ...@@ -396,6 +396,7 @@ class AssembleDiagonal(KernelType):
f' this->timingTree_->stop( "pre-communication" );\n' f' this->timingTree_->stop( "pre-communication" );\n'
f"\n" f"\n"
f"{indent(macro_loop(3), 2 * INDENT)}\n" f"{indent(macro_loop(3), 2 * INDENT)}\n"
f"{indent(self.dst.invert_elementwise(3), 2 * INDENT)}\n"
f" }}\n" f" }}\n"
f" else\n" f" else\n"
f" {{\n" f" {{\n"
...@@ -404,9 +405,9 @@ class AssembleDiagonal(KernelType): ...@@ -404,9 +405,9 @@ class AssembleDiagonal(KernelType):
f' this->timingTree_->stop( "pre-communication" );\n' f' this->timingTree_->stop( "pre-communication" );\n'
f"\n" f"\n"
f"{indent(macro_loop(2), 2 * INDENT)}\n" f"{indent(macro_loop(2), 2 * INDENT)}\n"
f"{indent(self.dst.invert_elementwise(2), 2 * INDENT)}\n"
f" }}\n" f" }}\n"
f"\n" f"\n"
f"{indent(self.dst.invert_elementwise(dim=0), INDENT)}\n"
f"}}\n" f"}}\n"
f"\n" f"\n"
f'this->stopTiming( "{self.name}" );' f'this->stopTiming( "{self.name}" );'
......
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Please register or to comment