fix build cinn pass check error (#58943)

phlrain · web-flow · commit 816b34e72956 · 2023-11-14T14:56:00.000+08:00
diff --git a/paddle/fluid/pir/transforms/build_cinn_pass.cc b/paddle/fluid/pir/transforms/build_cinn_pass.cc
@@ -140,7 +140,8 @@ bool IsSupportCinn(pir::Operation* op) {
   if (op->isa<paddle::dialect::FullOp>()) {
     auto out = op->result(0);
     // return IsSuportCinn( out.first_use().owern() )
-    if (out.first_use().owner()->isa<paddle::dialect::UniformOp>()) {
+    if (out.use_count() > 0 &&
+        out.first_use().owner()->isa<paddle::dialect::UniformOp>()) {
       return false;
     }
   }

Original file line number	Diff line number	Diff line change
`@@ -140,7 +140,8 @@ bool IsSupportCinn(pir::Operation* op) {`
`140`	`140`	`if (op->isa<paddle::dialect::FullOp>()) {`
`141`	`141`	`auto out = op->result(0);`
`142`	`142`	`// return IsSuportCinn( out.first_use().owern() )`
`143`		`- if (out.first_use().owner()->isa<paddle::dialect::UniformOp>()) {`
	`143`	`+ if (out.use_count() > 0 &&`
	`144`	`+ out.first_use().owner()->isa<paddle::dialect::UniformOp>()) {`
`144`	`145`	`return false;`
`145`	`146`	`}`
`146`	`147`	`}`