ROCm/jax

Pull Requests Commits

update wheel<0.46

kiran-thumma committed 71 days ago

Verified 39f73e9a

Skip testCudaArrayInterfaceOnNonCudaFails on ROCm platform (#675)

magaonka-amd committed 71 days ago

Verified fa227ed6

Update skip_on_devices and mosaic skip order in gpu_pallas_distributed_Test.py (#669)

gulsumgudukbay committed 73 days ago

Verified 6b06c904

Enable polynomial roots tests on ROCm platform on v0.8.0 branch (#665)

magaonka-amd committed 73 days ago

Verified 56321132

Skip test_batch_axis_sharding_jvp on ROCm (#654)

AratiGanesh committed 73 days ago

Verified d71b0868

Enable array interoperability tests on ROCm platform (#659)

magaonka-amd committed 73 days ago

Verified 4a400314

Update Skip Reason Output for run_on_devices decorator (#662)

gulsumgudukbay committed 73 days ago

Verified 7eabf75b

Fix test_cuda_array_interface test skip condition (#656)

magaonka-amd committed 75 days ago

Verified eacce54d

Added skip for "test_prim_tridiagonal_solve" tests on ROCm (#655)

tsrw2048 committed 75 days ago

Verified 011e44d9

Skip sparse tests on ROCm due to hipSPARSE issue (#651)

magaonka-amd committed 78 days ago

Verified caaf2a2b

Unskip supported dtypes for testConvolutionsPreferredElementType (#648)

gulsumgudukbay committed 79 days ago

Verified 114df047

Skipping elementwise_inline_asm unit test on ROCm. (#645)

tsrw2048 committed 79 days ago

Verified 9344b2b2

Enable test_variadic_reduce_window on ROCm (#646)

magaonka-amd committed 79 days ago

Verified 5be5318d

Enable reduce_window tests on ROCm (#641)

magaonka-amd committed 79 days ago

Verified 835e49d3

Enable RngShardingTests (#642)

gulsumgudukbay committed 79 days ago

Verified 68725119

Skip tridiagonal solve tests on ROCm (#634)

AratiGanesh committed 79 days ago

Verified c2783442

Enable nn tests on ROCm (#640)

phambinhfin committed 79 days ago

Verified 9baf4c70

Skip test_get_vmap for slice-only case with mismatched batch dims (#636)

phambinhfin committed 79 days ago

Verified f8dc4b1f

Enable testMultivariateNormalSingularCovariance on ROCm (#639)

AratiGanesh committed 79 days ago

Verified 0a40718e

Enable testDotProductAttentionBiasGradient on ROCm (#637)

phambinhfin committed 79 days ago

Verified fa0512a2

Unskip test_layout_donation_mismatching_in_and_out_fails for ROCm (#630)

phambinhfin committed 80 days ago

Verified 6ec7453d

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

phambinhfin committed 80 days ago

Verified ad0ffd93

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

tsrw2048 committed 80 days ago

Verified b74ae2cb

Enable optimizer tests on ROCm (#631)

magaonka-amd committed 80 days ago

Verified 23f28df8

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

tsrw2048 committed 80 days ago

Verified 61625381

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

tsrw2048 committed 80 days ago

Verified 3a8af3cb

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

magaonka-amd committed 80 days ago

Verified f6fd826f

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

phambinhfin committed 80 days ago

Verified 32a91b36

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

phambinhfin committed 81 days ago

Verified 7c1c2c20

Enable test_quantize_nvfp4 and test_quantize_requires_global_scale te… (#621)

phambinhfin committed 81 days ago

Verified 161cfcab

Older