Skip to content

Skip testEighTinyNorm test 0.8.0#695

Merged
AratiGanesh merged 2 commits intorocm-jaxlib-v0.8.0from
skip-eigh-tiny-norm
Feb 10, 2026
Merged

Skip testEighTinyNorm test 0.8.0#695
AratiGanesh merged 2 commits intorocm-jaxlib-v0.8.0from
skip-eigh-tiny-norm

Conversation

@AratiGanesh
Copy link

@AratiGanesh AratiGanesh commented Feb 6, 2026

Motivation

The testEighTinyNorm test fails on ROCm devices due to numerical precision issues in the underlying hipSolver https://ontrack-internal.amd.com/browse/SWDEV-578349

Technical Details

Added @jtu.skip_on_devices("rocm") decorator to skip testEighTinyNorm in `tests/linalg_test.py

Test Result

image

Submission Checklist
Look over the contributing guidelines at https://github.com/ROCm/ROCm/blob/develop/CONTRIBUTING.md#pull-requests.

@AratiGanesh AratiGanesh requested a review from a team February 6, 2026 21:56
@magaonka-amd
Copy link

magaonka-amd commented Feb 6, 2026

okay LGTM but one suggestion
can you add TODO similar to this : https://github.com/jax-ml/jax/pull/34610/changes#diff-f84d338951e2972a9623500032e6e479a30ad3d970e034a2cab1ca49cbf3156dR143

I was asked to do this change in upstream so I don't want you to get the same comment once you push upstream.

other than that changes looks okay.

something like

# numerical errors seen as of ROCm 7.2 due to hipsparse issue
# TODO: re-enable the test once the hipsparse is fixed

@AratiGanesh AratiGanesh changed the title Skip testEighTinyNorm test Skip testEighTinyNorm test 0.8.0 Feb 9, 2026
@AratiGanesh AratiGanesh added open-upstream Tag when you want a copy of this PR to be opened on upstream cherry-pick-candidate Mark a PR to be cherry-picked into the next ROCm JAX. Remove IIF the latest upstream contain the PR. labels Feb 9, 2026
@AratiGanesh AratiGanesh merged commit 05ffc4b into rocm-jaxlib-v0.8.0 Feb 10, 2026
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

cherry-pick-candidate Mark a PR to be cherry-picked into the next ROCm JAX. Remove IIF the latest upstream contain the PR. open-upstream Tag when you want a copy of this PR to be opened on upstream

Projects

None yet

Development

Successfully merging this pull request may close these issues.

3 participants

Comments