diff --git a/python/dune/perftool/sumfact/accumulation.py b/python/dune/perftool/sumfact/accumulation.py index 798cb689efaace387ab527824613dbd3239f2875..30640c2ff4c3b9ca26726229d2594933491cd73f 100644 --- a/python/dune/perftool/sumfact/accumulation.py +++ b/python/dune/perftool/sumfact/accumulation.py @@ -57,7 +57,7 @@ def sumfact_iname(bound, _type): @kernel_cached -def accum_iname(element, restriction, bound, i): +def accum_iname(element, bound, i): if element is None: suffix = "" else: @@ -310,7 +310,7 @@ def generate_accumulation_instruction(expr, visitor): depends_on=insn_dep, within_inames=frozenset(jacobian_inames))}) - inames = tuple(accum_iname(trial_leaf_element, (test_info.restriction, trial_info.restriction), mat.rows, i) + inames = tuple(accum_iname(trial_leaf_element, mat.rows, i) for i, mat in enumerate(vsf.matrix_sequence)) # Collect the lfs and lfs indices for the accumulate call @@ -340,7 +340,7 @@ def generate_accumulation_instruction(expr, visitor): vecinames = () if vsf.vectorized: - iname = accum_iname(trial_leaf_element, (test_info.restriction, trial_info.restriction), vsf.vector_width, "vec") + iname = accum_iname(trial_leaf_element, vsf.vector_width, "vec") vecinames = (iname,) transform(lp.tag_inames, [(iname, "vec")]) from dune.perftool.tools import maybe_wrap_subscript