Skip to content

Commit 3d54d03

Browse files
hawkinspGoogle-ML-Automation
authored andcommitted
Migrate StableHLO Python extension to nanobind.
PiperOrigin-RevId: 707543869
1 parent 66c8a2e commit 3d54d03

File tree

2 files changed

+6
-6
lines changed

2 files changed

+6
-6
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
)

third_party/xla/workspace.bzl

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -21,8 +21,8 @@ load("//third_party:repo.bzl", "tf_http_archive", "tf_mirror_urls")
2121
# curl -L https://github.com/openxla/xla/archive/<git hash>.tar.gz | sha256sum
2222
# and update XLA_SHA256 with the result.
2323

24-
XLA_COMMIT = "20a482597b7dd3067b26ca382b88084ee5a21cf7"
25-
XLA_SHA256 = "0870fcd86678cae31c56cfc57018f52ceec8e4691472af62c847ade746a0eb13"
24+
XLA_COMMIT = "0c9760db4399ded975e118f310f6976a60fcdc74"
25+
XLA_SHA256 = "bc5fdb66150557a5eaed3846d6f844ef9e82b986cedde3817496075c5bbb2eb1"
2626

2727
def repo():
2828
tf_http_archive(

0 commit comments

Comments
 (0)