diff --git a/jaxlib/mlir/_mlir_libs/BUILD.bazel b/jaxlib/mlir/_mlir_libs/BUILD.bazel index 511a2310b3b4..817c23a11636 100644 --- a/jaxlib/mlir/_mlir_libs/BUILD.bazel +++ b/jaxlib/mlir/_mlir_libs/BUILD.bazel @@ -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", ], ) @@ -85,7 +85,8 @@ py_extension( deps = [ ":jaxlib_mlir_capi_shared_library", "@llvm-project//mlir:CAPIGPUHeaders", - "@pybind11", + "@llvm-project//mlir:MLIRBindingsPythonNanobindHeaders", + "@nanobind", ], ) @@ -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", ], ) @@ -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", ], ) @@ -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", ], ) @@ -146,7 +147,8 @@ py_extension( deps = [ ":jaxlib_mlir_capi_shared_library", "@llvm-project//mlir:CAPISparseTensorHeaders", - "@pybind11", + "@llvm-project//mlir:MLIRBindingsPythonNanobindHeaders", + "@nanobind", ], )