fix porting

inducer · Oct 17, 2023 · af7a6ea · af7a6ea
1 parent 44a7a7d
commit af7a6ea
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/pytential/unregularized.py b/pytential/unregularized.py
@@ -321,7 +321,7 @@ def copy_targets_kernel(self):
         knl = lp.tag_array_axes(knl, "targets", "stride:auto, stride:1")
         knl = lp.tag_inames(knl, {"dim": "ilp"})
 
-        return knl.executor(self.cl_context)
+        return knl.executor(self.array_context.context)
 
     @property
     @memoize_method
@@ -336,6 +336,7 @@ def build_traversal(self):
         return FMMTraversalBuilder(self.array_context)
 
 
+@dataclass_array_container
 @dataclass(frozen=True)
 class _TargetInfo:
     """