diff options
Diffstat (limited to 'mlir/lib/Bindings/Python/MainModule.cpp')
| -rw-r--r-- | mlir/lib/Bindings/Python/MainModule.cpp | 4 |
1 files changed, 4 insertions, 0 deletions
diff --git a/mlir/lib/Bindings/Python/MainModule.cpp b/mlir/lib/Bindings/Python/MainModule.cpp index 17272472ccca..8da1ab16a451 100644 --- a/mlir/lib/Bindings/Python/MainModule.cpp +++ b/mlir/lib/Bindings/Python/MainModule.cpp @@ -11,6 +11,7 @@ #include "Globals.h" #include "IRModule.h" #include "Pass.h" +#include "Rewrite.h" namespace py = pybind11; using namespace mlir; @@ -116,6 +117,9 @@ PYBIND11_MODULE(_mlir, m) { populateIRInterfaces(irModule); populateIRTypes(irModule); + auto rewriteModule = m.def_submodule("rewrite", "MLIR Rewrite Bindings"); + populateRewriteSubmodule(rewriteModule); + // Define and populate PassManager submodule. auto passModule = m.def_submodule("passmanager", "MLIR Pass Management Bindings"); |