porting changes from 0.8.0 to 0.8.2 #601
Ruturaj4
merged
19 commits into
rocm-jaxlib-v0.8.2
from
cherry-8.2
Remove nvidia_wheel_versions
f4ffb74a
Make jaxlib targets visible
3d2c2a4b
hipblas typedef fix
5ff5b208
No GPU fail
078defc0
Wrap HIP inline functions in anonymous namespaces in vendor.h
d966d0e4
SWDEV-512768 - Replace hipGetLastError with hipExtGetLastError
c13ef105
Add shared utility function get_rocm_version to test_util.py
de13147d
Fix hipSparse CSR algorithm mappings for ROCm 7
a2abd8e5
Make nvidia version data optional for ROCm builds
2335b793
Fix v_pages quantization and adjust test params for ROCm compatibilit…
a77f3c78
Address LLVM assertion failure due to a multithreaded use. Update .gi…
47389d3c
Add skip of test_is_finite() on Cuda (#565)
14883422
Add rocm test requirements file (#570)
f1d95732
Let the unit tests use build.py for setting up Bazel commands for uni…
ca938480
adding abort logic to rocm/jax (#590)
4e25e347
Skip is_finite tests on ROCm (not in Triton lowering for jax 0.8.0) (…
f3ccd49b
Fix shared memory limit check for ROCm in test_dot (#596)
294da7df
Fix Numpy signatures test (#598)
48cb195a
Fix GPU lowering rule for SVD on ROCm devices (#600)
78547274
jayfurmanek
approved these changes
on 2026-01-15
Login to write a write a comment.
Login via GitHub