ROCm/jax

Pull Requests Commits

intro Mosaic to ROCm, bulk commit

Arech8 committed 23 days ago

a075fe15

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

tsrw2048 committed 29 days ago

98cc66ca

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

tsrw2048 committed 29 days ago

708d732f

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

tsrw2048 committed 29 days ago

10c7e75a

Unskip supported dtypes for testConvolutionsPreferredElementType (#649)

gulsumgudukbay committed 29 days ago

ee5581c8

Enable test_variadic_reduce_window on ROCm (#647)

magaonka-amd committed 29 days ago

134825e3

Enable reduce_window tests on ROCm (#643)

magaonka-amd committed 29 days ago

2557ff08

Enable RngShardingTests (#644)

gulsumgudukbay committed 29 days ago

4a075b9a

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

magaonka-amd committed 29 days ago

ca932346

Fix KeyError for bytes_reservable_limit on ROCm

magaonka-amd committed 29 days ago

0956a239

Enable testSvdSubsetByIndex for ROCm with subset_by_index skip (#603)

phambinhfin committed 29 days ago

fc3bc807

Enabled ToeplitzSymmetricConstruction unit tests for ROCm devices (#608)

tsrw2048 committed 29 days ago

455f6b10

Enabled testTridiagonal to run on ROCm devices (#607)

tsrw2048 committed 29 days ago

ee2c254e

Ruturaj4 committed 29 days ago

e8ac89ba

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

tsrw2048 committed 29 days ago

aaac58cb

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

tsrw2048 committed 29 days ago

fd0e42c9

Fix Numpy signatures test (#598)

magaonka-amd committed 29 days ago

1a823af2

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

phambinhfin committed 29 days ago

e5b66138

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

phambinhfin committed 29 days ago

17c7560c

adding abort logic to rocm/jax (#590)

gulsumgudukbay committed 29 days ago

a83cc9e9

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

charleshofer committed 29 days ago

6bd1a15e

Add rocm test requirements file (#570)

AratiGanesh committed 29 days ago

835856e0

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

Arech8 committed 29 days ago

6ab35026

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

Arech8 committed 29 days ago

8b0b3d10

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

phambinhfin committed 29 days ago

61d28322

Make nvidia version data optional for ROCm builds

phambinhfin committed 29 days ago

d88ca192

Fix hipSparse CSR algorithm mappings for ROCm 7

phambinhfin committed 29 days ago

81cdeb49

Add shared utility function get_rocm_version to test_util.py

charleshofer committed 29 days ago

1bac5e70

SWDEV-512768 - Replace hipGetLastError with hipExtGetLastError

dsicarov-amd committed 29 days ago

224b5baf

Wrap HIP inline functions in anonymous namespaces in vendor.h

AratiGanesh committed 29 days ago

2b08a20d

Older