jax
[0.5.0-UT] Skip CSR matmat and matvec float tests on ROCm <6.4 (NaN issue with beta==0)
#380

Merged

[0.5.0-UT] Skip CSR matmat and matvec float tests on ROCm <6.4 (NaN issue with beta==0) #380

mrodden merged 3 commits into rocm-jaxlib-v0.5.0 from bugfix-csr-beta0-nan-0.5.0-ut

Ruturaj4 approved these changes on 2025-04-23

Ruturaj4 requested changes on 2025-04-23

mrodden commented on 2025-04-23

Skip CSR matmat and matvec float tests on ROCm <6.4 (NaN issue with b…

ee8b6c58

psanal35 force pushed from efd56ee6 to ee8b6c58 251 days ago

Use ROCM_PATH if set with fallback to /opt/rocm, replace assert, note…

acc7cf7e

psanal35 force pushed from 7632093d to acc7cf7e 251 days ago

psanal35 requested a review from

Ruturaj4 250 days ago

psanal35 requested a review from

mrodden 250 days ago

Ruturaj4 approved these changes on 2025-04-23

mrodden changed the title ~~[BUGFIX][0.5.0-UT] Skip CSR matmat and matvec float tests on ROCm <6.4 (NaN issue with beta==0)~~ [0.5.0-UT] Skip CSR matmat and matvec float tests on ROCm <6.4 (NaN issue with beta==0) 250 days ago

mrodden requested changes on 2025-04-24

Generalize dtype check for future-proofing and fix import order

51b4c676

mrodden approved these changes on 2025-04-24

mrodden merged 5582e2aa into rocm-jaxlib-v0.5.0 249 days ago

Reviewers

Ruturaj4

mrodden

Assignees

No one assigned

Labels

None yet

Milestone

No milestone

jax [0.5.0-UT] Skip CSR matmat and matvec float tests on ROCm <6.4 (NaN issue with beta==0) #380 Merged

[0.5.0-UT] Skip CSR matmat and matvec float tests on ROCm <6.4 (NaN issue with beta==0) #380

jax
[0.5.0-UT] Skip CSR matmat and matvec float tests on ROCm <6.4 (NaN issue with beta==0)
#380

Merged