[MLIR] [Python] Add owner to mlir.ir.Block

Provides a way for python users to access the owning Operation from a Block.
llvm · Aug 19, 2021 · 96fbd5c · 96fbd5c
1 parent 234c4d2
commit 96fbd5c
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/mlir/lib/Bindings/Python/IRCore.cpp b/mlir/lib/Bindings/Python/IRCore.cpp
@@ -2200,6 +2200,12 @@ void mlir::python::populateIRCore(py::module &m) {
   // Mapping of PyBlock.
   //----------------------------------------------------------------------------
   py::class_<PyBlock>(m, "Block")
+      .def_property_readonly(
+          "owner",
+          [](PyBlock &self) {
+            return self.getParentOperation()->createOpView();
+          },
+          "Returns the owning operation of this block.")
       .def_property_readonly(
           "arguments",
           [](PyBlock &self) {