Update to rocm-jaxlib-v0.9.1 (#64)

cerisier · web-flow · commit ffd8138f9b05 · 2026-03-19T14:20:32.000+01:00
* Update to v0.9.1 rocm

* fix
diff --git a/openxla/commits.env b/openxla/commits.env
@@ -1,2 +1,2 @@
-XLA_COMMIT=9e9a0fb13b164c9875349246d1d8616a267c9bf5
-ROCM_XLA_COMMIT=c40171e49b63e8a12498fae2b64f90c2406802ec
+XLA_COMMIT=8352d6be49e512b1a422e693c17973d010939ea6
+ROCM_XLA_COMMIT=67f07743e4e80743315f7e4ddc0031eeb02b5f1a
diff --git a/openxla/patches/rocm/0002-Use-hermetic-cc-toolchain-for-Linux-CPU-use-glibc-2..patch b/openxla/patches/rocm/0002-Use-hermetic-cc-toolchain-for-Linux-CPU-use-glibc-2..patch
@@ -17,7 +17,7 @@ index 8d7f1384d8..8cd8df86a0 100644
  
 +HERMETIC_CC_TOOLCHAIN_VERSION = "v3.1.1"
 +
-+http_archive(
++tf_http_archive(
 +    name = "hermetic_cc_toolchain",
 +    sha256 = "907745bf91555f77e8234c0b953371e6cac5ba715d1cf12ff641496dd1bce9d1",
 +    urls = [
diff --git a/openxla/patches/rocm/0003-Fixed-issue-with-passing-arch_name-in-CreateTritonPi.patch b/openxla/patches/rocm/0003-Fixed-issue-with-passing-arch_name-in-CreateTritonPi.patch
diff --git a/tools/resolve-openxla-heads.sh b/tools/resolve-openxla-heads.sh
@@ -3,7 +3,7 @@
 set -euo pipefail
 
 xla_branch="main"
-rocm_xla_branch="rocm-jaxlib-v0.8.0"
+rocm_xla_branch="rocm-jaxlib-v0.9.1"
 
 xla_commit="$(
   gh api "repos/openxla/xla/commits/${xla_branch}" --jq '.sha'

Original file line number	Diff line number	Diff line change
`@@ -17,7 +17,7 @@ index 8d7f1384d8..8cd8df86a0 100644`
`17`	`17`
`18`	`18`	`+HERMETIC_CC_TOOLCHAIN_VERSION = "v3.1.1"`
`19`	`19`	`+`
`20`		`-+http_archive(`
	`20`	`++tf_http_archive(`
`21`	`21`	`+ name = "hermetic_cc_toolchain",`
`22`	`22`	`+ sha256 = "907745bf91555f77e8234c0b953371e6cac5ba715d1cf12ff641496dd1bce9d1",`
`23`	`23`	`+ urls = [`