[DRR] fix drr rewrite bug (#68642)

yuanlehome · web-flow · commit 092f378422ee · 2024-10-12T10:11:08.000+08:00
diff --git a/paddle/fluid/pir/drr/src/rewrite_pattern.cc b/paddle/fluid/pir/drr/src/rewrite_pattern.cc
@@ -545,7 +545,7 @@ MatchContextImpl DrrRewritePattern::CreateOperations(
     }
 
     // 3. insert new op at point max(max_res_idx+1, min_src_idx)
-    if (min_src_idx > max_res_idx) {
+    if (min_src_idx > max_res_idx || max_res_idx_op == nullptr) {
       rewriter.set_insertion_point(min_src_idx_op);
     } else {
       rewriter.SetInsertionPointAfter(max_res_idx_op);

Original file line number	Diff line number	Diff line change
`@@ -545,7 +545,7 @@ MatchContextImpl DrrRewritePattern::CreateOperations(`
`545`	`545`	`}`
`546`	`546`
`547`	`547`	`// 3. insert new op at point max(max_res_idx+1, min_src_idx)`
`548`		`- if (min_src_idx > max_res_idx) {`
	`548`	`+ if (min_src_idx > max_res_idx \|\| max_res_idx_op == nullptr) {`
`549`	`549`	`rewriter.set_insertion_point(min_src_idx_op);`
`550`	`550`	`} else {`
`551`	`551`	`rewriter.SetInsertionPointAfter(max_res_idx_op);`