From 5aeec96d3da17364bc37516a7813107ba9087b05 Mon Sep 17 00:00:00 2001 From: Philipp Schaad Date: Mon, 29 Jan 2024 17:23:23 +0100 Subject: [PATCH] Fix blunder --- dace/transformation/transformation.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/dace/transformation/transformation.py b/dace/transformation/transformation.py index 082a9028f1..8b87939ca8 100644 --- 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: