Add a pass to outline computations in a function#221

Merged

inducer merged 18 commits into

mainfrom

Jul 1, 2025

Commits on Jul 1, 2025

dataclass_array_container: Preserve returned type, fix tests
inducer
committed
move _ary_container_key_stringifier to utils.py

kaushikcfd
authored and
inducer
committed
don't tag NamedArray (they inherit tags from their corresponding _container entry)

majosm
authored and
inducer
committed
Add actx.outline decorator

committed
add test for outline

majosm
authored and
inducer
committed
adds an outlining example

kaushikcfd
authored and
inducer
committed
forbid calling _normalize_pt_expr on a DAG with function calls for now

majosm
authored and
inducer
committed
inline calls in freeze before _normalize_pt_expr

majosm
authored and
inducer
committed
disable concatenation in outlining example for now

majosm
authored and
inducer
committed
remove duplicates in _normalize_pt_expr

majosm
authored and
inducer
committed
remove duplicates in _to_input_for_compiled

majosm
authored and
inducer
committed
deduplicate in freeze

majosm
authored and
inducer
committed
update _DatawrapperToBoundPlaceholderMapper with CopyMapper changes

majosm
authored and
inducer
committed
add function inlining step to transform_dag

majosm
authored and
inducer
committed
add transform_dag implementation to pytato JAX array context in order to inline functions

majosm
authored and
inducer
committed
fix typing in make_loopy_program

majosm
authored and
inducer
committed
Misc typing fixes
inducer
committed
Update baseline (decrease only)
inducer
committed