diff --git a/src/gt4py/next/program_processors/runners/dace_iterator/itir_to_tasklet.py b/src/gt4py/next/program_processors/runners/dace_iterator/itir_to_tasklet.py index 2f32a07ce6..0949c80318 100644 --- a/src/gt4py/next/program_processors/runners/dace_iterator/itir_to_tasklet.py +++ b/src/gt4py/next/program_processors/runners/dace_iterator/itir_to_tasklet.py @@ -698,7 +698,7 @@ def _visit_shift(self, node: itir.FunCall) -> IteratorExpr: expr = f"{internals[0]} + {internals[1]}" shifted_value = self.add_expr_tasklet( - list(zip(args, internals)), expr, dace.dtypes.int64, "ind_addr" + list(zip(args, internals)), expr, dace.dtypes.int64, "shift" )[0].value shifted_index = {dim: value for dim, value in iterator.indices.items()}