ROCm/jax

Pull Requests Commits

Enable test_remat_checkpoint_dots tests on ROCm (#622)

phambinhfin committed 33 days ago

Verified 17aa7c33

enable nvfp4 tests rocm (#616)

phambinhfin committed 33 days ago

Verified 8360a36d

Enable test_nvfp4_gradient_clip tests on ROCm (#620)

phambinhfin committed 33 days ago

Verified 21f71a9f

Enable test_dot_general tests on ROCm (#619)

phambinhfin committed 33 days ago

Verified 2931e517

Enable test_dot_general_vmap tests on ROCm (#618)

phambinhfin committed 33 days ago

Verified ea794c2c

Enable test_dot_general_sharded tests on ROCm (#617)

phambinhfin committed 33 days ago

Verified e6e31a5f

Enable test_dot_general_sharded for ROCm (#615)

phambinhfin committed 33 days ago

Verified 228310bd

Enable rocm cuda array interface for ROCm (#625)

magaonka-amd committed 33 days ago

Verified 38345cea

Enabled RNN unit test: test_no_workspace_overflow for ROCm devices (#624)

tsrw2048 committed 33 days ago

Verified abbb2ef8

Enabled test for condition number on ROCm devices. (#613)

tsrw2048 committed 33 days ago

Verified 02cf0730

Enable test_comparison_scalar tests for ROCm GPUs (#610)

phambinhfin committed 33 days ago

Verified f8712b76

Fix KeyError for bytes_reservable_limit on ROCm

magaonka-amd committed 36 days ago

bb524b06

Enable testSvdSubsetByIndex for ROCm with subset_by_index skip (#603)

phambinhfin committed 36 days ago

Verified 894ba31f

Enabled ToeplitzSymmetricConstruction unit tests for ROCm devices (#608)

tsrw2048 committed 36 days ago

Verified cf97819a

Enabled testTridiagonal to run on ROCm devices (#607)

tsrw2048 committed 37 days ago

Verified 4f2e28ba

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

tsrw2048 committed 39 days ago

Verified 7134b1f4

Fix Numpy signatures test (#598)

magaonka-amd committed 39 days ago

Verified a3f8af53

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

phambinhfin committed 39 days ago

Verified b799d3a9

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

phambinhfin committed 39 days ago

Verified fb45b3ac

adding abort logic to rocm/jax (#590)

gulsumgudukbay committed 40 days ago

Verified ca39e75c

Fix psend test case failures (#595)

AratiGanesh committed 41 days ago

Verified e4f54fc9

GESVDJ support for ROCm GPUs in JAX (#580)

tsrw2048 committed 41 days ago

Verified 9dbc4119

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

charleshofer committed 69 days ago

Verified ad54f8b1

Add rocm test requirements file (#570)

AratiGanesh committed 69 days ago

Verified 552778a4

add fix for jaxlib build whl (#571)

kiran-thumma committed 73 days ago

Verified a1a217fa

Remove build scripts (#578)

gulsumgudukbay committed 73 days ago

Verified a1968610

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

gulsumgudukbay committed 73 days ago

Verified bf4facd8

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

gulsumgudukbay committed 74 days ago

Verified cafe7943

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

ScXfjiang committed 79 days ago

Verified a2b972f2

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

Arech8 committed 88 days ago

Verified 861cc763

Newer Older