ROCm/jax

Pull Requests Commits

Fix indentation for adding pytest marker

gulsumgudukbay committed 87 days ago

Verified 95a4d4ae

Implement Mosaic GPU detection and Auto-Skips

gulsumgudukbay committed 87 days ago

Verified 16044530

Abort detection CI workflow (#688)

gulsumgudukbay committed 88 days ago

f7a64071

Skipping testEighTinyNorm due to hipSolver issues (#697)

AratiGanesh committed 88 days ago

90a3578d

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

magaonka-amd committed 88 days ago

cd072cd8

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

magaonka-amd committed 88 days ago

c6326788

Add ROCm encoding for test_struct_encoding_determinism (#683)

AratiGanesh committed 88 days ago

30f61f4c

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

magaonka-amd committed 88 days ago

abfda2d5

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

magaonka-amd committed 88 days ago

991bc2fb

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

magaonka-amd committed 88 days ago

f3efc4d3

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

magaonka-amd committed 88 days ago

73824693

Skip testCudaArrayInterfaceOnNonCudaFails on ROCm platform (#677)

magaonka-amd committed 88 days ago

9df214c0

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

magaonka-amd committed 88 days ago

3f26dfeb

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

magaonka-amd committed 88 days ago

87d71117

Update Skip Reason Outputs (#663)

gulsumgudukbay committed 88 days ago

55c86bc9

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

AratiGanesh committed 88 days ago

25bc14dd

Enable testMultivariateNormalSingularCovariance on ROCm (#666)

AratiGanesh committed 88 days ago

2e76caea

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

magaonka-amd committed 88 days ago

c7ccce22

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

magaonka-amd committed 88 days ago

b798b3b3

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

magaonka-amd committed 88 days ago

2045fd0d

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

magaonka-amd committed 88 days ago

3f3518fc

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

magaonka-amd committed 88 days ago

2988d2ad

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

magaonka-amd committed 88 days ago

69381b1c

Enable test_variadic_reduce_window on ROCm (#647)

mminutoli committed 88 days ago

472b2279

Enable RngShardingTests (#644)

gulsumgudukbay committed 88 days ago

ce0783a5

Ruturaj4 committed 88 days ago

e5fddf8a

Fix Numpy signatures test (#598)

magaonka-amd committed 88 days ago

c979baff

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

phambinhfin committed 88 days ago

cbfb8422

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

phambinhfin committed 88 days ago

becc59ee

adding abort logic to rocm/jax (#590)

gulsumgudukbay committed 88 days ago

5a2d899d

Older