Skip to content
Snippets Groups Projects
Commit 809dd035 authored by Marcel Koch's avatar Marcel Koch Committed by Dominic Kempf
Browse files

fix vectorization

parent 5f921003
No related branches found
No related tags found
No related merge requests found
from dune.codegen.blockstructured.tools import sub_element_inames
from dune.codegen.generation import accumulation_mixin, instruction from dune.codegen.generation import accumulation_mixin, instruction
from dune.codegen.loopy.target import dtype_floatingpoint from dune.codegen.loopy.target import dtype_floatingpoint
from dune.codegen.options import get_form_option from dune.codegen.options import get_form_option
...@@ -66,7 +67,8 @@ def generate_accumulation_instruction(expr, visitor): ...@@ -66,7 +67,8 @@ def generate_accumulation_instruction(expr, visitor):
if visitor.trial_info: if visitor.trial_info:
lfs_inames = lfs_inames + visitor.trial_info.inames lfs_inames = lfs_inames + visitor.trial_info.inames
assignee = prim.Subscript(prim.Variable(accumvar_alias), tuple(prim.Variable(i) for i in lfs_inames)) assignee = prim.Subscript(prim.Variable(accumvar_alias),
tuple(prim.Variable(i) for i in sub_element_inames() + lfs_inames))
expr_with_weight = prim.Product((expr, prim.Call(prim.Variable(accumvar + '.weight'), ()))) expr_with_weight = prim.Product((expr, prim.Call(prim.Variable(accumvar + '.weight'), ())))
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment