ROCm/jax

Pull Requests Commits

Add ROCm gesdd SVD backend, export back-compat data, and relaxed GPU test tolerances

cj401-amd committed 16 days ago

78abe174

Phambinh/skip fused attention navi lds limit (#720)

phambinhfin committed 18 days ago

Verified 164fc708

Remove 'var': ['mean'] from unsupported_params in signature test (#719)

phambinhfin committed 24 days ago

Verified dc371953

Add ROCm platform to export memory_space tests (#715)

phambinhfin committed 26 days ago

Verified 342510fc

Backporting `device test` unit tests for ROCm from JAX v0.9.0 to v0.8.2. (#706)

tsrw2048 committed 34 days ago

Verified 2890b29c

Skip test_tridiagonal_solve_grad test 0.8.2 (#704)

AratiGanesh committed 34 days ago

Verified 8bf0d1f4

Skip test_batch_axis_sharding_jvp13 test on 0.8.2 (#707)

AratiGanesh committed 34 days ago

Verified 3b39f574

Skip testEighTinyNorm test 0.8.2 (#696)

AratiGanesh committed 40 days ago

Verified fd60968b

Backport memory space export test from JAX v0.9.0 to v0.8.2. (#700)

tsrw2048 committed 41 days ago

Verified 76568193

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

magaonka-amd committed 44 days ago

Verified 98ab4986

Add ROCm encoding for test_struct_encoding_determinism (#684)

AratiGanesh committed 45 days ago

Verified 21d5e0bf

Enable lax backend scipy tests on ROCm GPUs (#686)

magaonka-amd committed 46 days ago

Verified 4b67574a

Enable lobpcg tests on ROCm platform (#680)

magaonka-amd committed 47 days ago

Verified a1fe9281

Skip testCudaArrayInterfaceOnNonCudaFails on ROCm platform (#676)

magaonka-amd committed 52 days ago

Verified 6821ce3e

Update Skip Reason Outputs (#663)

gulsumgudukbay committed 53 days ago

Verified 5d8461b5

Enable array interoperability tests on ROCm platform (#660)

magaonka-amd committed 53 days ago

Verified 386f42b0

Skip test_tridiagonal_solve on ROCm due to hipSPARSE numerical errors (#668)

AratiGanesh committed 53 days ago

Verified ad44c78d

Skip test_batch_axis_sharding_jvp because of hipSPARSE issue (#667)

AratiGanesh committed 53 days ago

Verified 6166b23e

Enable testMultivariateNormalSingularCovariance on ROCm (#666)

AratiGanesh committed 53 days ago

Verified 5aa3a3ba

Fix test_cuda_array_interface test skip condition (#657)

magaonka-amd committed 55 days ago

Verified 792e978f

Port skip for "test_prim_tridiagonal_solve" tests from JAX v0.8.0 to v0.8.2 (#658)

tsrw2048 committed 55 days ago

Verified 9ef651c5

Update sparse test skip messages in v0.8.2 (#653)

magaonka-amd committed 58 days ago

Verified 775d1886

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

magaonka-amd committed 58 days ago

Verified b59b0b27

Added changes from PR #626 and PR #645. This also fixes merge conflicts from upstream.

tsrw2048 committed 59 days ago

694e630d

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

tsrw2048 committed 59 days ago

5b3e12a7

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

tsrw2048 committed 59 days ago

b3af7bfa

Unskip supported dtypes for testConvolutionsPreferredElementType (#649)

gulsumgudukbay committed 59 days ago

Verified fe0f30ac

Enable test_variadic_reduce_window on ROCm (#647)

magaonka-amd committed 59 days ago

Verified 1cc595cd

Enable reduce_window tests on ROCm (#643)

magaonka-amd committed 59 days ago

Verified fbfa695a

Enable RngShardingTests (#644)

gulsumgudukbay committed 59 days ago

Verified 9dc13113

Older