[Mlir-commits] [mlir] Added PyThreadPool as wrapper around MlirLlvmThreadPool in MLIR python bindings (PR #130109)

Mehdi Amini llvmlistbot at llvm.org
Thu Mar 6 11:39:14 PST 2025


================
@@ -47,3 +47,17 @@
 assert '"mlir.ir.Context._CAPIPtr"' in repr(c4_capsule)
 c5 = mlir.ir.Context._CAPICreate(c4_capsule)
 assert c4 is c5
+c4 = None
+c5 = None
+gc.collect()
+
+# Create a global threadpool and use it in two contexts
+tp = mlir.ir.ThreadPool()
+assert tp.get_max_concurrency() > 0
+c5 = mlir.ir.Context()
+c5.enable_multithreading(False)
----------------
joker-eph wrote:

We can also take it as optional at construction indeed!

But can't there be cases where you handed over an existing context and this "set_thread_pool" API can be also useful?

https://github.com/llvm/llvm-project/pull/130109


More information about the Mlir-commits mailing list