ROCm/jax

Pull Requests Commits

remove JAX_ENABLE_CUDA_XDIST

gulsumgudukbay committed 11 days ago

126ae8d0

ci: ROCm pytest abort support

gulsumgudukbay committed 11 days ago

fdc64175

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

tsrw2048 committed 18 days ago

22c253d8

Skipping testEighTinyNorm due to hipSolver issues (#697)

AratiGanesh committed 18 days ago

e682e0fe

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

magaonka-amd committed 18 days ago

3e9571ce

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

magaonka-amd committed 18 days ago

28d0d7f9

Add ROCm encoding for test_struct_encoding_determinism (#683)

AratiGanesh committed 18 days ago

9aa93d7c

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

magaonka-amd committed 18 days ago

1515692c

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

magaonka-amd committed 18 days ago

93967c61

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

magaonka-amd committed 18 days ago

fe141920

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

magaonka-amd committed 18 days ago

c30ce8db

Skip testCudaArrayInterfaceOnNonCudaFails on ROCm platform (#677)

magaonka-amd committed 18 days ago

5c9e35e4

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

magaonka-amd committed 18 days ago

f5486cd3

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

magaonka-amd committed 18 days ago

3b0050df

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

Ruturaj4 committed 18 days ago

b42f5e59

Update Skip Reason Outputs (#663)

gulsumgudukbay committed 18 days ago

1e70333a

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

AratiGanesh committed 18 days ago

35739353

Skip test_batch_axis_sharding_jvp because of hipSPARSE issue (#667)

AratiGanesh committed 18 days ago

fae6b30c

Enable testMultivariateNormalSingularCovariance on ROCm (#666)

AratiGanesh committed 18 days ago

88dad299

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

magaonka-amd committed 18 days ago

83476aa0

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

magaonka-amd committed 18 days ago

66c4688f

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

magaonka-amd committed 18 days ago

65ba0f1c

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

magaonka-amd committed 18 days ago

c3c93c89

Fix test_cuda_array_interface test skip condition (#657)

magaonka-amd committed 18 days ago

a002dc0a

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

magaonka-amd committed 18 days ago

6d85cf78

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

magaonka-amd committed 18 days ago

80239d73

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

tsrw2048 committed 18 days ago

d353733f

Enable test_variadic_reduce_window on ROCm (#647)

mminutoli committed 18 days ago

1c066db3

Enable RngShardingTests (#644)

gulsumgudukbay committed 18 days ago

49cac3ec

Backport cuda_array_interface , testDotAlgorithm, Optimizer test fixes from rocm-jaxlib-v0.8.0 (#635)

magaonka-amd committed 18 days ago

b200d457

Older