ROCm/jax

Pull Requests Commits

rocdl alternatives for nvvm: elect.sync, shfl.sync

Arech8 committed 2 hours ago

ca38907f

DISABLE important assertion in fragmented_array. RESTORE IT!

Arech8 committed 2 hours ago

c1472560

RocmModuleToBinaryPass WIP broken

Arech8 committed 1 day ago

743316f1

lift ROCDL dialect to Python/JAX/jaxlib

Arech8 committed 1 day ago

91b36bee

what? it works already?

Arech8 committed 13 days ago

324220d5

WIP build + custom_call

Arech8 committed 15 days ago

d545bc8a

reformat + todo

Arech8 committed 19 days ago

f0c22f1d

moved all to the standard custom_call.cc instead of customized file

Arech8 committed 19 days ago

31bdc301

reformat custom_call.cc

Arech8 committed 19 days ago

a98bef8a

move rocm/custom_call back to main ./gpu dir and drop ./rocm

Arech8 committed 19 days ago

e66f2139

first e2e work scaffolding

Arech8 committed 20 days ago

bd94e2f7

reformat my custom_call

Arech8 committed 23 days ago

382cd043

almost initial custom_call.cc with #if 0 and launder

Arech8 committed 23 days ago

fab96936

ROCm specifics for `primitives.py`

Arech8 committed 23 days ago

5df60d35

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

ScXfjiang committed 26 days ago

Verified a2b972f2

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

Arech8 committed 34 days ago

Verified 861cc763

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

Arech8 committed 35 days ago

Verified c972c55d

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

Arech8 committed 35 days ago

Verified 339bf1e8

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

phambinhfin committed 41 days ago

Verified f893175a

Fix ROCm lib names copy pattern to use wildcard version

phambinhfin committed 42 days ago

709963b8

Make nvidia version data optional for ROCm builds

phambinhfin committed 42 days ago

b5762449

Fix hipSparse CSR algorithm mappings for ROCm 7

phambinhfin committed 42 days ago

28e04303

Add shared utility function get_rocm_version to test_util.py

charleshofer committed 42 days ago

aac9e9d2

SWDEV-512768 - Replace hipGetLastError with hipExtGetLastError

dsicarov-amd committed 42 days ago

954fde6a

Wrap HIP inline functions in anonymous namespaces in vendor.h

AratiGanesh committed 42 days ago

13a4d047

charleshofer committed 42 days ago

f4df93f6

hipblas typedef fix

charleshofer committed 42 days ago

9ed3820b

Make jaxlib targets visible

charleshofer committed 42 days ago

3f0a5750

Remove nvidia_wheel_versions

charleshofer committed 42 days ago

f037da4e

Reverts f44bc6e37f770a9d79d24061f7960235a2892407

jakeharmon8 committed 76 days ago

403977d2

Older