ROCm/jax

Pull Requests Commits

Reduce the XLA thread pool size to 8.

mminutoli committed 45 days ago

049f5893

update conftest

gulsumgudukbay committed 47 days ago

b7a9ed7b

[TESTING ONLY] Strip nightly workflow to ROCm-only tests

mminutoli committed 47 days ago

2facb551

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

magaonka-amd committed 48 days ago

3f19d641

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

magaonka-amd committed 48 days ago

612fb377

Enable lax backend scipy tests on ROCm GPUs (#687)

magaonka-amd committed 48 days ago

b773cc41

Enable lobpcg tests on ROCm platform (#681)

magaonka-amd committed 48 days ago

fa2f01b1

Skip testCudaArrayInterfaceOnNonCudaFails on ROCm platform (#677)

magaonka-amd committed 48 days ago

db7807aa

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

magaonka-amd committed 48 days ago

ec4c2243

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

magaonka-amd committed 48 days ago

08511b63

[Pallas] Fix ROCm GPU architecture detection and route to Triton backend

Ruturaj4 committed 48 days ago

7cc741d5

Update Skip Reason Outputs (#663)

gulsumgudukbay committed 48 days ago

6e7023db

Enable array interoperability tests on ROCm platform (#660)

magaonka-amd committed 48 days ago

ad801639

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

AratiGanesh committed 48 days ago

9b5e320d

Skip test_batch_axis_sharding_jvp because of hipSPARSE issue (#667)

AratiGanesh committed 48 days ago

b46c49f1

Enable testMultivariateNormalSingularCovariance on ROCm (#666)

AratiGanesh committed 48 days ago

ea0550ec

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

magaonka-amd committed 48 days ago

c72f43ed

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

magaonka-amd committed 48 days ago

3ef6a335

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

magaonka-amd committed 48 days ago

223a8c02

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

magaonka-amd committed 48 days ago

8fa39167

Fix test_cuda_array_interface test skip condition (#657)

magaonka-amd committed 48 days ago

15c72e61

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

magaonka-amd committed 48 days ago

7320068f

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

magaonka-amd committed 48 days ago

8cc8132f

Added changes from PR #626 and PR #645. This also fixes merge conflicts from upstream.

tsrw2048 committed 48 days ago

155e11f0

Enabled RNN unit test: test_no_workspace_overflow for ROCm devices (#624)

tsrw2048 committed 48 days ago

7409e8d0

Enabled test for condition number on ROCm devices. (#613)

tsrw2048 committed 48 days ago

0f34f3d1

Unskip supported dtypes for testConvolutionsPreferredElementType (#649)

gulsumgudukbay committed 48 days ago

86022e92

Enable test_variadic_reduce_window on ROCm (#647)

magaonka-amd committed 48 days ago

3c79362f

Enable reduce_window tests on ROCm (#643)

magaonka-amd committed 48 days ago

0c0c4511

Enable RngShardingTests (#644)

gulsumgudukbay committed 48 days ago

93bc00b4

Older