ROCm/jax

Pull Requests Commits

Reduce python versions for testing

tsrw2048 committed 119 days ago

6bc68ad3

Point to new downstream XLA branch

tsrw2048 committed 119 days ago

6c99daef

Revert "Trim down release validation jobs for testing."

tsrw2048 committed 125 days ago

7ce0cc92

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

mminutoli committed 125 days ago

d7fa5ce2

Trim down release validation jobs for testing.

tsrw2048 committed 126 days ago

4e4bce83

Add expanded set of specific arch targets to support.

tsrw2048 committed 126 days ago

80c90840

Point JAX to downstream XLA and bump XLA commit

tsrw2048 committed 126 days ago

5d50bd7e

Add HSA_NO_SCRATCH_RECLAIM env var and update Rock images to 7.14 in release workflow

tsrw2048 committed 126 days ago

5c870c42

Added a release validation workflow for JAX ROCm wheels.

tsrw2048 committed 126 days ago

776f8a90

Cherry pick env var leak fix from open upstream.

tsrw2048 committed 126 days ago

c1876694

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

gulsumgudukbay committed 126 days ago

3d72c19b

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

gulsumgudukbay committed 126 days ago

d525d049

[ROCm] Set HSA_NO_SCRATCH_RECLAIM=1 in ROCm test workflows

gulsumgudukbay committed 126 days ago

20f77158

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

gulsumgudukbay committed 126 days ago

6bf8ce5e

Add org-based guards to all CI caller workflows

mminutoli committed 126 days ago

2bf7b3d7

[ROCm] Version-tier S3 layout, standardized CI

mminutoli committed 126 days ago

cf269722

[ROCm] Add `remote_download_outputs` option at build invocation.

tsrw2048 committed 126 days ago

f3fd9a7b

[ROCm] xdist: select GPU from runner-allocated ROCR slice (#794)

mminutoli committed 130 days ago

Verified 42886f9b

[ROCm] Use grid=(1,) in pallas test_non_range_while_loop

magaonka-amd committed 152 days ago

cbcc4e01

Update rules_ml_toolchain reference

alekstheod committed 166 days ago

b2e32655

Override HIP_VISIBLE_DEVICES per ROCm xdist worker

magaonka-amd committed 166 days ago

45b7af75

[ROCm] Split ROCm pytest into single- and multi-accelerator passes

magaonka-amd committed 166 days ago

95aba756

Use CloudFront download for ROCm plugin/PJRT wheels instead of S3

psanal35 committed 166 days ago

acd00f93

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

magaonka-amd committed 166 days ago

a23d3768

Add a libtpu version guard in test_dce_sink_prevents_xla_dce

yashk2810 committed 181 days ago

a33ed614

Remove unused EXTRA_INDEX environment variable

nitins17 committed 181 days ago

79e8ef93

Allow shell globbing for libtpu wheel path in pytest_tpu.yml

nitins17 committed 181 days ago

83e0ac50

Prepare for JAX release 0.10.0

yashk2810 committed 181 days ago

08a228a9

Fix a bug in WeakValueInterner.

hawkinsp committed 181 days ago

2362cc76

Addressed an old pytype TODO

superbobry committed 181 days ago

c12e417e

Older