ROCm/jax

Pull Requests Commits

Fix plugin init script

alekstheod committed 91 days ago

cf8bd43a

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

gulsumgudukbay committed 95 days ago

7d684aa5

Abort detection CI workflow (#688)

gulsumgudukbay committed 95 days ago

6957e42d

Skipping testEighTinyNorm due to hipSolver issues (#697)

AratiGanesh committed 95 days ago

f2f9dd18

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

magaonka-amd committed 95 days ago

02ae6c5b

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

magaonka-amd committed 95 days ago

717cd20f

Add ROCm encoding for test_struct_encoding_determinism (#683)

AratiGanesh committed 95 days ago

d18c7571

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

magaonka-amd committed 95 days ago

ec394330

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

magaonka-amd committed 95 days ago

14d1b171

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

magaonka-amd committed 95 days ago

c45c49a9

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

magaonka-amd committed 95 days ago

9983dd00

Skip testCudaArrayInterfaceOnNonCudaFails on ROCm platform (#677)

magaonka-amd committed 95 days ago

a186d625

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

magaonka-amd committed 95 days ago

e0d2d1e9

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

magaonka-amd committed 95 days ago

7648e11b

Update Skip Reason Outputs (#663)

gulsumgudukbay committed 95 days ago

3c295779

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

AratiGanesh committed 95 days ago

84ad452e

Enable testMultivariateNormalSingularCovariance on ROCm (#666)

AratiGanesh committed 95 days ago

e5714928

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

magaonka-amd committed 95 days ago

47321964

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

magaonka-amd committed 95 days ago

0cc3b988

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

magaonka-amd committed 95 days ago

f0e90395

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

magaonka-amd committed 95 days ago

1b6c9e72

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

magaonka-amd committed 95 days ago

cc543662

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

magaonka-amd committed 95 days ago

8ecdd8b7

Enable test_variadic_reduce_window on ROCm (#647)

mminutoli committed 95 days ago

22ac32aa

Enable RngShardingTests (#644)

gulsumgudukbay committed 95 days ago

4c60b2b0

Ruturaj4 committed 95 days ago

b202e1f4

Fix Numpy signatures test (#598)

magaonka-amd committed 95 days ago

92211a5d

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

phambinhfin committed 95 days ago

14b0125c

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

phambinhfin committed 95 days ago

6eb4fdf5

adding abort logic to rocm/jax (#590)

gulsumgudukbay committed 95 days ago

640e4a21

Older