ROCm/jax

Pull Requests Commits

pytest_rocm: full test suite on DPX with XLA_PYTHON_CLIENT_MEM_FRACTION=0.22

mkunredd committed 56 days ago

6e656707

fix: replace jq with python3 json parsing (jq not in JAX base image)

mkunredd committed 56 days ago

7383efbd

pytest_rocm: DPX smoke — n=1 worker, skip slow tests (avoid OOM)

mkunredd committed 56 days ago

1ed0e925

pytest_rocm: DPX smoke test — run lax_test.py only (30 min timeout)

mkunredd committed 56 days ago

aeecd944

pytest_rocm: remove container: block for DPX nodind (JAX base image as runner)

mkunredd committed 56 days ago

c75225a0

pytest_rocm: add amd-do-linux.jax.gpu.gfx950.1-dpx runner option

mkunredd committed 56 days ago

82d767c8

Enable full nightly wheel tests on true no-dind runners.

Kunreddy committed 56 days ago

20748c6c

fix: set safe.directory before git init in checkout step

mkunredd committed 56 days ago

fd0b9b36

fix: init git repo before fetch in shell checkout step

mkunredd committed 56 days ago

fb0a082a

pytest_rocm: replace JS actions with shell for no-dind container: support

Kunreddy committed 57 days ago

631d799e

Add 1-GPU isolated nodind label for manual pytest validation

Kunreddy committed 58 days ago

6fba2db2

Expose nodind runner label for manual ROCm pytest validation

Kunreddy committed 58 days ago

b32e5cbc

Mount /home/runner/_work in ROCm workflow containers

Kunreddy committed 58 days ago

0c275445

[ROCm] nightly: build jax/jaxlib on fork runners

mminutoli committed 59 days ago

d8b3f548

Remove stale ROCm skips from four GPU test cases

magaonka-amd committed 59 days ago

839c6488

Enable 12 more single-GPU bazel test targets in ROCm CI

magaonka-amd committed 59 days ago

e9978550

[ROCm] Enable previously skipped unit tests on ROCm platform

magaonka-amd committed 59 days ago

bde178f3

Simplify pytest parallelism calculation for ROCm CI

magaonka-amd committed 59 days ago

a0587192

Move Bazel test to the jax-base image (#831)

pelumi1163 committed 59 days ago

b22e303b

ci: retire ROCm 7.2.0 and TheRock 7.13 flows; run only TheRock 7.14 + latest

magaonka-amd committed 59 days ago

398f6bfb

ci: bump ROCm release wheel tests to TheRock 7.14 GA (#817)

Ruturaj4 committed 59 days ago

39bcc56e

Enable additional single-GPU bazel tests in ROCm CI (#816)

magaonka-amd committed 59 days ago

65aa7f4b

ci: run ROCm release wheel tests on TheRock 7.14.0rc3 images (#812)

Ruturaj4 committed 59 days ago

1247cc28

[ROCm] Add release pipeline CI workflows

mminutoli committed 59 days ago

18402cf5

[pallas:mgpu] Automatically infer `PROMISE_IN_BOUNDS` in `emit_pipeline`

superbobry committed 59 days ago

4861a40d

Replace closed form expansion of 2x2 and 3x3 determinants with a closed form LU decomposition with row pivoting.

hawkinsp committed 59 days ago

a25d587e

[triton] Added support for host-side TMA descriptors

superbobry committed 59 days ago

8585ff73

[mosaic_gpu] The `barrier` operand to `async_load` is now optional on pre-Hopper GPUs

superbobry committed 59 days ago

263eed93

Implement AOT compilation for autotuned JAX-Triton kernels.

EusebioDM committed 59 days ago

8e07bd40

Merge pull request #39895 from jax-ml:bzlmod

Google-ML-Automation committed 59 days ago

51b5b00f

Older