Skip to content

Commit db40a88

Browse files
majosminducer
authored andcommitted
deduplicate in _dag_to_compiled_func
1 parent 1242f0a commit db40a88

File tree

1 file changed

+5
-2
lines changed

1 file changed

+5
-2
lines changed

arraycontext/impl/pytato/compile.py

Lines changed: 5 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -287,8 +287,9 @@ def _dag_to_compiled_func(self, ary_or_dict_of_named_arrays,
287287
output_template):
288288
if isinstance(ary_or_dict_of_named_arrays, pt.Array):
289289
output_id = "_pt_out"
290-
dict_of_named_arrays = pt.make_dict_of_named_arrays(
291-
{output_id: ary_or_dict_of_named_arrays})
290+
dict_of_named_arrays = pt.transform.deduplicate(
291+
pt.make_dict_of_named_arrays(
292+
{output_id: ary_or_dict_of_named_arrays}))
292293
pytato_program, name_in_program_to_tags, name_in_program_to_axes = (
293294
self._dag_to_transformed_pytato_prg(dict_of_named_arrays,
294295
prg_id=self.f))
@@ -299,6 +300,8 @@ def _dag_to_compiled_func(self, ary_or_dict_of_named_arrays,
299300
output_axes=name_in_program_to_axes[output_id],
300301
output_name=output_id)
301302
elif isinstance(ary_or_dict_of_named_arrays, pt.DictOfNamedArrays):
303+
ary_or_dict_of_named_arrays = pt.transform.deduplicate(
304+
ary_or_dict_of_named_arrays)
302305
pytato_program, name_in_program_to_tags, name_in_program_to_axes = (
303306
self._dag_to_transformed_pytato_prg(ary_or_dict_of_named_arrays,
304307
prg_id=self.f))

0 commit comments

Comments
 (0)