Skip to content

Commit ade5900

Browse files
committed
[dace] Revert change to connectivity strides
1 parent 42ac16a commit ade5900

File tree

2 files changed

+3
-1
lines changed

2 files changed

+3
-1
lines changed

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

+2
Original file line numberDiff line numberDiff line change
@@ -242,13 +242,15 @@ def get_sdfg_args(sdfg: dace.SDFG, *args, check_args: bool = False, **kwargs) ->
242242
dace_shapes = get_shape_args(sdfg.arrays, dace_field_args)
243243
dace_conn_shapes = get_shape_args(sdfg.arrays, dace_conn_args)
244244
dace_strides = get_stride_args(sdfg.arrays, dace_field_args)
245+
dace_conn_strides = get_stride_args(sdfg.arrays, dace_conn_args)
245246
dace_offsets = get_offset_args(sdfg, args)
246247
all_args = {
247248
**dace_args,
248249
**dace_conn_args,
249250
**dace_shapes,
250251
**dace_conn_shapes,
251252
**dace_strides,
253+
**dace_conn_strides,
252254
**dace_offsets,
253255
}
254256

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

+1-1
Original file line numberDiff line numberDiff line change
@@ -189,7 +189,7 @@ def add_storage(
189189
name,
190190
shape=shape,
191191
strides=strides,
192-
offset=(offset if has_offset and name not in neighbor_tables else None),
192+
offset=(offset if has_offset else None),
193193
dtype=dtype,
194194
)
195195

0 commit comments

Comments
 (0)