jax
Port commits for UT fixes from rocm-jaxlib-v0.8.0 to rocm-jaxlib-v0.8.2: condition number, rnn workspace, and Pallas Triton tests
#650
Merged
Merged
Port commits for UT fixes from rocm-jaxlib-v0.8.0 to rocm-jaxlib-v0.8.2: condition number, rnn workspace, and Pallas Triton tests #650
Ruturaj4
merged
3 commits into
rocm-jaxlib-v0.8.2
from
move-commits-v0.8.0-to-v0.8.2
Enabled test for condition number on ROCm devices. (#613)
c4654ea4
Enabled RNN unit test: test_no_workspace_overflow for ROCm devices (#…
7e1ee069
Added changes from PR #626 and PR #645. This also fixes merge conflic…
e813a415
gulsumgudukbay
approved these changes
on 2026-01-23
Ruturaj4
approved these changes
on 2026-01-23
Reviewers
Assignees
No one assigned
Labels
open-upstream
cherry-pick-candidate
Milestone
No milestone
Login to write a write a comment.
Login via GitHub