From 736caca519b5bc657ac9b710f7cf3562add6da8a Mon Sep 17 00:00:00 2001 From: Lukas Truemper Date: Tue, 5 Sep 2023 07:27:04 +0200 Subject: [PATCH] Minor fix in WCR special case handling --- dace/transformation/dataflow/wcr_conversion.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/dace/transformation/dataflow/wcr_conversion.py b/dace/transformation/dataflow/wcr_conversion.py index 4c6164b6f1..f4151191cb 100644 --- a/dace/transformation/dataflow/wcr_conversion.py +++ b/dace/transformation/dataflow/wcr_conversion.py @@ -117,7 +117,7 @@ def can_be_applied(self, graph, expr_index, sdfg, permissive=False): func_rhs = r'^\s*%s\s*=\s*(%s)\(.*,\s*%s\s*\)\s*;$' % (re.escape(outconn), funcs, re.escape(inconn)) if re.match(lhs, cstr) is None and re.match(rhs, cstr) is None: if re.match(func_lhs, cstr) is None and re.match(func_rhs, cstr) is None: - inconns = [edge.dst_conn for edge in inedges] + inconns = list(self.tasklet.in_connectors) if len(inconns) != 2: continue @@ -236,7 +236,7 @@ def apply(self, state: SDFGState, sdfg: SDFG): re.escape(inconn)) match = re.match(func_lhs, cstr) if match is None: - inconns = [edge.dst_conn for edge in inedges] + inconns = list(self.tasklet.in_connectors) if len(inconns) != 2: continue