From 809dd0352a0b28c89fd637b0c5378f77356b16bb Mon Sep 17 00:00:00 2001
From: Marcel Koch <marcel.koch@uni-muenster.de>
Date: Mon, 21 Jan 2019 16:26:30 +0100
Subject: [PATCH] fix vectorization
---
python/dune/codegen/blockstructured/accumulation.py | 4 +++-
1 file changed, 3 insertions(+), 1 deletion(-)
diff --git a/python/dune/codegen/blockstructured/accumulation.py b/python/dune/codegen/blockstructured/accumulation.py
index dc987cb6..69e7c901 100644
--- a/python/dune/codegen/blockstructured/accumulation.py
+++ b/python/dune/codegen/blockstructured/accumulation.py
@@ -1,3 +1,4 @@
+from dune.codegen.blockstructured.tools import sub_element_inames
from dune.codegen.generation import accumulation_mixin, instruction
from dune.codegen.loopy.target import dtype_floatingpoint
from dune.codegen.options import get_form_option
@@ -66,7 +67,8 @@ def generate_accumulation_instruction(expr, visitor):
if visitor.trial_info:
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'), ())))
--
GitLab