What is the corresponding XLA version (commit) that works with the latest JAX version? #31898
Unanswered
Hanjiang9831
asked this question in
Q&A
Replies: 1 comment 1 reply
-
|
The XLA commit used by JAX is pinned here: https://github.com/jax-ml/jax/blob/main/third_party/xla/revision.bzl and updated nightly. |
Beta Was this translation helpful? Give feedback.
1 reply
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment
Uh oh!
There was an error while loading. Please reload this page.
-
I forked from JAX/xla a while ago (5/6 months ago) to experiment on my custom backend, and I found out that the version I forked from has issue initiating cpu + custom backend heterogenous computation on multihost scenarios. After more experiments I found out the CPU client in that version does not use the modern C API.
After checking the latest version, I found out the xla.cc that was in JAX/jaxlib is now gone, which indicates significant project changes, therefore I want to switch to work on the latest version (or a very recent one), so can anyone give me the very recent commits for JAX/xla that works fine together?
Beta Was this translation helpful? Give feedback.
All reactions