ROCm/jax

Pull Requests Commits

Skip test_batch_axis_sharding_jvp because of hipSPARSE issue

AratiGanesh committed 2 days ago

83a68c89

Unskip test_layout_donation_mismatching_in_and_out_fails for ROCm (#630)

phambinhfin committed 4 days ago

Verified 6ec7453d

Add proper ROCm skip for cuDNN attention tests (#604)

phambinhfin committed 4 days ago

Verified ad0ffd93

Cherry pick fix for unit test: "test_issubdtype" (#629)

tsrw2048 committed 4 days ago

Verified b74ae2cb

Enable optimizer tests on ROCm (#631)

magaonka-amd committed 4 days ago

Verified 23f28df8

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

tsrw2048 committed 4 days ago

Verified 61625381

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

tsrw2048 committed 4 days ago

Verified 3a8af3cb

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

magaonka-amd committed 4 days ago

Verified f6fd826f

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

phambinhfin committed 4 days ago

Verified 32a91b36

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

phambinhfin committed 5 days ago

Verified 7c1c2c20

Enable test_quantize_nvfp4 and test_quantize_requires_global_scale te… (#621)

phambinhfin committed 5 days ago

Verified 161cfcab

Enable test_remat_checkpoint_dots tests on ROCm (#622)

phambinhfin committed 5 days ago

Verified 17aa7c33

enable nvfp4 tests rocm (#616)

phambinhfin committed 5 days ago

Verified 8360a36d

Enable test_nvfp4_gradient_clip tests on ROCm (#620)

phambinhfin committed 5 days ago

Verified 21f71a9f

Enable test_dot_general tests on ROCm (#619)

phambinhfin committed 5 days ago

Verified 2931e517

Enable test_dot_general_vmap tests on ROCm (#618)

phambinhfin committed 5 days ago

Verified ea794c2c

Enable test_dot_general_sharded tests on ROCm (#617)

phambinhfin committed 5 days ago

Verified e6e31a5f

Enable test_dot_general_sharded for ROCm (#615)

phambinhfin committed 5 days ago

Verified 228310bd

Enable rocm cuda array interface for ROCm (#625)

magaonka-amd committed 5 days ago

Verified 38345cea

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

tsrw2048 committed 5 days ago

Verified abbb2ef8

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

tsrw2048 committed 5 days ago

Verified 02cf0730

Enable test_comparison_scalar tests for ROCm GPUs (#610)

phambinhfin committed 5 days ago

Verified f8712b76

Fix KeyError for bytes_reservable_limit on ROCm

magaonka-amd committed 8 days ago

bb524b06

Enable testSvdSubsetByIndex for ROCm with subset_by_index skip (#603)

phambinhfin committed 8 days ago

Verified 894ba31f

Enabled ToeplitzSymmetricConstruction unit tests for ROCm devices (#608)

tsrw2048 committed 8 days ago

Verified cf97819a

Enabled testTridiagonal to run on ROCm devices (#607)

tsrw2048 committed 9 days ago

Verified 4f2e28ba

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

tsrw2048 committed 11 days ago

Verified 7134b1f4

Fix Numpy signatures test (#598)

magaonka-amd committed 11 days ago

Verified a3f8af53

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

phambinhfin committed 11 days ago

Verified b799d3a9

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

phambinhfin committed 11 days ago

Verified fb45b3ac

Older