[pir] Add pybind property id of OpResult (#59064)

MarioLulab · web-flow · commit 483976620611 · 2023-11-20T10:27:45.000+08:00
* add OpResult  pybind id

* remove startswith 0x
diff --git a/paddle/fluid/pybind/pir.cc b/paddle/fluid/pybind/pir.cc
@@ -16,6 +16,7 @@
 #include <Python.h>
 #include <algorithm>
 #include <memory>
+#include <sstream>
 #include <string>
 #include <unordered_map>
 #include <unordered_set>
@@ -746,6 +747,19 @@ void BindOpResult(py::module *m) {
                   "persistable"));
             }
           })
+      .def_property_readonly(
+          "id",
+          [](OpResult &self) {
+            if (self.impl() == nullptr) {
+              PADDLE_THROW(phi::errors::InvalidArgument(
+                  "Currently, we can only get id of OpResult whose impl "
+                  "is not nullptr"));
+            } else {
+              std::stringstream ss;
+              ss << std::hex << self.impl();
+              return ss.str();
+            }
+          })
       .def("initialized",
            [](OpResult &self) {
              if (self.impl() == nullptr || self.type().storage() == nullptr) {
diff --git a/test/ir/pir/test_ir_pybind.py b/test/ir/pir/test_ir_pybind.py
@@ -205,6 +205,14 @@ def test_prog_seed(self):
         p.global_seed(10)
         self.assertEqual(p._seed, 10)
 
+    def test_opresult_id(self):
+        with paddle.pir_utils.IrGuard():
+            a = paddle.static.data(name='a', shape=[4, 4], dtype='float32')
+            result = paddle.tanh(a)
+
+        self.assertIsInstance(a.id, str)
+        self.assertIsInstance(result.id, str)
+
 
 if __name__ == "__main__":
     unittest.main()
