ROCm/jax

Pull Requests Commits

Move to new github standard for composite actions

pelumi1163 committed 76 days ago

12f6cec3

Add sleep to the build step

pelumi1163 committed 76 days ago

29f6655f

Revert CI-cert path

pelumi1163 committed 80 days ago

64542bff

pelumi1163 committed 80 days ago

00c6cc03

pelumi1163 committed 80 days ago

ff7b9f9b

Change CI-cert path

pelumi1163 committed 80 days ago

7aa606d4

Remove Volume mount

pelumi1163 committed 80 days ago

084d423d

Test K8 runners

pelumi1163 committed 82 days ago

65c2cf57

Trying 8 processes

pelumi1163 committed 85 days ago

30b901bd

Trying 3 processes

pelumi1163 committed 86 days ago

b0dc7aa2

Trying 3 processes

pelumi1163 committed 86 days ago

f523631f

Reverting to 4 processes

pelumi1163 committed 86 days ago

da408051

Testing 24 processes

pelumi1163 committed 86 days ago

64866471

Revert to 4 processes

pelumi1163 committed 87 days ago

9214d72b

pause concurrency

pelumi1163 committed 87 days ago

bb7f8ca5

test more processes

pelumi1163 committed 87 days ago

c953fb47

Merge branch 'amd-main' of github.com:ROCm/jax into amd-main

pelumi1163 committed 87 days ago

882eb65a

[ROCm] Enable previously skipped unit tests on ROCm platform

magaonka-amd committed 88 days ago

b691d000

Simplify pytest parallelism calculation for ROCm CI

magaonka-amd committed 88 days ago

7643edca

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

pelumi1163 committed 88 days ago

31f05e4d

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

magaonka-amd committed 88 days ago

027c3275

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

Ruturaj4 committed 88 days ago

141e326f

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

magaonka-amd committed 88 days ago

b998c2ca

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

Ruturaj4 committed 88 days ago

7705000e

[ROCm] Add release pipeline CI workflows

mminutoli committed 88 days ago

eea666e3

rollback of changelist 961233365.

jinliangwei committed 88 days ago

4f313089

Update XLA dependency to use revision

Google-ML-Automation committed 88 days ago

0c62bd53

Do not print the content of an array from `repr`/`str` if it's bigger than 1MiB

junwhanahn committed 89 days ago

f73e314d

Update XLA dependency to use revision

Google-ML-Automation committed 89 days ago

cec06d11

Merge pull request #39835 from mattjj:push-sqltwqqupsms

Google-ML-Automation committed 89 days ago

e2593fd2

Older