use deduplicate instead of Deduplicator

Author: majosm

arraycontext/impl/pytato/__init__.py

@@ -599,8 +599,8 @@ def _to_frozen(key: tuple[Any, ...], ary) -> TaggableCLArray:
         dag = pt.make_dict_of_named_arrays(
                 key_to_pt_arrays)
 
-        from pytato.transform import Deduplicator
-        dag = Deduplicator()(dag)
+        from pytato.transform import deduplicate
+        dag = deduplicate(dag)
 
         # FIXME: Remove this if/when _normalize_pt_expr gets support for functions
         dag = pt.tag_all_calls_to_be_inlined(dag)

arraycontext/impl/pytato/compile.py

@@ -8,7 +8,6 @@
 from __future__ import annotations
 
 from pytato.array import AxesT
-from pytato.transform import Deduplicator
 
 
 __copyright__ = """
@@ -196,7 +195,8 @@ def _to_input_for_compiled(
     """
     from arraycontext.impl.pyopencl.taggable_cl_array import to_tagged_cl_array
     if isinstance(ary, pt.Array):
-        dag = Deduplicator()(pt.make_dict_of_named_arrays({"_actx_out": ary}))
+        dag = pt.transform.deduplicate(
+            pt.make_dict_of_named_arrays({"_actx_out": ary}))
         # Transform the DAG to give metadata inference a chance to do its job
         return actx.transform_dag(dag)["_actx_out"].expr
     elif isinstance(ary, TaggableCLArray):

arraycontext/impl/pytato/outline.py

@@ -1,7 +1,5 @@
 from __future__ import annotations
 
-from pytato.transform import Deduplicator
-
 
 __doc__ = """
 .. autoclass:: OutlinedCall
@@ -214,7 +212,7 @@ def __call__(self, *args: object, **kwargs: object) -> ArrayOrContainer:
 
             prefixed_output = _call_with_placeholders(
                 self.f, args, kwargs, arg_id_to_prefixed_placeholder)
-            unpacked_prefixed_output = Deduplicator()(
+            unpacked_prefixed_output = pt.transform.deduplicate(
                 pt.make_dict_of_named_arrays(_unpack_output(prefixed_output)))
 
             prefixed_placeholders = frozenset(
@@ -232,7 +230,7 @@ def __call__(self, *args: object, **kwargs: object) -> ArrayOrContainer:
         arg_id_to_placeholder = _get_arg_id_to_placeholder(arg_id_to_arg)
 
         output = _call_with_placeholders(self.f, args, kwargs, arg_id_to_placeholder)
-        unpacked_output = Deduplicator()(
+        unpacked_output = pt.transform.deduplicate(
             pt.make_dict_of_named_arrays(_unpack_output(output)))
         if len(unpacked_output) == 1 and "_" in unpacked_output:
             ret_type = pt.function.ReturnType.ARRAY

arraycontext/impl/pytato/utils.py

@@ -59,8 +59,8 @@
     ArrayOrNames,
     ArrayOrNamesTc,
     CopyMapper,
-    Deduplicator,
     TransformMapperCache,
+    deduplicate,
 )
 from pytools import UniqueNameGenerator, memoize_method
 
@@ -151,7 +151,7 @@ def _normalize_pt_expr(
     Deterministic naming of placeholders permits more effective caching of
     equivalent graphs.
     """
-    expr = Deduplicator()(expr)
+    expr = deduplicate(expr)
 
     if get_num_call_sites(expr):
         raise NotImplementedError(