ROCm/jax

Pull Requests Commits

alekstheod committed 53 days ago

a5d48b4f

Fix KeyError for bytes_reservable_limit on ROCm

magaonka-amd committed 54 days ago

cef02f3a

Enable testSvdSubsetByIndex for ROCm with subset_by_index skip (#603)

phambinhfin committed 54 days ago

e3ffcb08

Enabled ToeplitzSymmetricConstruction unit tests for ROCm devices (#608)

tsrw2048 committed 54 days ago

b9b014dc

Enabled testTridiagonal to run on ROCm devices (#607)

tsrw2048 committed 54 days ago

9e3371ea

Ruturaj4 committed 54 days ago

fe81a738

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

tsrw2048 committed 57 days ago

Verified 9b333623

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

tsrw2048 committed 57 days ago

ed8596b0

Fix Numpy signatures test (#598)

magaonka-amd committed 57 days ago

651aded6

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

phambinhfin committed 57 days ago

d9324638

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

phambinhfin committed 57 days ago

f4bf6876

adding abort logic to rocm/jax (#590)

gulsumgudukbay committed 57 days ago

abbad31d

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

charleshofer committed 57 days ago

bd84e8b4

Add rocm test requirements file (#570)

AratiGanesh committed 57 days ago

333b6091

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

Arech8 committed 57 days ago

fb8ec61c

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

Arech8 committed 57 days ago

bb7522db

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

phambinhfin committed 57 days ago

1a33f0ea

Make nvidia version data optional for ROCm builds

phambinhfin committed 57 days ago

2cc64461

Fix hipSparse CSR algorithm mappings for ROCm 7

phambinhfin committed 57 days ago

4bab82c7

Add shared utility function get_rocm_version to test_util.py

charleshofer committed 57 days ago

90cd6419

SWDEV-512768 - Replace hipGetLastError with hipExtGetLastError

dsicarov-amd committed 57 days ago

c692f178

Wrap HIP inline functions in anonymous namespaces in vendor.h

AratiGanesh committed 57 days ago

1ccd416c

charleshofer committed 57 days ago

ba55fd63

hipblas typedef fix

charleshofer committed 57 days ago

0a3400c1

Make jaxlib targets visible

charleshofer committed 57 days ago

22d9fb19

Remove nvidia_wheel_versions

charleshofer committed 57 days ago

d26fc94f

Prepare for JAX release 0.8.2

belitskiy committed 85 days ago

3f125024

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

superbobry committed 85 days ago

e2815d51

[mpmd] Fix stage_id field nanobind compatability

Google-ML-Automation committed 85 days ago

69a6cca9

[MGPU] Doc fix and shape size fix for broadcast in WGSplatFragLayout.

golechwierowicz committed 85 days ago

315bb935

Older