ROCm/jax

Pull Requests Commits

Bazel workflow on ROCm now correctly processes freethreaded Python versions

tsrw2048 committed 3 days ago

927d3bb4

Added fix to Bazel ROCm CI script which will download JAX wheels from GCS.

tsrw2048 committed 3 days ago

75a21d7f

fix: add rocm_sysdeps/lib to wheel RUNPATH (#737)

WBobby committed 7 days ago

3a5fb555

Abort-Detection: Fix halt-for-connection input (#712)

gulsumgudukbay committed 7 days ago

6c448c55

Abort detection CI workflow (#688)

gulsumgudukbay committed 7 days ago

3fa5fa33

Skipping testEighTinyNorm due to hipSolver issues (#697)

AratiGanesh committed 7 days ago

54510f25

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

magaonka-amd committed 7 days ago

eea3302f

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

magaonka-amd committed 7 days ago

736275f8

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

magaonka-amd committed 7 days ago

9b8c58a7

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

magaonka-amd committed 7 days ago

c693f9f3

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

magaonka-amd committed 7 days ago

d6fe37cd

Skip testCudaArrayInterfaceOnNonCudaFails on ROCm platform (#677)

magaonka-amd committed 7 days ago

c6748183

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

magaonka-amd committed 7 days ago

566db14b

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

magaonka-amd committed 7 days ago

fc1ae633

Update Skip Reason Outputs (#663)

gulsumgudukbay committed 7 days ago

0559c4e1

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

AratiGanesh committed 7 days ago

4390cfba

Enable testMultivariateNormalSingularCovariance on ROCm (#666)

AratiGanesh committed 7 days ago

f0722943

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

magaonka-amd committed 7 days ago

0b74afb0

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

magaonka-amd committed 7 days ago

605d3eab

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

magaonka-amd committed 7 days ago

5a7dfb3d

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

magaonka-amd committed 7 days ago

6106f30b

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

magaonka-amd committed 7 days ago

4b02bff7

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

magaonka-amd committed 7 days ago

c21d1230

Enable test_variadic_reduce_window on ROCm (#647)

mminutoli committed 7 days ago

fd03607e

Enable RngShardingTests (#644)

gulsumgudukbay committed 7 days ago

7b7f106f

Ruturaj4 committed 7 days ago

17e10b1c

Fix Numpy signatures test (#598)

magaonka-amd committed 7 days ago

d3a731a7

Fix shared memory limit check for ROCm in test_dot (#596)

phambinhfin committed 7 days ago

6fa95347

Skip is_finite tests on ROCm (not in Triton lowering for jax 0.8.0) (#597)

phambinhfin committed 7 days ago

7628bff8

adding abort logic to rocm/jax (#590)

gulsumgudukbay committed 7 days ago

8303140f

Older