From 7ce3ea89db7649a77381197929c918b38f2a77c3 Mon Sep 17 00:00:00 2001 From: Andreas Kloeckner Date: Sun, 23 Jul 2023 21:15:11 -0500 Subject: [PATCH] _get_scalar_func_loopy_program: Allow offsets in GlobalArgs --- arraycontext/loopy.py | 6 ++++-- 1 file changed, 4 insertions(+), 2 deletions(-) diff --git a/arraycontext/loopy.py b/arraycontext/loopy.py index 0d304fe..af663e9 100644 --- a/arraycontext/loopy.py +++ b/arraycontext/loopy.py @@ -104,8 +104,10 @@ def _get_scalar_func_loopy_program(actx, 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(),)) -- GitLab