Closed
Description
Hello,
I wanted to ask whether there are plans to add free-threading support to MLIR python bindings?
Essentially, we may need to update the extensions this way (link):
- PYBIND11_MODULE(_mlirDialectsGPU, m) {
+ PYBIND11_MODULE(_mlirDialectsGPU, m, py::mod_gil_not_used()) {
I can help with providing a draft PR for that and we can iterate over if there are more updates required.
Thanks!
Context:
We are adding free-threading support to JAX: jax-ml/jax#23073.