From ae21a43f0cd6a80d83f36b00651f69a9b93e4170 Mon Sep 17 00:00:00 2001
From: Andreas Kloeckner <inform@tiker.net>
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 2b89d76..0d304fe 100644
--- a/arraycontext/loopy.py
+++ b/arraycontext/loopy.py
@@ -103,6 +103,10 @@ def _get_scalar_func_loopy_program(actx, 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(),))
--
GitLab