From e5ad81fc91cb580be5436ff00bd56c3b2c94d08b Mon Sep 17 00:00:00 2001 From: Andreas Kloeckner Date: Sat, 26 Jun 2021 23:59:18 -0500 Subject: [PATCH] Use ElementwiseMapKernelTag in special function kernel --- arraycontext/context.py | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/arraycontext/context.py b/arraycontext/context.py index 495dcff..ecfde93 100644 --- a/arraycontext/context.py +++ b/arraycontext/context.py @@ -218,6 +218,7 @@ class ArrayContext(ABC): import loopy as lp from .loopy import make_loopy_program + from arraycontext.transform_metadata import ElementwiseMapKernelTag return make_loopy_program( [domain_bset], [ @@ -226,7 +227,8 @@ class ArrayContext(ABC): var(c_name)(*[ var("inp%d" % i)[subscript] for i in range(nargs)])) ], - name="actx_special_%s" % c_name) + name="actx_special_%s" % c_name, + tags=(ElementwiseMapKernelTag(),)) @abstractmethod def freeze(self, array): -- GitLab