Skip to content

Add a pass to outline computations in a function #221

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Merged
inducer merged 18 commits into from
Jul 1, 2025
Merged

Add a pass to outline computations in a function #221

Changes from 1 commit
Commits
Show all changes
18 commits
Select commit Hold shift + click to select a range
d81f2d2
dataclass_array_container: Preserve returned type, fix tests
inducer Jun 30, 2025
c4e3bb1
move _ary_container_key_stringifier to utils.py
kaushikcfd Mar 14, 2023
666fb62
don't tag NamedArray (they inherit tags from their corresponding _con…
majosm Jun 13, 2024
f355ee3
Add actx.outline decorator
kaushikcfd Mar 14, 2023
93a2c67
add test for outline
majosm May 16, 2025
bc76a4f
adds an outlining example
kaushikcfd Mar 14, 2023
0fc5aac
forbid calling _normalize_pt_expr on a DAG with function calls for now
majosm Sep 10, 2024
0362dab
inline calls in freeze before _normalize_pt_expr
majosm Oct 24, 2024
88a6dc1
disable concatenation in outlining example for now
majosm Jan 17, 2025
7c14d10
remove duplicates in _normalize_pt_expr
majosm Jun 13, 2024
5585995
remove duplicates in _to_input_for_compiled
majosm Jun 13, 2024
af0b5bb
deduplicate in freeze
majosm Nov 14, 2024
6248a08
update _DatawrapperToBoundPlaceholderMapper with CopyMapper changes
majosm Sep 5, 2024
32210ae
add function inlining step to transform_dag
majosm May 16, 2025
1217c3e
add transform_dag implementation to pytato JAX array context in order…
majosm Jun 2, 2025
afdaed7
fix typing in make_loopy_program
majosm Jun 25, 2025
670d4be
Misc typing fixes
inducer Jul 1, 2025
cf14848
Update baseline (decrease only)
inducer Jul 1, 2025
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
Prev Previous commit
Next Next commit
inline calls in freeze before _normalize_pt_expr
  • Loading branch information
@majosm @inducer
majosm authored and inducer committed Jul 1, 2025
commit 0362dab5674cae8f68ecf9fc4b92757451dea522
6 changes: 6 additions & 0 deletions arraycontext/impl/pytato/__init__.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -633,6 +633,12 @@ def _to_frozen(

pt_dict_of_named_arrays = pt.make_dict_of_named_arrays(
key_to_pt_arrays)

# FIXME: Remove this if/when _normalize_pt_expr gets support for functions
pt_dict_of_named_arrays = pt.tag_all_calls_to_be_inlined(
pt_dict_of_named_arrays)
pt_dict_of_named_arrays = pt.inline_calls(pt_dict_of_named_arrays)

normalized_expr, bound_arguments = _normalize_pt_expr(
pt_dict_of_named_arrays)

Expand Down