From ae21a43f0cd6a80d83f36b00651f69a9b93e4170 Mon Sep 17 00:00:00 2001 From: Andreas Kloeckner Date: Thu, 20 Jul 2023 21:18:12 -0500 Subject: [PATCH] _get_scalar_func_loopy_program: specify that all arguments are GlobalArgs Fixes https://github.com/inducer/meshmode/issues/237 --- arraycontext/loopy.py | 4 ++++ 1 file changed, 4 insertions(+) diff --git a/arraycontext/loopy.py b/arraycontext/loopy.py index 2b89d76a..0d304fe7 100644 --- 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(),))