Make sure that the intermediate transient access nodes are not removed

if they are used elsewhere.
spcl · Oct 16, 2024 · 19525ef · 19525ef
1 parent 7f9333c
commit 19525ef
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/dace/transformation/dataflow/map_fusion.py b/dace/transformation/dataflow/map_fusion.py
@@ -283,12 +283,14 @@ def apply(self, graph: SDFGState, sdfg: SDFG):
             intermediate_nodes.add(dst)
             assert isinstance(dst, nodes.AccessNode)
 
-        # Check if an access node refers to non transient memory, or transient
+        # Check if an access node refers to non-transient memory, or transient
         # is used at another location (cannot erase)
         do_not_erase = set()
         for node in intermediate_nodes:
             if sdfg.arrays[node.data].transient is False:
                 do_not_erase.add(node)
+            elif len([n for n in sdfg.data_nodes() if n.data == node.data]) > 1:
+                do_not_erase.add(node)
             else:
                 for edge in graph.in_edges(node):
                     if edge.src != first_exit: