diff --git a/arraycontext/context.py b/arraycontext/context.py index 495dcff5b9907eda5216f67e00509a01483c0f3c..ecfde931bcc54646c30a6513cc7b5f56edb28c7c 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):