[NOMERGE] Production concatenate calls squashed

README.rst

@@ -33,7 +33,3 @@ Pytato is written to pose no particular restrictions on the version of numpy
 used for execution. To use mypy-based type checking on Pytato itself or
 packages using Pytato, numpy 1.20 or newer is required, due to the
 typing-based changes to numpy in that release.
-
-Furthermore, pytato now uses type promotion rules based on those in
-`numpy <https://numpy.org/devdocs/numpy_2_0_migration_guide.html#changes-to-numpy-data-type-promotion>`__ that should result in the same
-data types as the currently installed version of numpy.

examples/advection.py

@@ -156,6 +156,7 @@ def test_advection_convergence(order, flux_type):
         op = AdvectionOperator(discr, c=1, flux_type=flux_type,
                                dg_ops=dg_ops)
         result = op.apply(u)
+        result = pt.transform.Deduplicator()(result)
 
         prog = pt.generate_loopy(result, cl_device=queue.device)
 

pytato/__init__.py

@@ -158,7 +158,13 @@ def set_debug_enabled(flag: bool) -> None:
 from pytato.target.loopy import LoopyPyOpenCLTarget
 from pytato.target.loopy.codegen import generate_loopy
 from pytato.target.python.jax import generate_jax
-from pytato.transform.calls import inline_calls, tag_all_calls_to_be_inlined
+from pytato.transform import precompute_subexpressions
+from pytato.transform.calls import (
+    concatenate_calls,
+    inline_calls,
+    tag_all_calls_to_be_inlined,
+    zero_unused_call_bindings,
+)
 from pytato.transform.lower_to_index_lambda import to_index_lambda
 from pytato.transform.metadata import unify_axes_tags
 from pytato.transform.remove_broadcasts_einsum import rewrite_einsums_with_no_broadcasts
@@ -215,6 +221,7 @@ def set_debug_enabled(flag: bool) -> None:
     "arctan2",
     "broadcast_to",
     "concatenate",
+    "concatenate_calls",
     "conj",
     "cos",
     "cosh",
@@ -260,6 +267,7 @@ def set_debug_enabled(flag: bool) -> None:
     "number_distributed_tags",
     "ones",
     "pad",
+    "precompute_subexpressions",
     "prod",
     "real",
     "reshape",
@@ -287,5 +295,6 @@ def set_debug_enabled(flag: bool) -> None:
     "vdot",
     "verify_distributed_partition",
     "where",
+    "zero_unused_call_bindings",
     "zeros",
 )