ROCm/jax

Pull Requests Commits

Enabled bicgstab and gmres tests with identity and random systems for ROCm devices

tsrw2048 committed 86 days ago

f9c0dba9

Enable optimizer tests on ROCm (#631)

magaonka-amd committed 86 days ago

Verified 23f28df8

Enabled triton_pallas_test.py unit test file to run (#626)

tsrw2048 committed 86 days ago

Verified 61625381

Cherry pick fix for unit test: "donate_argnames_signature_fail" (#627)

tsrw2048 committed 86 days ago

Verified 3a8af3cb

Enable testDotAlgorithm BF16/TF32 tests on ROCm (#628)

magaonka-amd committed 86 days ago

Verified f6fd826f

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

phambinhfin committed 87 days ago

Verified 32a91b36

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

phambinhfin committed 87 days ago

Verified 7c1c2c20

Enable test_quantize_nvfp4 and test_quantize_requires_global_scale te… (#621)

phambinhfin committed 87 days ago

Verified 161cfcab

Enable test_remat_checkpoint_dots tests on ROCm (#622)

phambinhfin committed 87 days ago

Verified 17aa7c33

enable nvfp4 tests rocm (#616)

phambinhfin committed 87 days ago

Verified 8360a36d

Enable test_nvfp4_gradient_clip tests on ROCm (#620)

phambinhfin committed 87 days ago

Verified 21f71a9f

Enable test_dot_general tests on ROCm (#619)

phambinhfin committed 87 days ago

Verified 2931e517

Enable test_dot_general_vmap tests on ROCm (#618)

phambinhfin committed 87 days ago

Verified ea794c2c

Enable test_dot_general_sharded tests on ROCm (#617)

phambinhfin committed 87 days ago

Verified e6e31a5f

Enable test_dot_general_sharded for ROCm (#615)

phambinhfin committed 87 days ago

Verified 228310bd

Enable rocm cuda array interface for ROCm (#625)

magaonka-amd committed 87 days ago

Verified 38345cea

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

tsrw2048 committed 87 days ago

Verified abbb2ef8

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

tsrw2048 committed 88 days ago

Verified 02cf0730

Enable test_comparison_scalar tests for ROCm GPUs (#610)

phambinhfin committed 88 days ago

Verified f8712b76

Fix KeyError for bytes_reservable_limit on ROCm

magaonka-amd committed 90 days ago

bb524b06

Enable testSvdSubsetByIndex for ROCm with subset_by_index skip (#603)

phambinhfin committed 91 days ago

Verified 894ba31f

Enabled ToeplitzSymmetricConstruction unit tests for ROCm devices (#608)

tsrw2048 committed 91 days ago

Verified cf97819a

Enabled testTridiagonal to run on ROCm devices (#607)

tsrw2048 committed 91 days ago

Verified 4f2e28ba

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

tsrw2048 committed 93 days ago

Verified 7134b1f4

Fix Numpy signatures test (#598)

magaonka-amd committed 93 days ago

Verified a3f8af53

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

phambinhfin committed 93 days ago

Verified b799d3a9

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

phambinhfin committed 93 days ago

Verified fb45b3ac

adding abort logic to rocm/jax (#590)

gulsumgudukbay committed 94 days ago

Verified ca39e75c

Fix psend test case failures (#595)

AratiGanesh committed 95 days ago

Verified e4f54fc9

GESVDJ support for ROCm GPUs in JAX (#580)

tsrw2048 committed 96 days ago

Verified 9dbc4119

Older