ROCm/jax

Pull Requests Commits

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

magaonka-amd committed 2 days ago

203ede44

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

magaonka-amd committed 2 days ago

6dfbe5c2

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

tsrw2048 committed 2 days ago

1c0f7999

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

tsrw2048 committed 2 days ago

dc825a2a

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

tsrw2048 committed 2 days ago

cf8c8068

Unskip supported dtypes for testConvolutionsPreferredElementType (#649)

gulsumgudukbay committed 2 days ago

570cf20e

Enable test_variadic_reduce_window on ROCm (#647)

magaonka-amd committed 2 days ago

b1d8b7b6

Enable reduce_window tests on ROCm (#643)

magaonka-amd committed 2 days ago

22b0849b

Enable RngShardingTests (#644)

gulsumgudukbay committed 2 days ago

cd5be845

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

magaonka-amd committed 2 days ago

5ac0cc1e

Fix KeyError for bytes_reservable_limit on ROCm

magaonka-amd committed 2 days ago

998c8e57

Enable testSvdSubsetByIndex for ROCm with subset_by_index skip (#603)

phambinhfin committed 2 days ago

9a00eb25

Enabled ToeplitzSymmetricConstruction unit tests for ROCm devices (#608)

tsrw2048 committed 2 days ago

fd550ebf

Enabled testTridiagonal to run on ROCm devices (#607)

tsrw2048 committed 2 days ago

07c413bc

Ruturaj4 committed 2 days ago

69007e14

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

tsrw2048 committed 2 days ago

5c3deab4

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

tsrw2048 committed 2 days ago

41731e6a

Fix Numpy signatures test (#598)

magaonka-amd committed 2 days ago

4064b4e7

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

phambinhfin committed 2 days ago

091008f6

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

phambinhfin committed 2 days ago

3d1f58c6

adding abort logic to rocm/jax (#590)

gulsumgudukbay committed 2 days ago

707d0c03

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

charleshofer committed 2 days ago

cc7a2f3c

Add rocm test requirements file (#570)

AratiGanesh committed 2 days ago

b3d52442

Add skip of test_is_finite() on Cuda (#565)

Arech8 committed 2 days ago

bd386e8c

Address LLVM assertion failure due to a multithreaded use. Update .gitignore (#563)

Arech8 committed 2 days ago

851cfca9

Fix v_pages quantization and adjust test params for ROCm compatibilit… (#560)

phambinhfin committed 2 days ago

e56dc61a

Fix hipSparse CSR algorithm mappings for ROCm 7

phambinhfin committed 2 days ago

abe5dc6b

Add shared utility function get_rocm_version to test_util.py

charleshofer committed 2 days ago

f9c92561

SWDEV-512768 - Replace hipGetLastError with hipExtGetLastError

dsicarov-amd committed 2 days ago

ca9f48e9

Wrap HIP inline functions in anonymous namespaces in vendor.h

AratiGanesh committed 2 days ago

55f51ddd

Newer Older