Skip to content

Commit b490968

Browse files
majosminducer
authored andcommitted
add missing materialize_with_mpms import in pytato/__init__.py
1 parent cbdb9be commit b490968

File tree

1 file changed

+2
-0
lines changed

1 file changed

+2
-0
lines changed

pytato/__init__.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -160,6 +160,7 @@ def set_debug_enabled(flag: bool) -> None:
160160
from pytato.transform.calls import inline_calls, tag_all_calls_to_be_inlined
161161
from pytato.transform.dead_code_elimination import eliminate_dead_code
162162
from pytato.transform.lower_to_index_lambda import to_index_lambda
163+
from pytato.transform.materialize import materialize_with_mpms
163164
from pytato.transform.metadata import unify_axes_tags
164165
from pytato.transform.remove_broadcasts_einsum import rewrite_einsums_with_no_broadcasts
165166
from pytato.visualization import (
@@ -253,6 +254,7 @@ def set_debug_enabled(flag: bool) -> None:
253254
"make_distributed_send_ref_holder",
254255
"make_placeholder",
255256
"make_size_param",
257+
"materialize_with_mpms",
256258
"matmul",
257259
"maximum",
258260
"minimum",

0 commit comments

Comments
 (0)