From 2414108288f51ac4a182da83046af024a96ae54a Mon Sep 17 00:00:00 2001
From: Haruki Imai <imaihal@jp.ibm.com>
Date: Thu, 23 Jan 2025 01:46:17 -0500
Subject: [PATCH] Update the way to create pool.

Signed-off-by: Haruki Imai <imaihal@jp.ibm.com>
---
 src/onnx-mlir.cpp | 10 ++++++----
 1 file changed, 6 insertions(+), 4 deletions(-)

diff --git a/src/onnx-mlir.cpp b/src/onnx-mlir.cpp
index 762fcbb5c2..b6734a8ecc 100644
--- a/src/onnx-mlir.cpp
+++ b/src/onnx-mlir.cpp
@@ -70,10 +70,12 @@ int main(int argc, char *argv[]) {
 
   // Create context after MLIRContextCLOptions are registered and parsed.
   mlir::MLIRContext context(mlir::MLIRContext::Threading::DISABLED);
-  llvm::DefaultThreadPool pool(
-      llvm::hardware_concurrency(compilationNumThreads));
-  if (compilationNumThreads > 1)
-    context.setThreadPool(pool);
+  std::unique_ptr<llvm::ThreadPoolInterface> threadPoolPtr;
+  if (compilationNumThreads > 1) {
+    threadPoolPtr = std::make_unique<llvm::DefaultThreadPool>(
+        llvm::hardware_concurrency(compilationNumThreads));
+    context.setThreadPool(*threadPoolPtr);
+  }
 
   if (!context.isMultithreadingEnabled()) {
     assert(context.getNumThreads() == 1 && "1 thread if no multithreading");