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

Merged

Commits

Skip CSR matmat and matvec float tests on ROCm <6.4 (NaN issue with beta==0)

psanal35 committed 251 days ago
Use ROCM_PATH if set with fallback to /opt/rocm, replace assert, note fix in ROCm 6.4.0

psanal35 committed 251 days ago
Generalize dtype check for future-proofing and fix import order

psanal35 committed 250 days ago