ROCm/jax

Pull Requests Commits

Point the XLA pin at the merged bit_cast fix

gulsumgudukbay committed 41 days ago

acbebb22

Move the XLA pin past the device-code bit_cast fix

gulsumgudukbay committed 42 days ago

9209b576

[ROCm] Fix ROCm version in plugin wheel metadata (#843)

gulsumgudukbay committed 45 days ago

Verified 79ae4227

[ROCm] Simplify pytest parallelism calculation for ROCm CI

magaonka-amd committed 48 days ago

8a058bfd

[ROCm] Split the multi-accelerator tests into their own script (#834)

gulsumgudukbay committed 48 days ago

Verified 50e7d9cb

[ROCm] Drop the rocm-sdk PATH block from run_pytest_rocm.sh

gulsumgudukbay committed 54 days ago

49ec3f82

Derive the ROCm plugin package name from the ROCm major version

gulsumgudukbay committed 60 days ago

3a0d6b59

[ROCm][Pallas] Skip fused attention fwd tests that exceed device resources.

tsrw2048 committed 60 days ago

d4d63098

updated rocm/xla version

zahiqbal committed 70 days ago

7a0da134

updated XLA commit, JAX-side XLA patch that declares the missing trait in the shared header (mirroring the Bf16 struct with Eigen::half), which fixes both the ROCm and CUDA paths

zahiqbal committed 84 days ago

fd74cbe4

Deprecate 2d arguments to jax.numpy.cross.

hawkinsp committed 95 days ago

94a418dc

Update ROCm runner labels to new labels

psanal35 committed 95 days ago

8eb8f403

update XLA commit id on v0.10.1-draft

zahiqbal committed 95 days ago

416c7dce

Cherry-pick: Fix visibility in BUILD files (#757) (#759)

magaonka-amd committed 97 days ago

1e3ed533

[ROCm][Pallas] Skip fused attention bwd tests that exceed device resources

mminutoli committed 97 days ago

d62df769

Point JAX to newest downstream XLA

tsrw2048 committed 105 days ago

06158af4

Build all wheels but test with only one set for faster testing.

tsrw2048 committed 105 days ago

e4f0e43c

Revert "Trim down testing in release validation"

tsrw2048 committed 105 days ago

4c012443

[ROCm] CI improvements for ROCm

magaonka-amd committed 105 days ago

09b79d60

Update Rock manylinux image to 7.14

tsrw2048 committed 112 days ago

0a962d19

[ROCm] Put TheRock SDK bin on PATH via so rocminfo/rocm-smi resolve in /opt/rocm-less images

gulsumgudukbay committed 112 days ago

61cf8cdd

Disable hipblaslt in TheRock to avoid current Tensile issues.

tsrw2048 committed 112 days ago

6c5e4e20

Update Rock image to 7.14

tsrw2048 committed 112 days ago

4a737663

Add all specific target archs for ROCm

tsrw2048 committed 112 days ago

8a6e7292

Remove outdated ROCm skip that depends on ROCm versions < 6.4 (#776)

gulsumgudukbay committed 112 days ago

4e1e67cd

[ROCm] xdist: select GPU from runner-allocated ROCR slice

magaonka-amd committed 112 days ago

6dba736e

Add HSA_NO_SCRATCH_RECLAIM to release validation.

tsrw2048 committed 112 days ago

adec00da

Trim down testing in release validation

tsrw2048 committed 112 days ago

4fef3f46

Cherry pick env var leak fix from open upstream.

tsrw2048 committed 112 days ago

0776f25f

[ROCm] Set HSA_NO_SCRATCH_RECLAIM=1 in ROCm test workflows

gulsumgudukbay committed 112 days ago

f75948e8

Older