ROCm/jax

Pull Requests Commits

Create rocm-test-requirements.txt

jvadakat committed 330 days ago

ed283d1d

add fix for jaxlib build whl (#571)

kiran-thumma committed 331 days ago

Verified a1a217fa

Remove build scripts (#578)

gulsumgudukbay committed 331 days ago

Verified a1968610

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

gulsumgudukbay committed 331 days ago

Verified bf4facd8

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

gulsumgudukbay committed 332 days ago

Verified cafe7943

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

ScXfjiang committed 337 days ago

Verified a2b972f2

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

Arech8 committed 346 days ago

Verified 861cc763

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

Arech8 committed 346 days ago

Verified c972c55d

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

Arech8 committed 346 days ago

Verified 339bf1e8

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

phambinhfin committed 353 days ago

Verified f893175a

Fix ROCm lib names copy pattern to use wildcard version

phambinhfin committed 354 days ago

709963b8

Make nvidia version data optional for ROCm builds

phambinhfin committed 354 days ago

b5762449

Fix hipSparse CSR algorithm mappings for ROCm 7

phambinhfin committed 354 days ago

28e04303

Add shared utility function get_rocm_version to test_util.py

charleshofer committed 354 days ago

aac9e9d2

SWDEV-512768 - Replace hipGetLastError with hipExtGetLastError

dsicarov-amd committed 354 days ago

954fde6a

Wrap HIP inline functions in anonymous namespaces in vendor.h

AratiGanesh committed 354 days ago

13a4d047

charleshofer committed 354 days ago

f4df93f6

hipblas typedef fix

charleshofer committed 354 days ago

9ed3820b

Make jaxlib targets visible

charleshofer committed 354 days ago

3f0a5750

Remove nvidia_wheel_versions

charleshofer committed 354 days ago

f037da4e

Reverts f44bc6e37f770a9d79d24061f7960235a2892407

jakeharmon8 committed 1 year ago

403977d2

Prepare for JAX release 0.8.0

jakeharmon8 committed 1 year ago

b1ce4c57

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

justinjfu committed 1 year ago

07dad8e8

Update XLA hash for JAX release

jakeharmon8 committed 1 year ago

4f06e7a6

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

hawkinsp committed 1 year ago

1bd3402c

Clean up dead code.

brianwa84 committed 1 year ago

b942e894

Remove cusolver_potrf_ffi from the list of covered backwards compatibility targets.

gnecula committed 1 year ago

ac87cbe0

[Pallas:MGPU] Expose multicast TMA stores

apaszke committed 1 year ago

69fd2052

[Pallas:MGPU] Add an all-gather implementation

apaszke committed 1 year ago

8eed7dc5

Add tpu_sc.subcore_barrier primitive.

brianwa84 committed 1 year ago

6aae6f43

Older