diff --git a/python/dune/perftool/blockstructured/vectorization.py b/python/dune/perftool/blockstructured/vectorization.py index 7cd68f05f9445e31e01f9d9a5a1ffc47893e29d8..d514198f5c29c4cea95997adcaf44f3be456494b 100644 --- a/python/dune/perftool/blockstructured/vectorization.py +++ b/python/dune/perftool/blockstructured/vectorization.py @@ -318,20 +318,21 @@ def add_iname_array(knl, vec_iname): vector_name = vec_iname + '_vec' new_temporaries = dict() - new_temporaries[array_name] = DuneTemporaryVariable(array_name, managed=True, shape=(get_form_option('number_of_blocks'),), + new_temporaries[array_name] = DuneTemporaryVariable(array_name, managed=True, + shape=(get_form_option('number_of_blocks'),), scope=lp.temp_var_scope.PRIVATE, dtype=np.float64, base_storage=vec_iname + '_buff', _base_storage_access_may_be_aliasing=True) - new_temporaries[vector_name] = DuneTemporaryVariable(vector_name, managed=True, shape=(get_form_option('number_of_blocks'),), + new_temporaries[vector_name] = DuneTemporaryVariable(vector_name, managed=True, + shape=(get_form_option('number_of_blocks'),), scope=lp.temp_var_scope.PRIVATE, dtype=np.float64, base_storage=vec_iname + '_buff', _base_storage_access_may_be_aliasing=True) - new_insns = [] - new_insns.append(lp.Assignment(assignee=prim.Subscript(prim.Variable(array_name), (prim.Variable(vec_iname),)), - expression=prim.Variable(vec_iname), - id='init_{}_buffer'.format(vec_iname), - within_inames=frozenset({vec_iname}), within_inames_is_final=True)) + new_insns = [lp.Assignment(assignee=prim.Subscript(prim.Variable(array_name), (prim.Variable(vec_iname),)), + expression=prim.Variable(vec_iname), + id='init_{}_buffer'.format(vec_iname), + within_inames=frozenset({vec_iname}), within_inames_is_final=True)] replacemap = dict() replacemap[vec_iname] = prim.Subscript(prim.Variable(vector_name), (prim.Variable(vec_iname),))