diff options
| author | Mehdi Amini <aminim@google.com> | 2019-12-05 04:32:13 -0800 |
|---|---|---|
| committer | A. Unique TensorFlower <gardener@tensorflow.org> | 2019-12-05 04:32:46 -0800 |
| commit | b14ee5a9a149aaebb6574c0f29e1ce1d8d4ca305 (patch) | |
| tree | bfc132fc3a5977a27302b1d196e76d4b231ed1d6 /mlir/lib/ExecutionEngine | |
| parent | b8cd0c148644957ac48bc7a82b7c69e83a11a721 (diff) | |
| download | bcm5719-llvm-b14ee5a9a149aaebb6574c0f29e1ce1d8d4ca305.tar.gz bcm5719-llvm-b14ee5a9a149aaebb6574c0f29e1ce1d8d4ca305.zip | |
Fix MLIR Build after LLVM upstream JIT changes (getMainJITDylib removed)
The getMainJITDylib() method was removed in 4fc68b9b7f, replace it by creating a JITDylib on the fly.
PiperOrigin-RevId: 283948595
Diffstat (limited to 'mlir/lib/ExecutionEngine')
| -rw-r--r-- | mlir/lib/ExecutionEngine/ExecutionEngine.cpp | 5 |
1 files changed, 4 insertions, 1 deletions
diff --git a/mlir/lib/ExecutionEngine/ExecutionEngine.cpp b/mlir/lib/ExecutionEngine/ExecutionEngine.cpp index 58f783824bc..2913c436ad5 100644 --- a/mlir/lib/ExecutionEngine/ExecutionEngine.cpp +++ b/mlir/lib/ExecutionEngine/ExecutionEngine.cpp @@ -234,9 +234,12 @@ Expected<std::unique_ptr<ExecutionEngine>> ExecutionEngine::create( auto objectLayer = std::make_unique<RTDyldObjectLinkingLayer>( session, []() { return std::make_unique<SectionMemoryManager>(); }); auto dataLayout = deserModule->getDataLayout(); + llvm::orc::JITDylib *mainJD = session.getJITDylibByName("<main>"); + if (!mainJD) + mainJD = &session.createJITDylib("<main>"); // Resolve symbols that are statically linked in the current process. - session.getMainJITDylib().addGenerator( + mainJD->addGenerator( cantFail(DynamicLibrarySearchGenerator::GetForCurrentProcess( dataLayout.getGlobalPrefix()))); |
