We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent db40a88 commit 84553a1Copy full SHA for 84553a1
arraycontext/impl/pytato/__init__.py
@@ -954,7 +954,9 @@ def _to_frozen(
954
rec_keyed_map_array_container(_to_frozen, array),
955
actx=None)
956
957
- pt_dict_of_named_arrays = pt.make_dict_of_named_arrays(key_to_pt_arrays)
+ pt_dict_of_named_arrays = pt.transform.deduplicate(
958
+ pt.make_dict_of_named_arrays(key_to_pt_arrays))
959
+
960
transformed_dag = self.transform_dag(pt_dict_of_named_arrays)
961
pt_prg = pt.generate_jax(transformed_dag, jit=True)
962
out_dict = pt_prg()
0 commit comments