[PIR] fix ci error. (#57513)

PaddlePaddle · Sep 20, 2023 · 4fd9767 · 4fd9767
1 parent e30b274
commit 4fd9767
Showing 1 changed file with 6 additions and 1 deletion.
diff --git a/paddle/fluid/pybind/ir.cc b/paddle/fluid/pybind/ir.cc
@@ -450,7 +450,12 @@ void BindOpResult(py::module *m) {
            })
       .def("__hash__",
            [](OpResult &self) { return std::hash<pir::Value>{}(self); })
-      .def("get_defining_op", &OpResult::owner, return_value_policy::reference)
+      .def(
+          "get_defining_op",
+          [](const OpResult &self) -> pir::Operation * {
+            return self ? self.owner() : nullptr;
+          },
+          return_value_policy::reference)
       .def_property_readonly(
           "block",
           [](OpResult &self) { return self.owner()->GetParent(); },