Skip to content

Commit 84553a1

Browse files
majosminducer
authored andcommitted
avoid duplication in PytatoJAXArrayContext.freeze
1 parent db40a88 commit 84553a1

File tree

1 file changed

+3
-1
lines changed

1 file changed

+3
-1
lines changed

arraycontext/impl/pytato/__init__.py

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -954,7 +954,9 @@ def _to_frozen(
954954
rec_keyed_map_array_container(_to_frozen, array),
955955
actx=None)
956956

957-
pt_dict_of_named_arrays = pt.make_dict_of_named_arrays(key_to_pt_arrays)
957+
pt_dict_of_named_arrays = pt.transform.deduplicate(
958+
pt.make_dict_of_named_arrays(key_to_pt_arrays))
959+
958960
transformed_dag = self.transform_dag(pt_dict_of_named_arrays)
959961
pt_prg = pt.generate_jax(transformed_dag, jit=True)
960962
out_dict = pt_prg()

0 commit comments

Comments
 (0)