_get_scalar_func_loopy_program: specify that all arguments are Global…

…Args Fixes inducer/meshmode#237
inducer · Jul 21, 2023 · ae21a43 · ae21a43
1 parent dbc087c
commit ae21a43
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/arraycontext/loopy.py b/arraycontext/loopy.py
@@ -103,6 +103,10 @@ def get(c_name, nargs, naxes):
                         var(c_name)(*[
                             var("inp%d" % i)[subscript] for i in range(nargs)]))
                     ],
+                [
+                    lp.GlobalArg("out", dtype=None, shape=lp.auto)] + [
+                        lp.GlobalArg("inp%d" % i, dtype=None, shape=lp.auto)
+                        for i in range(nargs)] + [...],
                 name="actx_special_%s" % c_name,
                 tags=(ElementwiseMapKernelTag(),))