Allow literals for symbolic_domain_sizes

GridTools · Feb 14, 2025 · 329bd21 · 329bd21
1 parent c7b66df
commit 329bd21
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/src/gt4py/next/iterator/ir_utils/domain_utils.py b/src/gt4py/next/iterator/ir_utils/domain_utils.py
@@ -126,7 +126,9 @@ def translate(
                 ]
                 horizontal_sizes: dict[str, itir.Expr]
                 if symbolic_domain_sizes is not None:
-                    horizontal_sizes = {k: im.ref(v) for k, v in symbolic_domain_sizes.items()}
+                    horizontal_sizes = {
+                        k: im.ensure_expr(v) for k, v in symbolic_domain_sizes.items()
+                    }
                 else:
                     # note: ugly but cheap re-computation, but should disappear
                     assert common.is_offset_provider(offset_provider)