feat[next-dace]: Lowering mixed-GTIR to SDFG

src/gt4py/next/iterator/transforms/pass_manager.py

@@ -5,6 +5,7 @@
 #
 # Please, refer to the LICENSE file in the root directory.
 # SPDX-License-Identifier: BSD-3-Clause
+import warnings
 from typing import Optional, Protocol
 
 from gt4py.next import common, utils
@@ -113,15 +114,15 @@ def _process_symbolic_domains_option(
 
     if use_max_domain_range_on_unstructured_shift is None:
         use_max_domain_range_on_unstructured_shift = _has_dynamic_domains(ir)
+    elif use_max_domain_range_on_unstructured_shift:
+        if not _has_dynamic_domains(ir):
+            warnings.warn(
+                "You are using static domains together with "
+                "'use_max_domain_range_on_unstructured_shift'. This is "
+                "likely not what you wanted.",
+                stacklevel=2,
+            )
     if use_max_domain_range_on_unstructured_shift:
-        # TODO(havogt): ICON4Py uses this codepath as default for now. Once we use the minimal domain range, we should re-enable this warning.
-        # if not _has_dynamic_domains(ir):
-        #     warnings.warn(
-        #         "You are using static domains together with "
-        #         "'use_max_domain_range_on_unstructured_shift'. This is "
-        #         "likely not what you wanted.",
-        #         stacklevel=2,  # noqa: ERA001
-        #     )  # noqa: ERA001, RUF100
         assert not symbolic_domain_sizes, "Options are mutually exclusive."
         symbolic_domain_sizes = _max_domain_range_sizes(offset_provider)  # type: ignore[assignment]
     return symbolic_domain_sizes
@@ -137,6 +138,7 @@ def apply_common_transforms(
     unroll_reduce=False,
     common_subexpression_elimination=True,
     force_inline_lambda_args=False,
+    transform_concat_where_to_as_fieldop=True,
     #: A dictionary mapping axes names to their length. See :func:`infer_domain.infer_expr` for
     #: more details.
     symbolic_domain_sizes: Optional[dict[str, itir.Expr]] = None,
@@ -185,7 +187,8 @@ def apply_common_transforms(
     ir = prune_empty_concat_where.prune_empty_concat_where(ir)
     ir = remove_broadcast.RemoveBroadcast.apply(ir)
 
-    ir = concat_where.transform_to_as_fieldop(ir)
+    if transform_concat_where_to_as_fieldop:
+        ir = concat_where.transform_to_as_fieldop(ir)
 
     for _ in range(10):
         inlined = ir
@@ -257,6 +260,12 @@ def apply_common_transforms(
         ir, opcount_preserving=True, force_inline_lambda_args=force_inline_lambda_args
     )
 
+    ir = infer_domain.infer_program(
+        ir,
+        offset_provider=offset_provider,
+        symbolic_domain_sizes=symbolic_domain_sizes,
+    )
+
     assert isinstance(ir, itir.Program)
     return ir
 

src/gt4py/next/program_processors/runners/dace/__init__.py

@@ -12,6 +12,8 @@
     make_dace_backend,
     run_dace_cpu,
     run_dace_cpu_cached,
+    run_dace_cpu_gt,
+    run_dace_cpu_gt_noopt,
     run_dace_cpu_noopt,
     run_dace_gpu,
     run_dace_gpu_cached,
@@ -24,6 +26,8 @@
     "make_dace_backend",
     "run_dace_cpu",
     "run_dace_cpu_cached",
+    "run_dace_cpu_gt",
+    "run_dace_cpu_gt_noopt",
     "run_dace_cpu_noopt",
     "run_dace_gpu",
     "run_dace_gpu_cached",