ROCm/jax

Pull Requests Commits

alekstheod committed 116 days ago

42a02dc0

Make deps parametrized

alekstheod committed 128 days ago

79f790d0

Support wheels as build deps

alekstheod committed 129 days ago

49cb507a

alekstheod committed 131 days ago

a5d48b4f

Fix KeyError for bytes_reservable_limit on ROCm

magaonka-amd committed 133 days ago

cef02f3a

Enable testSvdSubsetByIndex for ROCm with subset_by_index skip (#603)

phambinhfin committed 133 days ago

e3ffcb08

Enabled ToeplitzSymmetricConstruction unit tests for ROCm devices (#608)

tsrw2048 committed 133 days ago

b9b014dc

Enabled testTridiagonal to run on ROCm devices (#607)

tsrw2048 committed 133 days ago

9e3371ea

Ruturaj4 committed 133 days ago

fe81a738

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

tsrw2048 committed 135 days ago

Verified 9b333623

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

tsrw2048 committed 135 days ago

ed8596b0

Fix Numpy signatures test (#598)

magaonka-amd committed 135 days ago

651aded6

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

phambinhfin committed 135 days ago

d9324638

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

phambinhfin committed 135 days ago

f4bf6876

adding abort logic to rocm/jax (#590)

gulsumgudukbay committed 135 days ago

abbad31d

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

charleshofer committed 135 days ago

bd84e8b4

Add rocm test requirements file (#570)

AratiGanesh committed 135 days ago

333b6091

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

Arech8 committed 135 days ago

fb8ec61c

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

Arech8 committed 135 days ago

bb7522db

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

phambinhfin committed 135 days ago

1a33f0ea

Make nvidia version data optional for ROCm builds

phambinhfin committed 135 days ago

2cc64461

Fix hipSparse CSR algorithm mappings for ROCm 7

phambinhfin committed 135 days ago

4bab82c7

Add shared utility function get_rocm_version to test_util.py

charleshofer committed 135 days ago

90cd6419

SWDEV-512768 - Replace hipGetLastError with hipExtGetLastError

dsicarov-amd committed 135 days ago

c692f178

Wrap HIP inline functions in anonymous namespaces in vendor.h

AratiGanesh committed 135 days ago

1ccd416c

charleshofer committed 135 days ago

ba55fd63

hipblas typedef fix

charleshofer committed 135 days ago

0a3400c1

Make jaxlib targets visible

charleshofer committed 135 days ago

22d9fb19

Remove nvidia_wheel_versions

charleshofer committed 135 days ago

d26fc94f

Prepare for JAX release 0.8.2

belitskiy committed 164 days ago

3f125024

Older