diff options
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( |
