ROCm/jax

Pull Requests Commits

Update build/rocm/run_jax_ut.sh

alekstheod committed 31 days ago

6694656b

Add additional args

alekstheod committed 31 days ago

25158ca8

Add deploy wheel script

alekstheod committed 31 days ago

5ff2ebd1

Use plain bazel to test jax, use hermetic rocm dependency

alekstheod committed 31 days ago

24d5725c

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

charleshofer committed 34 days ago

Verified ad54f8b1

Add rocm test requirements file (#570)

AratiGanesh committed 34 days ago

Verified 552778a4

add fix for jaxlib build whl (#571)

kiran-thumma committed 38 days ago

Verified a1a217fa

Remove build scripts (#578)

gulsumgudukbay committed 38 days ago

Verified a1968610

Revert "Remove old scripts and break build for ROCm" (#575)

gulsumgudukbay committed 38 days ago

Verified bf4facd8

Remove old scripts and break build for ROCm (#569)

gulsumgudukbay committed 39 days ago

Verified cafe7943

register rocm platform to mx datatype lowering path (#567)

ScXfjiang committed 45 days ago

Verified a2b972f2

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

Arech8 committed 53 days ago

Verified 861cc763

Implement Pallas lowering of lax.is_finite() for Triton (#564)

Arech8 committed 53 days ago

Verified c972c55d

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

Arech8 committed 53 days ago

Verified 339bf1e8

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

phambinhfin committed 60 days ago

Verified f893175a

Fix ROCm lib names copy pattern to use wildcard version

phambinhfin committed 61 days ago

709963b8

Make nvidia version data optional for ROCm builds

phambinhfin committed 61 days ago

b5762449

Fix hipSparse CSR algorithm mappings for ROCm 7

phambinhfin committed 61 days ago

28e04303

Add shared utility function get_rocm_version to test_util.py

charleshofer committed 61 days ago

aac9e9d2

SWDEV-512768 - Replace hipGetLastError with hipExtGetLastError

dsicarov-amd committed 61 days ago

954fde6a

Wrap HIP inline functions in anonymous namespaces in vendor.h

AratiGanesh committed 61 days ago

13a4d047

charleshofer committed 61 days ago

f4df93f6

hipblas typedef fix

charleshofer committed 61 days ago

9ed3820b

Make jaxlib targets visible

charleshofer committed 61 days ago

3f0a5750

Remove nvidia_wheel_versions

charleshofer committed 61 days ago

f037da4e

Reverts f44bc6e37f770a9d79d24061f7960235a2892407

jakeharmon8 committed 95 days ago

403977d2

Prepare for JAX release 0.8.0

jakeharmon8 committed 95 days ago

b1ce4c57

[Pallas] Fix 64-bit multiplication emulation in Philox kernel

justinjfu committed 95 days ago

07dad8e8

Update XLA hash for JAX release

jakeharmon8 committed 95 days ago

4f06e7a6

Fix race when dialects were registered in an MLIR context when it was being accessed concurrently.

hawkinsp committed 95 days ago

1bd3402c

Older