diff --git a/grudge/symbolic/compiler.py b/grudge/symbolic/compiler.py index fd6b9c78979f6f0b0d8e7fb55c77a6dcecd1b8d8..207ae1897b200931945c11107c8c3e83c3009a42 100644 --- a/grudge/symbolic/compiler.py +++ b/grudge/symbolic/compiler.py @@ -1024,6 +1024,11 @@ class ToLoopyInstructionMapper: % {"iel": iel, "idof": idof}, insns, + [ + lp.GlobalArg(name, shape=lp.auto, is_input=False) + for name, dnr in zip(insn.names, insn.do_not_return) + if not dnr + ] + [...], name="grudge_assign_%d" % self.insn_count, # Single-insn kernels may have their no_sync_with resolve to an