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

Make sure user-visible layout nesting levels are non-negative

parent 43e7d7a3
No related branches found
No related tags found
No related merge requests found
...@@ -342,6 +342,14 @@ def parse_array_dim_tags(dim_tags, use_increasing_target_axes=False): ...@@ -342,6 +342,14 @@ def parse_array_dim_tags(dim_tags, use_increasing_target_axes=False):
for nl in ta_nesting_levels), for nl in ta_nesting_levels),
target_axis)) target_axis))
ta_nesting_level_increment = -min(ta_nesting_levels)
for i in range(len(result)):
if (isinstance(result[i], _StrideArrayDimTagBase)
and result[i].target_axis == target_axis):
result[i] = result[i].copy(
layout_nesting_level=result[i].layout_nesting_level
+ ta_nesting_level_increment)
# }}} # }}}
return result return result
......
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