Skip to content

Commit aeb2989

Browse files
hawkinspGoogle-ML-Automation
authored andcommitted
Migrate StableHLO Python extension to nanobind.
I'm working towards moving the MLIR Python core code to use nanobind instead of pybind11: * llvm/llvm-project#117922, which was merged recently, allows downstream Python dialect extensions to be defined using either pybind11 or nanobind. * llvm/llvm-project#118583 is a PR in review that ports the Python core code to use nanobind instead of pybind11. This PR migrates StableHLO and related dialects to use nanobind rather than pybind11, with the goal of migrating JAX away from pybind11. PiperOrigin-RevId: 705197067
1 parent 05ad393 commit aeb2989

File tree

1 file changed

+4
-4
lines changed

1 file changed

+4
-4
lines changed

jaxlib/mlir/_mlir_libs/BUILD.bazel

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -333,9 +333,9 @@ py_extension(
333333
deps = [
334334
":jaxlib_mlir_capi_shared_library",
335335
"@llvm-project//mlir:CAPIIRHeaders",
336-
"@llvm-project//mlir:MLIRBindingsPythonHeaders",
336+
"@llvm-project//mlir:MLIRBindingsPythonNanobindHeaders",
337337
"@local_config_python//:headers",
338-
"@pybind11",
338+
"@nanobind",
339339
"@stablehlo//:chlo_capi_headers",
340340
],
341341
)
@@ -351,9 +351,9 @@ py_extension(
351351
":jaxlib_mlir_capi_shared_library",
352352
"@llvm-project//llvm:Support",
353353
"@llvm-project//mlir:CAPIIRHeaders",
354-
"@llvm-project//mlir:MLIRBindingsPythonHeaders",
354+
"@llvm-project//mlir:MLIRBindingsPythonNanobindHeaders",
355355
"@local_config_python//:headers",
356-
"@pybind11",
356+
"@nanobind",
357357
"@stablehlo//:stablehlo_capi_headers",
358358
],
359359
)

0 commit comments

Comments
 (0)