[DRR] Fix bug of matching source pattern in DRR (PaddlePaddle#58673)

* fix bug of drr * fix bug
danleifeng · Nov 6, 2023 · ee24bff · ee24bff
1 parent 4b0abdb
commit ee24bff
Showing 1 changed file with 9 additions and 1 deletion.
diff --git a/paddle/fluid/pir/drr/drr_rewrite_pattern.cc b/paddle/fluid/pir/drr/drr_rewrite_pattern.cc
@@ -271,11 +271,19 @@ bool DrrRewritePattern::MatchFromOutputToInput(
         break;
       }
       // bfs producer_op of current_op
-      if (!drr_visited.count(drr_producer_op)) {
+      if (drr_visited.count(drr_producer_op) &&
+          ir_visited.count(ir_producer_op)) {
+        continue;
+      }
+      if (!drr_visited.count(drr_producer_op) &&
+          !ir_visited.count(ir_producer_op)) {
         drr_q.push(drr_producer_op);
         ir_q.push(ir_producer_op);
         drr_visited.insert(drr_producer_op);
         ir_visited.insert(ir_producer_op);
+      } else {
+        matched = false;
+        break;
       }
     }
     // binding output tensor of current_op