[IR Operation] ⚔Elden chapter 1.3⚔ (#55657)

paddle/fluid/pybind/ir.cc

@@ -94,7 +94,12 @@ void BindBlock(py::module *m) {
 void BindOperation(py::module *m) {
   py::class_<Operation> op(*m, "Operation");
   op.def("name", &Operation::name)
-      .def("get_parent", &Operation::GetParent, return_value_policy::reference)
+      .def("get_parent",
+           py::overload_cast<>(&Operation::GetParent),
+           return_value_policy::reference)
+      .def("get_parent",
+           py::overload_cast<>(&Operation::GetParent, py::const_),
+           return_value_policy::reference)
       .def("num_results", &Operation::num_results)
       .def("result", &Operation::result)
       .def("operands",

paddle/ir/core/operation.cc

@@ -210,7 +210,7 @@ Attribute Operation::attribute(const std::string &key) const {
   return attributes_.at(key);
 }
 
-Region *Operation::GetParentRegion() const {
+Region *Operation::GetParentRegion() {
   return parent_ ? parent_->GetParent() : nullptr;
 }
 
@@ -218,8 +218,8 @@ Operation *Operation::GetParentOp() const {
   return parent_ ? parent_->GetParentOp() : nullptr;
 }
 
-Program *Operation::GetParentProgram() {
-  Operation *op = this;
+const Program *Operation::GetParentProgram() const {
+  Operation *op = const_cast<Operation *>(this);
   while (Operation *parent_op = op->GetParentOp()) {
     op = parent_op;
   }

paddle/ir/core/operation.h

@@ -109,13 +109,23 @@ class IR_API alignas(8) Operation final {
     return info_.HasInterface<Interface>();
   }
 
-  Block *GetParent() const { return parent_; }
+  const Block *GetParent() const { return parent_; }
 
-  Region *GetParentRegion() const;
+  Block *GetParent() {
+    return const_cast<Block *>(
+        const_cast<const Operation *>(this)->GetParent());
+  }
+
+  Region *GetParentRegion();
 
   Operation *GetParentOp() const;
 
-  Program *GetParentProgram();
+  const Program *GetParentProgram() const;
+
+  Program *GetParentProgram() {
+    return const_cast<Program *>(
+        const_cast<const Operation *>(this)->GetParentProgram());
+  }
 
   operator Block::iterator() { return position_; }