ROCm/jax

Pull Requests Commits

Move xla pin to a9f7986abbfecaf82097fc90abcc8ecbfa4c7b48

draganmladjenovic committed 85 days ago

2aa3f9e2

Add single and multi gpu versions of testing scripts.

tsrw2048 committed 127 days ago

98d8ab72

Point to new XLA branch/commit.

tsrw2048 committed 128 days ago

a5921c1b

Pin numpy version

tsrw2048 committed 128 days ago

d4433253

Deprecate 2d arguments to jax.numpy.cross.

hawkinsp committed 133 days ago

94a418dc

Update ROCm runner labels to new labels

psanal35 committed 133 days ago

8eb8f403

update XLA commit id on v0.10.1-draft

zahiqbal committed 133 days ago

416c7dce

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

magaonka-amd committed 135 days ago

1e3ed533

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

mminutoli committed 135 days ago

d62df769

Point JAX to newest downstream XLA

tsrw2048 committed 143 days ago

06158af4

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

tsrw2048 committed 143 days ago

e4f0e43c

Revert "Trim down testing in release validation"

tsrw2048 committed 143 days ago

4c012443

[ROCm] CI improvements for ROCm

magaonka-amd committed 143 days ago

09b79d60

Update Rock manylinux image to 7.14

tsrw2048 committed 150 days ago

0a962d19

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

gulsumgudukbay committed 150 days ago

61cf8cdd

Disable hipblaslt in TheRock to avoid current Tensile issues.

tsrw2048 committed 150 days ago

6c5e4e20

Update Rock image to 7.14

tsrw2048 committed 150 days ago

4a737663

Add all specific target archs for ROCm

tsrw2048 committed 150 days ago

8a6e7292

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

gulsumgudukbay committed 150 days ago

4e1e67cd

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

magaonka-amd committed 150 days ago

6dba736e

Add HSA_NO_SCRATCH_RECLAIM to release validation.

tsrw2048 committed 150 days ago

adec00da

Trim down testing in release validation

tsrw2048 committed 150 days ago

4fef3f46

Cherry pick env var leak fix from open upstream.

tsrw2048 committed 150 days ago

0776f25f

[ROCm] Set HSA_NO_SCRATCH_RECLAIM=1 in ROCm test workflows

gulsumgudukbay committed 150 days ago

f75948e8

[ROCm] Embed TheRock-aware wheel RUNPATH at link time

gulsumgudukbay committed 150 days ago

f1a95235

Revert "temp push condition on workflow"

zahiqbal committed 157 days ago

d4698160

temp push condition on workflow

zahiqbal committed 157 days ago

7fcdd3a5

Added a release validation workflow for JAX ROCm wheels.

tsrw2048 committed 157 days ago

bd53c71d

[ROCm] Version-tier S3 layout, standardized CI

mminutoli committed 157 days ago

544a7b4d

Add org-based guards to all CI caller workflows

mminutoli committed 157 days ago

f81c3003

Older