Skip to content

Commit 45947fe

Browse files
committed
add function inlining step to transform_dag
1 parent 4d29185 commit 45947fe

1 file changed

Lines changed: 2 additions & 0 deletions

File tree

arraycontext/impl/pytato/__init__.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -745,6 +745,8 @@ def compile(self, f: Callable[..., Any]) -> Callable[..., Any]:
745745
def transform_dag(self, dag: pytato.DictOfNamedArrays
746746
) -> pytato.DictOfNamedArrays:
747747
import pytato as pt
748+
dag = pt.tag_all_calls_to_be_inlined(dag)
749+
dag = pt.inline_calls(dag)
748750
dag = pt.transform.materialize_with_mpms(dag)
749751
return dag
750752

0 commit comments

Comments
 (0)