Skip to content

Commit 581134f

Browse files
committed
feat: use pt.materialize_with_mpms directly
1 parent f326c84 commit 581134f

1 file changed

Lines changed: 1 addition & 1 deletion

File tree

‎arraycontext/impl/pytato/__init__.py‎

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -829,7 +829,7 @@ def transform_dag(self, dag: pytato.AbstractResultWithNamedArrays
829829
import pytato as pt
830830
dag = pt.tag_all_calls_to_be_inlined(dag)
831831
dag = pt.inline_calls(dag)
832-
return pt.transform.materialize_with_mpms(dag)
832+
return pt.materialize_with_mpms(dag)
833833

834834
@override
835835
def einsum(self, spec, *args, arg_names=None, tagged=()):

0 commit comments

Comments
 (0)