From 7f3b26011afb2a91945b5fc3872948196b8b5d46 Mon Sep 17 00:00:00 2001
From: hong <43953930+phlrain@users.noreply.github.com>
Date: Tue, 14 Nov 2023 14:56:00 +0800
Subject: [PATCH] fix build cinn pass check error (#58943)

---
 paddle/fluid/pir/transforms/build_cinn_pass.cc | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/paddle/fluid/pir/transforms/build_cinn_pass.cc b/paddle/fluid/pir/transforms/build_cinn_pass.cc
index 8b5f171d427a34..79a742e0849f07 100644
--- 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;
     }
   }