diff options
| author | River Riddle <riverriddle@google.com> | 2019-06-04 19:18:23 -0700 |
|---|---|---|
| committer | Mehdi Amini <joker.eph@gmail.com> | 2019-06-09 16:17:59 -0700 |
| commit | f1b848e4701a4cd3fa781c259e3728faff1c31df (patch) | |
| tree | 5587429894d04d9e78eb0e91b76aea3003c94bd1 /mlir/bindings/python/pybind.cpp | |
| parent | f59f64e838c82399f7b31949e8de75547223f42b (diff) | |
| download | bcm5719-llvm-f1b848e4701a4cd3fa781c259e3728faff1c31df.tar.gz bcm5719-llvm-f1b848e4701a4cd3fa781c259e3728faff1c31df.zip | |
NFC: Rename FuncBuilder to OpBuilder and refactor to take a top level region instead of a function.
PiperOrigin-RevId: 251563898
Diffstat (limited to 'mlir/bindings/python/pybind.cpp')
| -rw-r--r-- | mlir/bindings/python/pybind.cpp | 4 |
1 files changed, 2 insertions, 2 deletions
diff --git a/mlir/bindings/python/pybind.cpp b/mlir/bindings/python/pybind.cpp index 76cec271621..6ec0860ff7f 100644 --- a/mlir/bindings/python/pybind.cpp +++ b/mlir/bindings/python/pybind.cpp @@ -248,7 +248,7 @@ struct PythonFunctionContext { PythonFunction enter() { assert(function.function && "function is not set up"); auto *mlirFunc = static_cast<mlir::Function *>(function.function); - contextBuilder.emplace(mlirFunc); + contextBuilder.emplace(mlirFunc->getBody()); context = new mlir::edsc::ScopedContext(*contextBuilder, mlirFunc->getLoc()); return function; @@ -262,7 +262,7 @@ struct PythonFunctionContext { PythonFunction function; mlir::edsc::ScopedContext *context; - llvm::Optional<FuncBuilder> contextBuilder; + llvm::Optional<OpBuilder> contextBuilder; }; PythonFunctionContext PythonMLIRModule::makeFunctionContext( |
