ROCm/jax

Pull Requests Commits

update conftest

gulsumgudukbay committed 21 days ago

7b785c15

[TESTING ONLY] Strip nightly workflow to ROCm-only tests

mminutoli committed 21 days ago

f06a3dfa

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

tsrw2048 committed 21 days ago

f51a7a55

Skipping testEighTinyNorm due to hipSolver issues (#697)

AratiGanesh committed 21 days ago

2ff51937

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

magaonka-amd committed 21 days ago

04625c07

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

magaonka-amd committed 21 days ago

918560f9

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

magaonka-amd committed 21 days ago

4993f1d5

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

magaonka-amd committed 21 days ago

2584bfa6

Add ROCm encoding for test_struct_encoding_determinism (#683)

AratiGanesh committed 21 days ago

5c32a623

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

magaonka-amd committed 21 days ago

bc2fe040

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

magaonka-amd committed 21 days ago

f8ad1416

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

magaonka-amd committed 21 days ago

ed160fc6

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

magaonka-amd committed 21 days ago

d7d4bff7

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

magaonka-amd committed 21 days ago

3753c427

Enable lobpcg tests on ROCm platform (#681)

magaonka-amd committed 21 days ago

c162bd0e

Skip testCudaArrayInterfaceOnNonCudaFails on ROCm platform (#677)

magaonka-amd committed 21 days ago

ff14de3d

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

magaonka-amd committed 21 days ago

87066861

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

magaonka-amd committed 21 days ago

26b19d66

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

Ruturaj4 committed 21 days ago

324c86ee

Update Skip Reason Outputs (#663)

gulsumgudukbay committed 21 days ago

981b167a

Enable array interoperability tests on ROCm platform (#660)

magaonka-amd committed 21 days ago

8c24c67f

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

AratiGanesh committed 21 days ago

8cb89151

Skip test_batch_axis_sharding_jvp because of hipSPARSE issue (#667)

AratiGanesh committed 21 days ago

4330a5f1

Enable testMultivariateNormalSingularCovariance on ROCm (#666)

AratiGanesh committed 21 days ago

1db0a486

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

magaonka-amd committed 21 days ago

9f5d729d

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

magaonka-amd committed 21 days ago

af3f0a57

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

magaonka-amd committed 21 days ago

79714e0a

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

magaonka-amd committed 21 days ago

4966b99c

Fix test_cuda_array_interface test skip condition (#657)

magaonka-amd committed 21 days ago

b042066c

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

magaonka-amd committed 21 days ago

203ede44

Older