Skip to content

Commit

Permalink
Switch mlir bindings from pybind11 to nanobind
Browse files Browse the repository at this point in the history
PiperOrigin-RevId: 709143995
  • Loading branch information
Google-ML-Automation committed Dec 23, 2024
1 parent 3c79b98 commit 347ef1b
Showing 1 changed file with 12 additions and 10 deletions.
22 changes: 12 additions & 10 deletions jaxlib/mlir/_mlir_libs/BUILD.bazel
Original file line number Diff line number Diff line change
Expand Up @@ -70,8 +70,8 @@ py_extension(
":jaxlib_mlir_capi_shared_library",
"@llvm-project//mlir:CAPIGPUHeaders",
"@llvm-project//mlir:CAPIIRHeaders",
"@llvm-project//mlir:MLIRBindingsPythonHeaders",
"@pybind11",
"@llvm-project//mlir:MLIRBindingsPythonNanobindHeaders",
"@nanobind",
],
)

Expand All @@ -85,7 +85,8 @@ py_extension(
deps = [
":jaxlib_mlir_capi_shared_library",
"@llvm-project//mlir:CAPIGPUHeaders",
"@pybind11",
"@llvm-project//mlir:MLIRBindingsPythonNanobindHeaders",
"@nanobind",
],
)

Expand All @@ -100,8 +101,8 @@ py_extension(
":jaxlib_mlir_capi_shared_library",
"@llvm-project//mlir:CAPIIRHeaders",
"@llvm-project//mlir:CAPINVGPUHeaders",
"@llvm-project//mlir:MLIRBindingsPythonHeaders",
"@pybind11",
"@llvm-project//mlir:MLIRBindingsPythonNanobindHeaders",
"@nanobind",
],
)

Expand All @@ -116,8 +117,8 @@ py_extension(
":jaxlib_mlir_capi_shared_library",
"@llvm-project//mlir:CAPIIRHeaders",
"@llvm-project//mlir:CAPILLVMHeaders",
"@llvm-project//mlir:MLIRBindingsPythonHeaders",
"@pybind11",
"@llvm-project//mlir:MLIRBindingsPythonNanobindHeaders",
"@nanobind",
],
)

Expand All @@ -131,8 +132,8 @@ py_extension(
deps = [
":jaxlib_mlir_capi_shared_library",
"@llvm-project//mlir:CAPISparseTensorHeaders",
"@llvm-project//mlir:MLIRBindingsPythonHeaders",
"@pybind11",
"@llvm-project//mlir:MLIRBindingsPythonNanobindHeaders",
"@nanobind",
],
)

Expand All @@ -146,7 +147,8 @@ py_extension(
deps = [
":jaxlib_mlir_capi_shared_library",
"@llvm-project//mlir:CAPISparseTensorHeaders",
"@pybind11",
"@llvm-project//mlir:MLIRBindingsPythonNanobindHeaders",
"@nanobind",
],
)

Expand Down

0 comments on commit 347ef1b

Please sign in to comment.