ROCm/jax

Pull Requests Commits

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

charleshofer committed 7 days ago

Verified ad54f8b1

Add rocm test requirements file (#570)

AratiGanesh committed 7 days ago

Verified 552778a4

add fix for jaxlib build whl (#571)

kiran-thumma committed 11 days ago

Verified a1a217fa

Remove build scripts (#578)

gulsumgudukbay committed 11 days ago

Verified a1968610

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

gulsumgudukbay committed 11 days ago

Verified bf4facd8

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

gulsumgudukbay committed 11 days ago

Verified cafe7943

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

ScXfjiang committed 17 days ago

Verified a2b972f2

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

Arech8 committed 26 days ago

Verified 861cc763

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

Arech8 committed 26 days ago

Verified c972c55d

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

Arech8 committed 26 days ago

Verified 339bf1e8

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

phambinhfin committed 32 days ago

Verified f893175a

Fix ROCm lib names copy pattern to use wildcard version

phambinhfin committed 33 days ago

709963b8

Make nvidia version data optional for ROCm builds

phambinhfin committed 33 days ago

b5762449

Fix hipSparse CSR algorithm mappings for ROCm 7

phambinhfin committed 33 days ago

28e04303

Add shared utility function get_rocm_version to test_util.py

charleshofer committed 33 days ago

aac9e9d2

SWDEV-512768 - Replace hipGetLastError with hipExtGetLastError

dsicarov-amd committed 33 days ago

954fde6a

Wrap HIP inline functions in anonymous namespaces in vendor.h

AratiGanesh committed 33 days ago

13a4d047

charleshofer committed 33 days ago

f4df93f6

hipblas typedef fix

charleshofer committed 33 days ago

9ed3820b

Make jaxlib targets visible

charleshofer committed 33 days ago

3f0a5750

Remove nvidia_wheel_versions

charleshofer committed 33 days ago

f037da4e

Reverts f44bc6e37f770a9d79d24061f7960235a2892407

jakeharmon8 committed 67 days ago

403977d2

Prepare for JAX release 0.8.0

jakeharmon8 committed 68 days ago

b1ce4c57

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

justinjfu committed 68 days ago

07dad8e8

Update XLA hash for JAX release

jakeharmon8 committed 68 days ago

4f06e7a6

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

hawkinsp committed 68 days ago

1bd3402c

Clean up dead code.

brianwa84 committed 68 days ago

b942e894

Remove cusolver_potrf_ffi from the list of covered backwards compatibility targets.

gnecula committed 68 days ago

ac87cbe0

[Pallas:MGPU] Expose multicast TMA stores

apaszke committed 68 days ago

69fd2052

[Pallas:MGPU] Add an all-gather implementation

apaszke committed 68 days ago

8eed7dc5

Older