ROCm/jax

Pull Requests Commits

Add ROCm encoding for test_struct_encoding_determinism

AratiGanesh committed 95 days ago

5ad62b7a

Skip testCudaArrayInterfaceOnNonCudaFails on ROCm platform (#676)

magaonka-amd committed 99 days ago

Verified 6821ce3e

Update Skip Reason Outputs (#663)

gulsumgudukbay committed 100 days ago

Verified 5d8461b5

Enable array interoperability tests on ROCm platform (#660)

magaonka-amd committed 100 days ago

Verified 386f42b0

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

AratiGanesh committed 100 days ago

Verified ad44c78d

Skip test_batch_axis_sharding_jvp because of hipSPARSE issue (#667)

AratiGanesh committed 100 days ago

Verified 6166b23e

Enable testMultivariateNormalSingularCovariance on ROCm (#666)

AratiGanesh committed 100 days ago

Verified 5aa3a3ba

Fix test_cuda_array_interface test skip condition (#657)

magaonka-amd committed 102 days ago

Verified 792e978f

Port skip for "test_prim_tridiagonal_solve" tests from JAX v0.8.0 to v0.8.2 (#658)

tsrw2048 committed 102 days ago

Verified 9ef651c5

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

magaonka-amd committed 105 days ago

Verified 775d1886

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

magaonka-amd committed 105 days ago

Verified b59b0b27

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

tsrw2048 committed 106 days ago

694e630d

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

tsrw2048 committed 106 days ago

5b3e12a7

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

tsrw2048 committed 106 days ago

b3af7bfa

Unskip supported dtypes for testConvolutionsPreferredElementType (#649)

gulsumgudukbay committed 106 days ago

Verified fe0f30ac

Enable test_variadic_reduce_window on ROCm (#647)

magaonka-amd committed 106 days ago

Verified 1cc595cd

Enable reduce_window tests on ROCm (#643)

magaonka-amd committed 106 days ago

Verified fbfa695a

Enable RngShardingTests (#644)

gulsumgudukbay committed 106 days ago

Verified 9dc13113

Backport cuda_array_interface , testDotAlgorithm, Optimizer test fixes from rocm-jaxlib-v0.8.0 (#635)

magaonka-amd committed 106 days ago

Verified 84c7adfd

Fix KeyError for bytes_reservable_limit on ROCm

magaonka-amd committed 110 days ago

cef02f3a

Enable testSvdSubsetByIndex for ROCm with subset_by_index skip (#603)

phambinhfin committed 110 days ago

e3ffcb08

Enabled ToeplitzSymmetricConstruction unit tests for ROCm devices (#608)

tsrw2048 committed 110 days ago

b9b014dc

Enabled testTridiagonal to run on ROCm devices (#607)

tsrw2048 committed 110 days ago

9e3371ea

Ruturaj4 committed 110 days ago

fe81a738

Fixed merge conflicts when moving GESVDJ commit from JAX 0.8.0 to 0.8.2 (#602)

tsrw2048 committed 112 days ago

Verified 9b333623

Fix GPU lowering rule for SVD on ROCm devices (#600)

tsrw2048 committed 113 days ago

ed8596b0

Fix Numpy signatures test (#598)

magaonka-amd committed 113 days ago

651aded6

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

phambinhfin committed 113 days ago

d9324638

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

phambinhfin committed 113 days ago

f4bf6876

adding abort logic to rocm/jax (#590)

gulsumgudukbay committed 113 days ago

abbad31d

Older