_get_scalar_func_loopy_program: Allow offsets in GlobalArgs

inducer · Jul 24, 2023 · 7ce3ea8 · 7ce3ea8
1 parent ae21a43
commit 7ce3ea8
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/arraycontext/loopy.py b/arraycontext/loopy.py
@@ -104,8 +104,10 @@ def get(c_name, nargs, naxes):
                             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)
+                    lp.GlobalArg("out",
+                                 dtype=None, shape=lp.auto, offset=lp.auto)] + [
+                        lp.GlobalArg("inp%d" % i,
+                                     dtype=None, shape=lp.auto, offset=lp.auto)
                         for i in range(nargs)] + [...],
                 name="actx_special_%s" % c_name,
                 tags=(ElementwiseMapKernelTag(),))