ROCm/jax

Pull Requests Commits

kiran-thumma committed 32 days ago

Verified ec77adb5

Fix HIP memory leaks in RNN kernels (#726)

magaonka-amd committed 32 days ago

Verified 48d2ef14

Update skip message version from 0.8.0 to 0.9.0 for test_is_finite on… (#702)

phambinhfin committed 53 days ago

Verified 32ba3e95

Skip test_batch_axis_sharding_jvp13 test 0.9.0 (#709)

AratiGanesh committed 55 days ago

Verified 8cb19cfc

Skip test_tridiagonal_solve_grad test 0.9.0 (#703)

AratiGanesh committed 55 days ago

Verified 57c00805

Add `device test` unit tests for ROCm (JAX v0.9.0) (#705)

tsrw2048 committed 55 days ago

Verified 0988e6da

Modified memory space export test to run on ROCm (for some tests). (#698)

tsrw2048 committed 56 days ago

Verified ac4ba1cb

Skipping testEighTinyNorm due to hipSolver issues (#697)

AratiGanesh committed 56 days ago

Verified 6df5f178

Enable test deviceless aot compile test on ROCm (#694)

magaonka-amd committed 59 days ago

Verified 76c5165a

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

magaonka-amd committed 59 days ago

Verified 7071f715

Enable memory space export tests on ROCm GPUs (#690)

magaonka-amd committed 59 days ago

Verified 2c7ef615

Remove 'mean' from unsupported params for jnp.var (#689)

magaonka-amd committed 59 days ago

Verified 4803ba8d

Add ROCm encoding for test_struct_encoding_determinism (#683)

AratiGanesh committed 60 days ago

Verified 6701f5d1

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

magaonka-amd committed 61 days ago

Verified 498f7350

Enable lobpcg tests on ROCm platform (#681)

magaonka-amd committed 62 days ago

Verified d3cce970

Skip testCudaArrayInterfaceOnNonCudaFails on ROCm platform (#677)

magaonka-amd committed 67 days ago

Verified 103da597

Update Skip Reason Outputs (#663)

gulsumgudukbay committed 67 days ago

3747d88a

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

AratiGanesh committed 67 days ago

93f5fdc1

Skip test_batch_axis_sharding_jvp because of hipSPARSE issue (#667)

AratiGanesh committed 67 days ago

fc10735b

Enable testMultivariateNormalSingularCovariance on ROCm (#666)

AratiGanesh committed 67 days ago

ed14fc6b

Fix test_cuda_array_interface test skip condition (#657)

magaonka-amd committed 67 days ago

34676201

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

tsrw2048 committed 67 days ago

a988aebf

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

magaonka-amd committed 67 days ago

e0739e6c

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

magaonka-amd committed 67 days ago

95e1df00

Enable array interoperability tests on ROCm platform (#660)

magaonka-amd committed 67 days ago

91be368b

[Pallas] Fix ROCm GPU architecture detection and route to Triton backend

Ruturaj4 committed 68 days ago

b77014fd

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

tsrw2048 committed 73 days ago

98cc66ca

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

tsrw2048 committed 73 days ago

708d732f

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

tsrw2048 committed 73 days ago

10c7e75a

Unskip supported dtypes for testConvolutionsPreferredElementType (#649)

gulsumgudukbay committed 73 days ago

ee5581c8

Older