Skip to content

Use xla from rocm-jaxlib-v0.8.0 release branch#599

Open
alekstheod wants to merge 1 commit intorocm-jaxlib-v0.8.0from
switch_to_xla_from_release_branch
Open

Use xla from rocm-jaxlib-v0.8.0 release branch#599
alekstheod wants to merge 1 commit intorocm-jaxlib-v0.8.0from
switch_to_xla_from_release_branch

Conversation

@alekstheod
Copy link

Use xla from internal rocm release branch rocm-jaxlib-v0.8.0

@gulsumgudukbay
Copy link

@alekstheod could you please update the PR comment with the goal of this PR? why was this needed?

@alekstheod
Copy link
Author

alekstheod commented Jan 16, 2026

@gulsumgudukbay
It seems we do build jax from rocm-jaxlib-v0.8.0 with an old version of xla by default. I am not sure if that is intentional but I believe we shall be using xla from rocm-jaxlib-v0.8.0 instead

@gulsumgudukbay
Copy link

@gulsumgudukbay It seems we do build jax from rocm-jaxlib-v0.8.0 with an old version of xla by default. I am not sure if that is intentional but I believe we shall be using xla from rocm-jaxlib-v0.8.0 instead

since we weren't using the build scripts from this repo and we migrated to rocm-jax, that wasn't updated.

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

2 participants