[MLIR] [Python] Add Operation.parent

[jdd]

Summary

Attribute to get the parent operation of an operation.

[jdd]

Adding test

[stellaraccident]

Thanks!

mlir/lib/Bindings/Python/IRCore.cpp

void mlir::python::populateIRCore(py::module &m) {
   py::class_<PyOperation, PyOperationBase>(m, "Operation")
       .def_static("create", &PyOperation::create, py::arg("name"),
                   py::arg("results") = py::none(),
                   py::arg("operands") = py::none(),
                   py::arg("attributes") = py::none(),
                   py::arg("successors") = py::none(), py::arg("regions") = 0,
                   py::arg("loc") = py::none(), py::arg("ip") = py::none(),
                   kOperationCreateDocstring)
+      .def_property_readonly("parent",
+                             [](PyOperation &self) {
+                               return self.getParentOperation().getObject();
+                             })
       .def("erase", &PyOperation::erase)
       .def_property_readonly(MLIR_PYTHON_CAPI_PTR_ATTR,
                              &PyOperation::getCapsule)
       .def(MLIR_PYTHON_CAPI_FACTORY_ATTR, &PyOperation::createFromCapsule)
       .def_property_readonly("name",
                              [](PyOperation &self) {
                                self.checkValid();
                                MlirOperation operation = self.get();

mlir/test/python/ir/operation.py

for i in range(len(op.regions)):
       region = op.regions[i]
       print(f"{indent}REGION {i}:")
       for j in range(len(region.blocks)):
         block = region.blocks[j]
         print(f"{indent}  BLOCK {j}:")
         for k in range(len(block.operations)):
           child_op = block.operations[k]
           print(f"{indent}    OP {k}: {child_op}")
+          print(f"{indent}    OP {k}: parent {child_op.operation.parent.name}")
           walk_operations(indent + "      ", child_op)
 
   # CHECK: REGION 0:
   # CHECK:   BLOCK 0:
   # CHECK:     OP 0: func
+  # CHECK:     OP 0: parent module
   # CHECK:       REGION 0:
   # CHECK:         BLOCK 0:
   # CHECK:           OP 0: %0 = "custom.addi"
+  # CHECK:           OP 0: parent func
   # CHECK:           OP 1: return
+  # CHECK:           OP 1: parent func
   walk_operations("", module.operation)
 
 run(testTraverseOpRegionBlockIndices)
 
 
 # CHECK-LABEL: TEST: testBlockArgumentList
 def testBlockArgumentList():
   with Context() as ctx: