Skip to content
Snippets Groups Projects
Commit 1980520b authored by Andreas Klöckner's avatar Andreas Klöckner
Browse files

Prevent infinite recursion in automatic axis assignment

parent 3c391614
No related branches found
No related tags found
No related merge requests found
......@@ -773,9 +773,12 @@ def assign_automatic_axes(kernel, axis=0, local_size=None):
desired_length = kernel.get_constant_iname_length(iname)
except isl.Error:
# Likely unbounded, automatic assignment is not
# going to happen for this iname
return assign_automatic_axes(kernel,
axis=recursion_axis, local_size=local_size)
# going to happen for this iname.
new_iname_to_tag = kernel.iname_to_tag.copy()
new_iname_to_tag[iname] = None
return assign_automatic_axes(
kernel.copy(iname_to_tag=new_iname_to_tag),
axis=recursion_axis)
if axis is None:
# {{{ find a suitable axis
......
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