Add python_version input to jax release workflow#70
Merged
Conversation
1 task
ScottTodd
added a commit
to ROCm/TheRock
that referenced
this pull request
Jun 30, 2026
## Motivation Progress on #5634 and #6218. Follow-up to these PRs: * #4366 * #5727 This brings PyTorch and JAX back into alignment with optional triggering, a script to define the job matrix, and the ability to trigger builds for a single python version at a time to save CI resources. ## Technical Details > [!IMPORTANT] > The `python_version` input will need to be added to https://github.com/ROCm/rockrel/blob/main/.github/workflows/multi_arch_release_linux_jax_wheels.yml, like with ROCm/rockrel#54 and ROCm/rockrel#55. > > I've prepared this PR for that: ROCm/rockrel#70. ## Test Plan * New unit tests * Trigger a dev release with `multi_arch_release_linux.yml`, setting `build_jax` to `false` * https://github.com/ROCm/TheRock/actions/runs/28469583076 * Trigger `multi_arch_release_linux_jax_wheels.yml` for a single `python_version` * https://github.com/ROCm/TheRock/actions/runs/28469829633 ## Submission Checklist - [x] Look over the contributing guidelines at https://github.com/ROCm/ROCm/blob/develop/CONTRIBUTING.md#pull-requests.
This file contains hidden or bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment
2 participants
Add this suggestion to a batch that can be applied as a single commit.This suggestion is invalid because no changes were made to the code.Suggestions cannot be applied while the pull request is closed.Suggestions cannot be applied while viewing a subset of changes.Only one suggestion per line can be applied in a batch.Add this suggestion to a batch that can be applied as a single commit.Applying suggestions on deleted lines is not supported.You must change the existing code in this line in order to create a valid suggestion.Outdated suggestions cannot be applied.This suggestion has been applied or marked resolved.Suggestions cannot be applied from pending reviews.Suggestions cannot be applied on multi-line comments.Suggestions cannot be applied while the pull request is queued to merge.Suggestion cannot be applied right now. Please check back later.
Motivation
Progress on ROCm/TheRock#5634 and ROCm/TheRock#6218.
This will be used by ROCm/TheRock#6117 which applies what ROCm/TheRock#5727 added for pytorch to jax: an optional
build_jaxtoggle and thepython_versioninput to build for a single version instead of all versions.Test Plan
Dev release in TheRock: https://github.com/ROCm/TheRock/actions/runs/28469583076
Submission Checklist