Fix ExecutionEngine post-update in upstream LLVM

LLVM r367686 changed the locking scheme to avoid potential deadlocks and the
related llvm::orc::ThreadSafeModule APIs ExecutionEngine was relying upon,
breaking the MLIR build.  Update our use of ThreadSafeModule to unbreak the
build.

PiperOrigin-RevId: 261566571

diff --git a/mlir/lib/ExecutionEngine/ExecutionEngine.cpp b/mlir/lib/ExecutionEngine/ExecutionEngine.cpp
index 0317a92..b830c53 100644 (file)
--- a/mlir/lib/ExecutionEngine/ExecutionEngine.cpp
+++ b/mlir/lib/ExecutionEngine/ExecutionEngine.cpp
@@ -183,7 +183,9 @@ private:
       (void)resp;
       if (!irTransformer)
         return std::move(module);
-      if (Error err = irTransformer(module.getModule()))
+      Error err = module.withModuleDo(
+          [this](llvm::Module &module) { return irTransformer(&module); });
+      if (err)
         return std::move(err);
       return std::move(module);
     };