ROCm/jax

Pull Requests Commits

Bazel workflow on ROCm now correctly processes freethreaded Python versions

tsrw2048 committed 88 days ago

927d3bb4

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

tsrw2048 committed 88 days ago

75a21d7f

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

WBobby committed 91 days ago

3a5fb555

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

gulsumgudukbay committed 91 days ago

6c448c55

Abort detection CI workflow (#688)

gulsumgudukbay committed 91 days ago

3fa5fa33

Skipping testEighTinyNorm due to hipSolver issues (#697)

AratiGanesh committed 91 days ago

54510f25

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

magaonka-amd committed 91 days ago

eea3302f

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

magaonka-amd committed 91 days ago

736275f8

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

magaonka-amd committed 91 days ago

9b8c58a7

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

magaonka-amd committed 91 days ago

c693f9f3

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

magaonka-amd committed 91 days ago

d6fe37cd

Skip testCudaArrayInterfaceOnNonCudaFails on ROCm platform (#677)

magaonka-amd committed 91 days ago

c6748183

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

magaonka-amd committed 91 days ago

566db14b

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

magaonka-amd committed 91 days ago

fc1ae633

Update Skip Reason Outputs (#663)

gulsumgudukbay committed 91 days ago

0559c4e1

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

AratiGanesh committed 91 days ago

4390cfba

Enable testMultivariateNormalSingularCovariance on ROCm (#666)

AratiGanesh committed 91 days ago

f0722943

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

magaonka-amd committed 91 days ago

0b74afb0

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

magaonka-amd committed 91 days ago

605d3eab

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

magaonka-amd committed 91 days ago

5a7dfb3d

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

magaonka-amd committed 91 days ago

6106f30b

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

magaonka-amd committed 91 days ago

4b02bff7

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

magaonka-amd committed 91 days ago

c21d1230

Enable test_variadic_reduce_window on ROCm (#647)

mminutoli committed 91 days ago

fd03607e

Enable RngShardingTests (#644)

gulsumgudukbay committed 91 days ago

7b7f106f

Ruturaj4 committed 91 days ago

17e10b1c

Fix Numpy signatures test (#598)

magaonka-amd committed 91 days ago

d3a731a7

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

phambinhfin committed 91 days ago

6fa95347

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

phambinhfin committed 91 days ago

7628bff8

adding abort logic to rocm/jax (#590)

gulsumgudukbay committed 91 days ago

8303140f

Older