Copy2Map: Check for happens-before memlets

spcl · Dec 2, 2023 · 5e1bdfa · 5e1bdfa
1 parent e17f0f4
commit 5e1bdfa
Showing 1 changed file with 5 additions and 0 deletions.
diff --git a/dace/transformation/dataflow/copy_to_map.py b/dace/transformation/dataflow/copy_to_map.py
@@ -35,6 +35,11 @@ def can_be_applied(self, graph: SDFGState, expr_index: int, sdfg: SDFG, permissi
         if self.a.desc(sdfg).strides == self.b.desc(sdfg).strides:
             return False
 
+        # Happens-before edges
+        for edge in graph.edges_between(self.a, self.b):
+            if edge.data.data is None:
+                return False
+
         return True
 
     def delinearize_linearize(self, desc: data.Array, copy_shape: Tuple[symbolic.SymbolicType],