ROCm/jax

Pull Requests Commits

Fix deprecated use of arrays in place of dtypes

jakevdp committed 22 days ago

68189624

Implement approx_tanh for ROCm using OCML tanh function (#614)

phambinhfin committed 22 days ago

Verified 32a91b36

Skip scaled_matmul tests on ROCm pending upstream JAX merge (#606)

phambinhfin committed 22 days ago

Verified 7c1c2c20

Enable test_quantize_nvfp4 and test_quantize_requires_global_scale te… (#621)

phambinhfin committed 22 days ago

Verified 161cfcab

Enable test_remat_checkpoint_dots tests on ROCm (#622)

phambinhfin committed 22 days ago

Verified 17aa7c33

enable nvfp4 tests rocm (#616)

phambinhfin committed 22 days ago

Verified 8360a36d

Enable test_nvfp4_gradient_clip tests on ROCm (#620)

phambinhfin committed 22 days ago

Verified 21f71a9f

Enable test_dot_general tests on ROCm (#619)

phambinhfin committed 22 days ago

Verified 2931e517

Enable test_dot_general_vmap tests on ROCm (#618)

phambinhfin committed 22 days ago

Verified ea794c2c

Enable test_dot_general_sharded tests on ROCm (#617)

phambinhfin committed 22 days ago

Verified e6e31a5f

Enable test_dot_general_sharded for ROCm (#615)

phambinhfin committed 22 days ago

Verified 228310bd

Enable rocm cuda array interface for ROCm (#625)

magaonka-amd committed 22 days ago

Verified 38345cea

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

tsrw2048 committed 23 days ago

Verified abbb2ef8

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

tsrw2048 committed 23 days ago

Verified 02cf0730

Enable test_comparison_scalar tests for ROCm GPUs (#610)

phambinhfin committed 23 days ago

Verified f8712b76

Fix KeyError for bytes_reservable_limit on ROCm

magaonka-amd committed 26 days ago

bb524b06

Enable testSvdSubsetByIndex for ROCm with subset_by_index skip (#603)

phambinhfin committed 26 days ago

Verified 894ba31f

Enabled ToeplitzSymmetricConstruction unit tests for ROCm devices (#608)

tsrw2048 committed 26 days ago

Verified cf97819a

Enabled testTridiagonal to run on ROCm devices (#607)

tsrw2048 committed 27 days ago

Verified 4f2e28ba

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

tsrw2048 committed 28 days ago

Verified 7134b1f4

Fix Numpy signatures test (#598)

magaonka-amd committed 29 days ago

Verified a3f8af53

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

phambinhfin committed 29 days ago

Verified b799d3a9

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

phambinhfin committed 29 days ago

Verified fb45b3ac

adding abort logic to rocm/jax (#590)

gulsumgudukbay committed 30 days ago

Verified ca39e75c

Fix psend test case failures (#595)

AratiGanesh committed 30 days ago

Verified e4f54fc9

GESVDJ support for ROCm GPUs in JAX (#580)

tsrw2048 committed 31 days ago

Verified 9dbc4119

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

charleshofer committed 59 days ago

Verified ad54f8b1

Add rocm test requirements file (#570)

AratiGanesh committed 59 days ago

Verified 552778a4

add fix for jaxlib build whl (#571)

kiran-thumma committed 63 days ago

Verified a1a217fa

Remove build scripts (#578)

gulsumgudukbay committed 63 days ago

Verified a1968610

Older