Fix blunder

spcl · Jan 29, 2024 · 5aeec96 · 5aeec96
1 parent 27a350e
commit 5aeec96
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/dace/transformation/transformation.py b/dace/transformation/transformation.py
@@ -224,8 +224,8 @@ def apply_pattern(self, append: bool = True, annotate: bool = True) -> Union[Any
         tsdfg = tcfg.sdfg if not isinstance(tcfg, SDFG) else tcfg
         tgraph = tcfg.node(self.state_id) if self.state_id >= 0 else tcfg
         retval = self.apply(tgraph, tsdfg)
-        if annotate and not self.annotates_memlets(tsdfg):
-            propagation.propagate_memlets_sdfg()
+        if annotate and not self.annotates_memlets():
+            propagation.propagate_memlets_sdfg(tsdfg)
         return retval
 
     def __lt__(self, other: 'PatternTransformation') -> bool: