Skip to content

0.8.2 ci_build to check dev tarball#293

Merged
kiran-thumma merged 2 commits intorocm-jaxlib-v0.8.2from
rocm-jaxlib-v0.8.2-fixdevtar
Feb 11, 2026
Merged

0.8.2 ci_build to check dev tarball#293
kiran-thumma merged 2 commits intorocm-jaxlib-v0.8.2from
rocm-jaxlib-v0.8.2-fixdevtar

Conversation

@kiran-thumma
Copy link
Contributor

@kiran-thumma kiran-thumma commented Feb 5, 2026

Motivation

Fix packaging.version.InvalidVersion errors during JAX wheel builds caused by ROCm version strings containing multiple + characters (e.g., 7.12.0.dev0+<sha>), which violate PEP 440 version specifier rules.

Technical Details

Modified jax_rocm_plugin/build/rocm/ci_build to sanitize rocm_version by replacing + with . for PEP 440 compliance:

  1. Line 60: Sanitize when setting ML_WHEEL_VERSION_SUFFIX for Bazel jaxlib builds
  2. Line 178: Sanitize when setting ROCM_VERSION_EXTRA environment variable for plugin/pjrt wheel builds

PEP 440 local version identifiers only allow one + character, and the local segment can only contain alphanumerics and periods. This ensures wheel versions like 0.8.0+rocm7.12.0.dev0.6475192225f38160e614c71cc0543aed2be8fa03 are valid.

Test Plan

  • Build JAX wheels with dev rocm_version containing + (e.g., 7.12.0.dev0+6475192225f38160e614c71cc0543aed2be8fa03)
  • Verify jaxlib, jax-rocm7-plugin, and jax-rocm7-pjrt wheels build successfully
  • Verify wheel filenames contain sanitized version strings

Test Result

Submission Checklist

Cherry-picked from: #292

@kiran-thumma kiran-thumma changed the title update ci_build to check dev tarball 0.8.2 ci_build to check dev tarball Feb 5, 2026
@kiran-thumma kiran-thumma merged commit 8d0d9be into rocm-jaxlib-v0.8.2 Feb 11, 2026
18 checks passed
JeniferC99 pushed a commit that referenced this pull request Mar 3, 2026
…-encoding fix

- build/ci_build: replace '+' with '.' in rocm_ver_tag for valid Docker tags
- jax_rocm_plugin/build/rocm/ci_build: same for image name and ROCM_VERSION_EXTRA
- get_rocm.py: unquote then quote therock URL to avoid double-encoding; create amdgcn symlink only if missing

Made-with: Cursor
JeniferC99 pushed a commit that referenced this pull request Mar 3, 2026
…-encoding fix

- build/ci_build: replace '+' with '.' in rocm_ver_tag and rocm_version_tag for valid Docker tags
- jax_rocm_plugin/build/rocm/ci_build: same for image name and ROCM_VERSION_EXTRA
- get_rocm.py: unquote then quote therock URL to avoid double-encoding; create amdgcn symlink only if missing

Made-with: Cursor
JeniferC99 pushed a commit that referenced this pull request Mar 3, 2026
…-encoding fix

- build/ci_build: replace '+' with '.' in rocm_ver_tag and rocm_version_tag for valid Docker tags
- jax_rocm_plugin/build/rocm/ci_build: ROCM_VERSION_EXTRA use replace('+', '.')
- tools/get_rocm.py: unquote then quote therock URL to avoid double-encoding; create amdgcn symlink only if missing

Made-with: Cursor
JeniferC99 added a commit that referenced this pull request Mar 3, 2026
…-encoding fix (#333)

* Cherry-pick PR #293 + #306: dev tarball Docker tag fix and double URL-encoding fix

- build/ci_build: replace '+' with '.' in rocm_ver_tag for valid Docker tags
- jax_rocm_plugin/build/rocm/ci_build: same for image name and ROCM_VERSION_EXTRA
- get_rocm.py: unquote then quote therock URL to avoid double-encoding; create amdgcn symlink only if missing

Made-with: Cursor

* Fix pylint C0301: shorten comment line (line-too-long 103/100)

Made-with: Cursor

* Use try/except FileExistsError for amdgcn symlink (PR #330)

Made-with: Cursor

---------

Co-authored-by: Jenkins <jenkins-compute@amd.com>
JeniferC99 added a commit that referenced this pull request Mar 3, 2026
…-encoding fix (#335)

* Cherry-pick PR #293 + #306: dev tarball Docker tag fix and double URL-encoding fix

- build/ci_build: replace '+' with '.' in rocm_ver_tag and rocm_version_tag for valid Docker tags
- jax_rocm_plugin/build/rocm/ci_build: ROCM_VERSION_EXTRA use replace('+', '.')
- tools/get_rocm.py: unquote then quote therock URL to avoid double-encoding; create amdgcn symlink only if missing

Made-with: Cursor

* Fix pylint C0301: shorten comment line (line-too-long 103/100)

Made-with: Cursor

* Use try/except FileExistsError for amdgcn symlink (PR #330)

Made-with: Cursor

---------

Co-authored-by: Jenkins <jenkins-compute@amd.com>
JeniferC99 added a commit that referenced this pull request Mar 4, 2026
…-encoding fix (#334)

* Cherry-pick PR #293 + #306: dev tarball Docker tag fix and double URL-encoding fix

- build/ci_build: replace '+' with '.' in rocm_ver_tag and rocm_version_tag for valid Docker tags
- jax_rocm_plugin/build/rocm/ci_build: same for image name and ROCM_VERSION_EXTRA
- get_rocm.py: unquote then quote therock URL to avoid double-encoding; create amdgcn symlink only if missing

Made-with: Cursor

* Fix pylint C0301: shorten comment line (line-too-long 103/100)

Made-with: Cursor

* Use try/except FileExistsError for amdgcn symlink (PR #330)

Made-with: Cursor

* Update get_rocm.py

* Update get_rocm.py

---------

Co-authored-by: Jenkins <jenkins-compute@amd.com>
gulsumgudukbay pushed a commit that referenced this pull request Mar 10, 2026
…-encoding fix (#335)

* Cherry-pick PR #293 + #306: dev tarball Docker tag fix and double URL-encoding fix

- build/ci_build: replace '+' with '.' in rocm_ver_tag and rocm_version_tag for valid Docker tags
- jax_rocm_plugin/build/rocm/ci_build: ROCM_VERSION_EXTRA use replace('+', '.')
- tools/get_rocm.py: unquote then quote therock URL to avoid double-encoding; create amdgcn symlink only if missing

Made-with: Cursor

* Fix pylint C0301: shorten comment line (line-too-long 103/100)

Made-with: Cursor

* Use try/except FileExistsError for amdgcn symlink (PR #330)

Made-with: Cursor

---------

Co-authored-by: Jenkins <jenkins-compute@amd.com>
(cherry picked from commit ea9a021)
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