Minor fix in WCR special case handling

spcl · Sep 5, 2023 · 736caca · 736caca
1 parent 9c73a2f
commit 736caca
Showing 1 changed file with 2 additions and 2 deletions.
diff --git 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