fix subset num_elements

GridTools · Jan 13, 2025 · 81b5fd3 · 81b5fd3
1 parent 4a82810
commit 81b5fd3
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/gt4py/next/program_processors/runners/dace_fieldview/gtir_dataflow.py b/src/gt4py/next/program_processors/runners/dace_fieldview/gtir_dataflow.py
@@ -605,7 +605,7 @@ def visit_arg(arg: IteratorExpr | DataExpr) -> IteratorExpr | ValueExpr:
                 arg_node = arg.dc_node
                 arg_desc = arg_node.desc(self.sdfg)
                 if isinstance(arg, MemletExpr):
-                    assert arg.subset.sum_elements() == 1
+                    assert arg.subset.num_elements() == 1
                     arg_desc = dace.data.Scalar(arg_desc.dtype)
             elif isinstance(arg, IteratorExpr):
                 arg_node = arg.field