Skip to content

[mlir] free threading support for mlir python bindings #105522

Closed
@vfdev-5

Description

@vfdev-5

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.

Metadata

Metadata

Assignees

No one assigned

    Labels

    Type

    No type

    Projects

    No projects

    Milestone

    No milestone

    Relationships

    None yet

    Development

    No branches or pull requests

    Issue actions