File tree Expand file tree Collapse file tree 1 file changed +2
-0
lines changed
Expand file tree Collapse file tree 1 file changed +2
-0
lines changed Original file line number Diff line number Diff line change @@ -160,6 +160,7 @@ def set_debug_enabled(flag: bool) -> None:
160160from pytato .transform .calls import inline_calls , tag_all_calls_to_be_inlined
161161from pytato .transform .dead_code_elimination import eliminate_dead_code
162162from pytato .transform .lower_to_index_lambda import to_index_lambda
163+ from pytato .transform .materialize import materialize_with_mpms
163164from pytato .transform .metadata import unify_axes_tags
164165from pytato .transform .remove_broadcasts_einsum import rewrite_einsums_with_no_broadcasts
165166from 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" ,
You can’t perform that action at this time.
0 commit comments