Skip to content

Commit 72d3ff0

Browse files
wence-sv2518
authored andcommitted
loopy: Respect instruction label inames
1 parent bfaf487 commit 72d3ff0

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

pyop2/codegen/rep2loopy.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -231,7 +231,7 @@ def loop_nesting(instructions, deps, outer_inames, kernel_name):
231231
if isinstance(insn.children[1], (Zero, Literal)):
232232
nesting[insn] = outer_inames
233233
else:
234-
nesting[insn] = runtime_indices([insn])
234+
nesting[insn] = runtime_indices([insn]) | runtime_indices(insn.label.within_inames)
235235
else:
236236
assert isinstance(insn, FunctionCall)
237237
if insn.name in (petsc_functions | {kernel_name}):

0 commit comments

Comments
 (0)