Skip to content

Commit 6798732

Browse files
committed
inline calls in freeze before _normalize_pt_expr
1 parent 0ecf082 commit 6798732

1 file changed

Lines changed: 6 additions & 0 deletions

File tree

arraycontext/impl/pytato/__init__.py

Lines changed: 6 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -522,6 +522,12 @@ def _to_frozen(key: tuple[Any, ...], ary) -> TaggableCLArray:
522522

523523
pt_dict_of_named_arrays = pt.make_dict_of_named_arrays(
524524
key_to_pt_arrays)
525+
526+
# FIXME: Remove this if/when _normalize_pt_expr gets support for functions
527+
pt_dict_of_named_arrays = pt.tag_all_calls_to_be_inlined(
528+
pt_dict_of_named_arrays)
529+
pt_dict_of_named_arrays = pt.inline_calls(pt_dict_of_named_arrays)
530+
525531
normalized_expr, bound_arguments = _normalize_pt_expr(
526532
pt_dict_of_named_arrays)
527533

0 commit comments

Comments
 (0)