[dace] Remove stride and offset symbols for connectity arrays

Author: edopao

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

@@ -140,16 +140,23 @@ def get_shape_args(
     return shape_args
 
 
-def get_offset_args(
-    sdfg: dace.SDFG,
-    args: Sequence[Any],
-) -> Mapping[str, int]:
+def get_offset_args(sdfg: dace.SDFG, args: Sequence[Any]) -> Mapping[str, int]:
     sdfg_arrays: Mapping[str, dace.data.Array] = sdfg.arrays
     sdfg_params: Sequence[str] = sdfg.arg_names
+    fied_args = {param: arg for param, arg in zip(sdfg_params, args) if common.is_field(arg)}
+
+    # assume that arrays for connectivity tables do not use offset
+    assert all(
+        drange.start == 0
+        for sdfg_param, arg in fied_args.items()
+        if sdfg_param.startswith("__connectivity")
+        for drange in arg.domain.ranges
+    )
+
     return {
         str(sym): -drange.start
-        for sdfg_param, arg in zip(sdfg_params, args)
-        if common.is_field(arg)
+        for sdfg_param, arg in fied_args.items()
+        if not sdfg_param.startswith("__connectivity")
         for sym, drange in zip(sdfg_arrays[sdfg_param].offset, get_sorted_dim_ranges(arg.domain))
     }
 
@@ -235,15 +242,13 @@ def get_sdfg_args(sdfg: dace.SDFG, *args, check_args: bool = False, **kwargs) ->
     dace_shapes = get_shape_args(sdfg.arrays, dace_field_args)
     dace_conn_shapes = get_shape_args(sdfg.arrays, dace_conn_args)
     dace_strides = get_stride_args(sdfg.arrays, dace_field_args)
-    dace_conn_strides = get_stride_args(sdfg.arrays, dace_conn_args)
     dace_offsets = get_offset_args(sdfg, args)
     all_args = {
         **dace_args,
         **dace_conn_args,
         **dace_shapes,
         **dace_conn_shapes,
         **dace_strides,
-        **dace_conn_strides,
         **dace_offsets,
     }
 

src/gt4py/next/program_processors/runners/dace_iterator/itir_to_sdfg.py

@@ -189,7 +189,7 @@ def add_storage(
                 name,
                 shape=shape,
                 strides=strides,
-                offset=(offset if has_offset else None),
+                offset=(offset if has_offset and name not in neighbor_tables else None),
                 dtype=dtype,
             )