ROCm/jax

Pull Requests Commits

Fix missing ROCm wheel download step name

Kunreddy committed 84 days ago

57f0fb09

Clear dist/logs after checkout on nodind GPU jobs.

Kunreddy committed 84 days ago

fb4f6629

Fix nodind nightly: restore dind build AWS/checkout, harden GPU shell steps.

Kunreddy committed 84 days ago

414ff337

Merge branch 'test/nodind-shell-steps' of https://github.com/ROCm/jax into test/nodind-shell-steps

Kunreddy committed 84 days ago

6a72732e

Keep nightly ROCm builds on dind CPU; GPU tests on nodind.

Kunreddy committed 84 days ago

27e44a60

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

mkunredd committed 84 days ago

82d767c8

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

Kunreddy committed 84 days ago

20748c6c

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

mkunredd committed 84 days ago

fd0b9b36

fix: init git repo before fetch in shell checkout step

mkunredd committed 84 days ago

fb0a082a

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

Kunreddy committed 84 days ago

631d799e

Add 1-GPU isolated nodind label for manual pytest validation

Kunreddy committed 86 days ago

6fba2db2

Expose nodind runner label for manual ROCm pytest validation

Kunreddy committed 86 days ago

b32e5cbc

Mount /home/runner/_work in ROCm workflow containers

Kunreddy committed 86 days ago

0c275445

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

mminutoli committed 87 days ago

d8b3f548

Remove stale ROCm skips from four GPU test cases

magaonka-amd committed 87 days ago

839c6488

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

magaonka-amd committed 87 days ago

e9978550

[ROCm] Enable previously skipped unit tests on ROCm platform

magaonka-amd committed 87 days ago

bde178f3

Simplify pytest parallelism calculation for ROCm CI

magaonka-amd committed 87 days ago

a0587192

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

pelumi1163 committed 87 days ago

b22e303b

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

magaonka-amd committed 87 days ago

398f6bfb

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

Ruturaj4 committed 87 days ago

39bcc56e

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

magaonka-amd committed 87 days ago

65aa7f4b

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

Ruturaj4 committed 87 days ago

1247cc28

[ROCm] Add release pipeline CI workflows

mminutoli committed 87 days ago

18402cf5

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

superbobry committed 87 days ago

4861a40d

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

hawkinsp committed 87 days ago

a25d587e

[triton] Added support for host-side TMA descriptors

superbobry committed 87 days ago

8585ff73

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

superbobry committed 87 days ago

263eed93

Implement AOT compilation for autotuned JAX-Triton kernels.

EusebioDM committed 87 days ago

8e07bd40

Merge pull request #39895 from jax-ml:bzlmod

Google-ML-Automation committed 87 days ago

51b5b00f

Older