Skip to content

Commit 300da3c

Browse files
majosminducer
authored andcommitted
add function inlining step to transform_dag
1 parent 0c98ff1 commit 300da3c

File tree

1 file changed

+2
-0
lines changed

1 file changed

+2
-0
lines changed

arraycontext/impl/pytato/__init__.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -789,6 +789,8 @@ def compile(self, f: Callable[..., Any]) -> Callable[..., Any]:
789789
def transform_dag(self, dag: pytato.DictOfNamedArrays
790790
) -> pytato.DictOfNamedArrays:
791791
import pytato as pt
792+
dag = pt.tag_all_calls_to_be_inlined(dag)
793+
dag = pt.inline_calls(dag)
792794
dag = pt.transform.materialize_with_mpms(dag)
793795
return dag
794796

0 commit comments

Comments
 (0)