You signed in with another tab or window. Reload to refresh your session.You signed out in another tab or window. Reload to refresh your session.You switched accounts on another tab or window. Reload to refresh your session.Dismiss alert
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.