Fix a multithreaded bug in llvm LazyInitJIT (#3158)

4ac93a53 · Yinghai Lu · Tianqi Chen · 9089e196 · 4ac93a53
Commit 4ac93a53 authored May 09, 2019 by Yinghai Lu Committed by Tianqi Chen May 09, 2019
Show whitespace changes
Inline Side-by-side

Showing with 3 additions and 1 deletions

src/codegen/llvm/llvm_module.cc
+3 -1

No files found.
--- a/src/codegen/llvm/llvm_module.cc
+++ b/src/codegen/llvm/llvm_module.cc
@@ -226,8 +226,10 @@ class LLVMModuleNode final : public runtime::ModuleNode {

 private:
  void LazyInitJIT() {
-    CHECK(ee_ == nullptr);
    std::lock_guard<std::mutex> lock(mutex_);
+    if (ee_) {
+      return;
+    }
    llvm::EngineBuilder builder(std::move(module_));
    std::string triple, mcpu, mattr;
    llvm::TargetOptions opt;