Finally, fix the bug in subgraph fusion that was causing index error.

spcl · Oct 16, 2024 · cab4d26 · cab4d26
1 parent 8348b53
commit cab4d26
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/dace/transformation/subgraph/subgraph_fusion.py b/dace/transformation/subgraph/subgraph_fusion.py
@@ -606,7 +606,7 @@ def compress_transient_arrays(sdfg: SDFG, graph: SDFGState, transient_allocation
 
             in_edges_iter = iter(in_edges)
             in_edge = next(in_edges_iter)
-            target_subset = dcpy(in_edge.data.subset)
+            target_subset = dcpy(in_edge.data.dst_subset)
             target_subset.pop(invariant_dimensions[data_name])
             while True:
                 try:  # executed if there are multiple in_edges
@@ -957,7 +957,7 @@ def can_be_applied(self, sdfg: SDFG, subgraph: SubgraphView) -> bool:
                                     for ie in graph.in_edges(e.src):
                                         # get corresponding inner memlet and join its subset to our access set
                                         if ie.dst_conn[2:] == e.src_conn[3:]:
-                                            current_subset = dcpy(ie.data.subset)
+                                            current_subset = dcpy(ie.data.dst_subset)
                                             current_subset.pop(invariant_dimensions[node_data])
 
                                             access_set = subsets.union(access_set, current_subset)