Skip to content

Commit 957be2f

Browse files
kaushikcfdinducer
authored andcommitted
PytatoPyOpenCLArrayContext: always invoke MPMS materializer
1 parent 64d639b commit 957be2f

File tree

1 file changed

+4
-0
lines changed

1 file changed

+4
-0
lines changed

arraycontext/impl/pytato/__init__.py

Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -200,6 +200,10 @@ def transform_dag(self, dag: "pytato.DictOfNamedArrays"
200200
:arg dag: An instance of :class:`pytato.DictOfNamedArrays`
201201
:returns: A transformed version of *dag*.
202202
"""
203+
import pytato as pt
204+
205+
dag = pt.transform.materialize_with_mpms(dag)
206+
203207
return dag
204208

205209
def tag(self, tags: Union[Sequence[Tag], Tag], array):

0 commit comments

Comments
 (0)