ROCm/jax

Pull Requests Commits

Abort detection CI workflow (#688)

gulsumgudukbay committed 1 day ago

f7a64071

Skipping testEighTinyNorm due to hipSolver issues (#697)

AratiGanesh committed 1 day ago

90a3578d

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

magaonka-amd committed 1 day ago

cd072cd8

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

magaonka-amd committed 1 day ago

c6326788

Add ROCm encoding for test_struct_encoding_determinism (#683)

AratiGanesh committed 1 day ago

30f61f4c

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

magaonka-amd committed 1 day ago

abfda2d5

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

magaonka-amd committed 1 day ago

991bc2fb

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

magaonka-amd committed 1 day ago

f3efc4d3

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

magaonka-amd committed 1 day ago

73824693

Skip testCudaArrayInterfaceOnNonCudaFails on ROCm platform (#677)

magaonka-amd committed 1 day ago

9df214c0

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

magaonka-amd committed 1 day ago

3f26dfeb

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

magaonka-amd committed 1 day ago

87d71117

Update Skip Reason Outputs (#663)

gulsumgudukbay committed 1 day ago

55c86bc9

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

AratiGanesh committed 1 day ago

25bc14dd

Enable testMultivariateNormalSingularCovariance on ROCm (#666)

AratiGanesh committed 1 day ago

2e76caea

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

magaonka-amd committed 1 day ago

c7ccce22

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

magaonka-amd committed 1 day ago

b798b3b3

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

magaonka-amd committed 1 day ago

2045fd0d

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

magaonka-amd committed 1 day ago

3f3518fc

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

magaonka-amd committed 1 day ago

2988d2ad

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

magaonka-amd committed 1 day ago

69381b1c

Enable test_variadic_reduce_window on ROCm (#647)

mminutoli committed 1 day ago

472b2279

Enable RngShardingTests (#644)

gulsumgudukbay committed 1 day ago

ce0783a5

Ruturaj4 committed 1 day ago

e5fddf8a

Fix Numpy signatures test (#598)

magaonka-amd committed 1 day ago

c979baff

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

phambinhfin committed 1 day ago

cbfb8422

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

phambinhfin committed 1 day ago

becc59ee

adding abort logic to rocm/jax (#590)

gulsumgudukbay committed 1 day ago

5a2d899d

Let the unit tests use build.py for setting up Bazel commands for unit tests (#582)

charleshofer committed 1 day ago

11a084ba

Add rocm test requirements file (#570)

AratiGanesh committed 1 day ago

3a63ef39

Older