diff --git a/arraycontext/impl/pytato/__init__.py b/arraycontext/impl/pytato/__init__.py index a7a0b1b2bbde4118e94d43fc07971992881d2e30..108079a975bc515a2880edf1f4ae47fb092ff5f5 100644 --- a/arraycontext/impl/pytato/__init__.py +++ b/arraycontext/impl/pytato/__init__.py @@ -200,6 +200,10 @@ class PytatoPyOpenCLArrayContext(ArrayContext): :arg dag: An instance of :class:`pytato.DictOfNamedArrays` :returns: A transformed version of *dag*. """ + import pytato as pt + + dag = pt.transform.materialize_with_mpms(dag) + return dag def tag(self, tags: Union[Sequence[Tag], Tag], array):