Skip to content
Snippets Groups Projects
Commit 957be2f5 authored by Kaushik Kulkarni's avatar Kaushik Kulkarni Committed by Andreas Klöckner
Browse files

PytatoPyOpenCLArrayContext: always invoke MPMS materializer

parent 64d639b5
No related branches found
No related tags found
No related merge requests found
Pipeline #236415 passed
......@@ -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):
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment