Update the way to create pool.

Signed-off-by: Haruki Imai <[email protected]>
onnx · Jan 23, 2025 · 2414108 · 2414108
1 parent da52a06
commit 2414108
Showing 1 changed file with 6 additions and 4 deletions.
diff --git 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");