Skip to content

Commit 2be828e

Browse files
committed
deduplicate in freeze
1 parent c3a17d6 commit 2be828e

1 file changed

Lines changed: 3 additions & 0 deletions

File tree

arraycontext/impl/pytato/__init__.py

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -582,6 +582,9 @@ def _to_frozen(key: tuple[Any, ...], ary) -> TaggableCLArray:
582582
pt_dict_of_named_arrays = pt.make_dict_of_named_arrays(
583583
key_to_pt_arrays)
584584

585+
pt_dict_of_named_arrays = pt.transform.Deduplicator()(
586+
pt_dict_of_named_arrays)
587+
585588
# FIXME: Remove this if/when _normalize_pt_expr gets support for functions
586589
pt_dict_of_named_arrays = pt.tag_all_calls_to_be_inlined(
587590
pt_dict_of_named_arrays)

0 commit comments

Comments
 (0)