ROCm/jax

Pull Requests Commits

Enable Pallas tests on ROCm

AratiGanesh committed 8 days ago

1521ffff

Fix KeyError for bytes_reservable_limit on ROCm

magaonka-amd committed 10 days ago

bb524b06

Enable testSvdSubsetByIndex for ROCm with subset_by_index skip (#603)

phambinhfin committed 11 days ago

Verified 894ba31f

Enabled ToeplitzSymmetricConstruction unit tests for ROCm devices (#608)

tsrw2048 committed 11 days ago

Verified cf97819a

Enabled testTridiagonal to run on ROCm devices (#607)

tsrw2048 committed 11 days ago

Verified 4f2e28ba

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

tsrw2048 committed 13 days ago

Verified 7134b1f4

Fix Numpy signatures test (#598)

magaonka-amd committed 13 days ago

Verified a3f8af53

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

phambinhfin committed 13 days ago

Verified b799d3a9

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

phambinhfin committed 13 days ago

Verified fb45b3ac

adding abort logic to rocm/jax (#590)

gulsumgudukbay committed 14 days ago

Verified ca39e75c

Fix psend test case failures (#595)

AratiGanesh committed 15 days ago

Verified e4f54fc9

GESVDJ support for ROCm GPUs in JAX (#580)

tsrw2048 committed 15 days ago

Verified 9dbc4119

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

charleshofer committed 43 days ago

Verified ad54f8b1

Add rocm test requirements file (#570)

AratiGanesh committed 43 days ago

Verified 552778a4

add fix for jaxlib build whl (#571)

kiran-thumma committed 47 days ago

Verified a1a217fa

Remove build scripts (#578)

gulsumgudukbay committed 47 days ago

Verified a1968610

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

gulsumgudukbay committed 47 days ago

Verified bf4facd8

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

gulsumgudukbay committed 48 days ago

Verified cafe7943

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

ScXfjiang committed 54 days ago

Verified a2b972f2

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

Arech8 committed 62 days ago

Verified 861cc763

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

Arech8 committed 62 days ago

Verified c972c55d

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

Arech8 committed 62 days ago

Verified 339bf1e8

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

phambinhfin committed 69 days ago

Verified f893175a

Fix ROCm lib names copy pattern to use wildcard version

phambinhfin committed 70 days ago

709963b8

Make nvidia version data optional for ROCm builds

phambinhfin committed 70 days ago

b5762449

Fix hipSparse CSR algorithm mappings for ROCm 7

phambinhfin committed 70 days ago

28e04303

Add shared utility function get_rocm_version to test_util.py

charleshofer committed 70 days ago

aac9e9d2

SWDEV-512768 - Replace hipGetLastError with hipExtGetLastError

dsicarov-amd committed 70 days ago

954fde6a

Wrap HIP inline functions in anonymous namespaces in vendor.h

AratiGanesh committed 70 days ago

13a4d047

charleshofer committed 70 days ago

f4df93f6

Older