ROCm/jax

Pull Requests Commits

Revert CI-cert path

pelumi1163 committed 2 days ago

64542bff

pelumi1163 committed 2 days ago

00c6cc03

pelumi1163 committed 2 days ago

ff7b9f9b

Change CI-cert path

pelumi1163 committed 2 days ago

7aa606d4

Remove Volume mount

pelumi1163 committed 2 days ago

084d423d

Test K8 runners

pelumi1163 committed 4 days ago

65c2cf57

Trying 8 processes

pelumi1163 committed 8 days ago

30b901bd

Trying 3 processes

pelumi1163 committed 8 days ago

b0dc7aa2

Trying 3 processes

pelumi1163 committed 8 days ago

f523631f

Reverting to 4 processes

pelumi1163 committed 9 days ago

da408051

Testing 24 processes

pelumi1163 committed 9 days ago

64866471

Revert to 4 processes

pelumi1163 committed 9 days ago

9214d72b

pause concurrency

pelumi1163 committed 10 days ago

bb7f8ca5

test more processes

pelumi1163 committed 10 days ago

c953fb47

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

pelumi1163 committed 10 days ago

882eb65a

[ROCm] Enable previously skipped unit tests on ROCm platform

magaonka-amd committed 10 days ago

b691d000

Simplify pytest parallelism calculation for ROCm CI

magaonka-amd committed 10 days ago

7643edca

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

pelumi1163 committed 10 days ago

31f05e4d

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

magaonka-amd committed 10 days ago

027c3275

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

Ruturaj4 committed 10 days ago

141e326f

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

magaonka-amd committed 10 days ago

b998c2ca

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

Ruturaj4 committed 10 days ago

7705000e

[ROCm] Add release pipeline CI workflows

mminutoli committed 10 days ago

eea666e3

rollback of changelist 961233365.

jinliangwei committed 10 days ago

4f313089

Update XLA dependency to use revision

Google-ML-Automation committed 10 days ago

0c62bd53

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

junwhanahn committed 11 days ago

f73e314d

Update XLA dependency to use revision

Google-ML-Automation committed 11 days ago

cec06d11

Merge pull request #39835 from mattjj:push-sqltwqqupsms

Google-ML-Automation committed 11 days ago

e2593fd2

Merge pull request #39837 from mattjj:push-yvqskzomlqvl

Google-ML-Automation committed 11 days ago

787b3a3e

[cleanup] remove process_call, now dead code

mattjj committed 11 days ago

82eef9fb

Older