ROCm/jax

Pull Requests Commits

Enabled ToeplitzSymmetricConstruction unit tests for ROCm devices as certain data types are supported. All relevant tests either pass or skip.

tsrw2048 committed 14 days ago

c7359545

Enabled testTridiagonal to run on ROCm devices (#607)

tsrw2048 committed 14 days ago

Verified 4f2e28ba

Fix GPU lowering rule for SVD on ROCm devices (#600)

tsrw2048 committed 16 days ago

Verified 7134b1f4

Fix Numpy signatures test (#598)

magaonka-amd committed 16 days ago

Verified a3f8af53

Fix shared memory limit check for ROCm in test_dot (#596)

phambinhfin committed 16 days ago

Verified b799d3a9

Skip is_finite tests on ROCm (not in Triton lowering for jax 0.8.0) (#597)

phambinhfin committed 16 days ago

Verified fb45b3ac

adding abort logic to rocm/jax (#590)

gulsumgudukbay committed 17 days ago

Verified ca39e75c

Fix psend test case failures (#595)

AratiGanesh committed 18 days ago

Verified e4f54fc9

GESVDJ support for ROCm GPUs in JAX (#580)

tsrw2048 committed 18 days ago

Verified 9dbc4119

Let the unit tests use build.py for setting up Bazel commands for unit tests (#582)

charleshofer committed 46 days ago

Verified ad54f8b1

Add rocm test requirements file (#570)

AratiGanesh committed 46 days ago

Verified 552778a4

add fix for jaxlib build whl (#571)

kiran-thumma committed 50 days ago

Verified a1a217fa

Remove build scripts (#578)

gulsumgudukbay committed 50 days ago

Verified a1968610

Revert "Remove old scripts and break build for ROCm" (#575)

gulsumgudukbay committed 50 days ago

Verified bf4facd8

Remove old scripts and break build for ROCm (#569)

gulsumgudukbay committed 51 days ago

Verified cafe7943

register rocm platform to mx datatype lowering path (#567)

ScXfjiang committed 56 days ago

Verified a2b972f2

Add skip of test_is_finite() on Cuda (#565)

Arech8 committed 65 days ago

Verified 861cc763

Implement Pallas lowering of lax.is_finite() for Triton (#564)

Arech8 committed 65 days ago

Verified c972c55d

Address LLVM assertion failure due to a multithreaded use. Update .gitignore (#563)

Arech8 committed 65 days ago

Verified 339bf1e8

Fix v_pages quantization and adjust test params for ROCm compatibilit… (#560)

phambinhfin committed 72 days ago

Verified f893175a

Fix ROCm lib names copy pattern to use wildcard version

phambinhfin committed 73 days ago

709963b8

Make nvidia version data optional for ROCm builds

phambinhfin committed 73 days ago

b5762449

Fix hipSparse CSR algorithm mappings for ROCm 7

phambinhfin committed 73 days ago

28e04303

Add shared utility function get_rocm_version to test_util.py

charleshofer committed 73 days ago

aac9e9d2

SWDEV-512768 - Replace hipGetLastError with hipExtGetLastError

dsicarov-amd committed 73 days ago

954fde6a

Wrap HIP inline functions in anonymous namespaces in vendor.h

AratiGanesh committed 73 days ago

13a4d047

charleshofer committed 73 days ago

f4df93f6

hipblas typedef fix

charleshofer committed 73 days ago

9ed3820b

Make jaxlib targets visible

charleshofer committed 73 days ago

3f0a5750

Remove nvidia_wheel_versions

charleshofer committed 73 days ago

f037da4e

Older