Merge remote-tracking branch 'philip/dace-gtir-better-strides' into d…

…ace-gtir-scan
GridTools · Dec 20, 2024 · f2396c4 · f2396c4
2 parents 462f3c5 + 360baae
commit f2396c4
Show file tree

Hide file tree

Showing 4 changed files with 1,099 additions and 113 deletions.
diff --git a/src/gt4py/next/program_processors/runners/dace_fieldview/transformations/__init__.py b/src/gt4py/next/program_processors/runners/dace_fieldview/transformations/__init__.py
@@ -39,6 +39,8 @@
     gt_change_transient_strides,
     gt_map_strides_to_dst_nested_sdfg,
     gt_map_strides_to_src_nested_sdfg,
+    gt_propagate_strides_from_access_node,
+    gt_propagate_strides_of,
 )
 from .util import gt_find_constant_arguments, gt_make_transients_persistent
 
@@ -65,6 +67,8 @@
     "gt_make_transients_persistent",
     "gt_map_strides_to_dst_nested_sdfg",
     "gt_map_strides_to_src_nested_sdfg",
+    "gt_propagate_strides_from_access_node",
+    "gt_propagate_strides_of",
     "gt_reduce_distributed_buffering",
     "gt_set_gpu_blocksize",
     "gt_set_iteration_order",

diff --git a/src/gt4py/next/program_processors/runners/dace_fieldview/transformations/simplify.py b/src/gt4py/next/program_processors/runners/dace_fieldview/transformations/simplify.py
@@ -971,7 +971,7 @@ def apply(
             tmp_out_subset = dace_subsets.Range.from_array(tmp_desc)
         assert glob_in_subset is not None
 
-        # Recursively visit the nested SDFGs for mapping from inner to outer strides on the vertical dimension
+        # Recursively visit the nested SDFGs for mapping of strides from inner to outer array
         gtx_transformations.gt_map_strides_to_src_nested_sdfg(sdfg, graph, map_to_tmp_edge, glob_ac)
 
         # We now remove the `tmp` node, and create a new connection between