ROCm/jax

Pull Requests Commits

Enable Pallas tests on ROCm

AratiGanesh committed 28 days ago

1521ffff

Fix KeyError for bytes_reservable_limit on ROCm

magaonka-amd committed 30 days ago

bb524b06

Enable testSvdSubsetByIndex for ROCm with subset_by_index skip (#603)

phambinhfin committed 31 days ago

Verified 894ba31f

Enabled ToeplitzSymmetricConstruction unit tests for ROCm devices (#608)

tsrw2048 committed 31 days ago

Verified cf97819a

Enabled testTridiagonal to run on ROCm devices (#607)

tsrw2048 committed 31 days ago

Verified 4f2e28ba

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

tsrw2048 committed 33 days ago

Verified 7134b1f4

Fix Numpy signatures test (#598)

magaonka-amd committed 33 days ago

Verified a3f8af53

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

phambinhfin committed 33 days ago

Verified b799d3a9

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

phambinhfin committed 33 days ago

Verified fb45b3ac

adding abort logic to rocm/jax (#590)

gulsumgudukbay committed 35 days ago

Verified ca39e75c

Fix psend test case failures (#595)

AratiGanesh committed 35 days ago

Verified e4f54fc9

GESVDJ support for ROCm GPUs in JAX (#580)

tsrw2048 committed 36 days ago

Verified 9dbc4119

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

charleshofer committed 63 days ago

Verified ad54f8b1

Add rocm test requirements file (#570)

AratiGanesh committed 64 days ago

Verified 552778a4

add fix for jaxlib build whl (#571)

kiran-thumma committed 67 days ago

Verified a1a217fa

Remove build scripts (#578)

gulsumgudukbay committed 67 days ago

Verified a1968610

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

gulsumgudukbay committed 68 days ago

Verified bf4facd8

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

gulsumgudukbay committed 68 days ago

Verified cafe7943

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

ScXfjiang committed 74 days ago

Verified a2b972f2

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

Arech8 committed 83 days ago

Verified 861cc763

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

Arech8 committed 83 days ago

Verified c972c55d

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

Arech8 committed 83 days ago

Verified 339bf1e8

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

phambinhfin committed 89 days ago

Verified f893175a

Fix ROCm lib names copy pattern to use wildcard version

phambinhfin committed 90 days ago

709963b8

Make nvidia version data optional for ROCm builds

phambinhfin committed 90 days ago

b5762449

Fix hipSparse CSR algorithm mappings for ROCm 7

phambinhfin committed 90 days ago

28e04303

Add shared utility function get_rocm_version to test_util.py

charleshofer committed 90 days ago

aac9e9d2

SWDEV-512768 - Replace hipGetLastError with hipExtGetLastError

dsicarov-amd committed 90 days ago

954fde6a

Wrap HIP inline functions in anonymous namespaces in vendor.h

AratiGanesh committed 90 days ago

13a4d047

charleshofer committed 90 days ago

f4df93f6

Older