Skip to content
Snippets Groups Projects
Commit 4ed824c8 authored by Dominic Kempf's avatar Dominic Kempf
Browse files

Fix all DG vectorization examples

parent e417fa5d
No related branches found
No related tags found
No related merge requests found
...@@ -138,7 +138,6 @@ def collect_vector_data_rotate(knl): ...@@ -138,7 +138,6 @@ def collect_vector_data_rotate(knl):
else: else:
raise NotImplementedError raise NotImplementedError
elif quantity in knl.temporary_variables: elif quantity in knl.temporary_variables:
expr, = quantities[quantity]
if all(get_pymbolic_tag(expr) == 'vector' for expr in quantities[quantity]): if all(get_pymbolic_tag(expr) == 'vector' for expr in quantities[quantity]):
# #
# There is a vector quantity to be vectorized! That requires register rotation! # There is a vector quantity to be vectorized! That requires register rotation!
...@@ -160,7 +159,6 @@ def collect_vector_data_rotate(knl): ...@@ -160,7 +159,6 @@ def collect_vector_data_rotate(knl):
for expr in quantities[quantity]: for expr in quantities[quantity]:
assert isinstance(expr, prim.Subscript) assert isinstance(expr, prim.Subscript)
last_index = expr.index[-1] last_index = expr.index[-1]
assert last_index in tuple(range(4))
replacemap_vec[expr] = prim.Subscript(prim.Variable(get_vector_view_name(quantity)), replacemap_vec[expr] = prim.Subscript(prim.Variable(get_vector_view_name(quantity)),
(prim.Variable("vec_index") + last_index, prim.Variable(new_iname)), (prim.Variable("vec_index") + last_index, prim.Variable(new_iname)),
) )
......
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