ROCm/jax

Pull Requests Commits

Support wheels as build deps

alekstheod committed 7 hours ago

49cb507a

alekstheod committed 3 days ago

a5d48b4f

Fix KeyError for bytes_reservable_limit on ROCm

magaonka-amd committed 4 days ago

cef02f3a

Enable testSvdSubsetByIndex for ROCm with subset_by_index skip (#603)

phambinhfin committed 4 days ago

e3ffcb08

Enabled ToeplitzSymmetricConstruction unit tests for ROCm devices (#608)

tsrw2048 committed 4 days ago

b9b014dc

Enabled testTridiagonal to run on ROCm devices (#607)

tsrw2048 committed 4 days ago

9e3371ea

Ruturaj4 committed 4 days ago

fe81a738

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

tsrw2048 committed 6 days ago

Verified 9b333623

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

tsrw2048 committed 6 days ago

ed8596b0

Fix Numpy signatures test (#598)

magaonka-amd committed 6 days ago

651aded6

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

phambinhfin committed 6 days ago

d9324638

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

phambinhfin committed 6 days ago

f4bf6876

adding abort logic to rocm/jax (#590)

gulsumgudukbay committed 6 days ago

abbad31d

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

charleshofer committed 6 days ago

bd84e8b4

Add rocm test requirements file (#570)

AratiGanesh committed 6 days ago

333b6091

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

Arech8 committed 6 days ago

fb8ec61c

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

Arech8 committed 6 days ago

bb7522db

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

phambinhfin committed 6 days ago

1a33f0ea

Make nvidia version data optional for ROCm builds

phambinhfin committed 6 days ago

2cc64461

Fix hipSparse CSR algorithm mappings for ROCm 7

phambinhfin committed 6 days ago

4bab82c7

Add shared utility function get_rocm_version to test_util.py

charleshofer committed 6 days ago

90cd6419

SWDEV-512768 - Replace hipGetLastError with hipExtGetLastError

dsicarov-amd committed 6 days ago

c692f178

Wrap HIP inline functions in anonymous namespaces in vendor.h

AratiGanesh committed 6 days ago

1ccd416c

charleshofer committed 6 days ago

ba55fd63

hipblas typedef fix

charleshofer committed 6 days ago

0a3400c1

Make jaxlib targets visible

charleshofer committed 6 days ago

22d9fb19

Remove nvidia_wheel_versions

charleshofer committed 6 days ago

d26fc94f

Prepare for JAX release 0.8.2

belitskiy committed 35 days ago

3f125024

[pallas:mosaic] `pltpu.emit_pipeline` now accepts block specs in HBM

superbobry committed 35 days ago

e2815d51

[mpmd] Fix stage_id field nanobind compatability

Google-ML-Automation committed 35 days ago

69a6cca9

Older