Skip to content

Actions: jax-ml/jax

Actions

ROCm GPU Post-Merge Check

Actions

Loading...
Loading

Show workflow options

Create status badge

Loading
rocm-ci.yml
4,627 workflow runs
4,627 workflow runs

Filter by Event

Filter by Status

Filter by Branch

Filter by Actor

Update XLA dependency to use revision http://github.com/openxla/xla/c… ROCm GPU Post-Merge Check #4627: Commit 8dd92e5 pushed by copybara-service bot
Pending main
main
Pending
[Mosaic] Add sign_extended attribute to tpu::UnpackSubelementsOp. ROCm GPU Post-Merge Check #4626: Commit fbf4610 pushed by copybara-service bot
6h 41m 39s main
main
6h 41m 39s
Make eager shmap + check_vma=False + axis_index work with explicit mode. ROCm GPU Post-Merge Check #4625: Commit 9855a2e pushed by copybara-service bot
3h 14m 1s main
main
3h 14m 1s
Disallow nesting with jax.experimental.shard_map. This means to remov… ROCm GPU Post-Merge Check #4624: Commit dfadef7 pushed by copybara-service bot
5h 22m 49s main
main
5h 22m 49s
Update XLA dependency to use revision http://github.com/openxla/xla/c… ROCm GPU Post-Merge Check #4623: Commit bae1fbd pushed by copybara-service bot
Queued main
main
Queued
Add a fancy transpose rule for reshard_p so that we can do the accu… ROCm GPU Post-Merge Check #4622: Commit c755701 pushed by copybara-service bot
3h 2m 23s main
main
3h 2m 23s
[JAX] Make Colocated Python object destruction use a sorted device list ROCm GPU Post-Merge Check #4621: Commit 5cdc116 pushed by copybara-service bot
1h 34m 38s main
main
1h 34m 38s
Merge pull request #32535 from jakevdp:pallas-shard-map ROCm GPU Post-Merge Check #4620: Commit 16a8ca4 pushed by copybara-service bot
1m 37s main
main
1m 37s
[IFRT] Migrate Array::pjrt_layout() callers to interpret nullptr … ROCm GPU Post-Merge Check #4619: Commit e64fe93 pushed by copybara-service bot
15m 38s main
main
15m 38s
Update pip and uv default index with mirror artifacts repository … ROCm GPU Post-Merge Check #4618: Commit 653a846 pushed by copybara-service bot
58m 29s main
main
58m 29s
bump rules_ml_toolchain version ROCm GPU Post-Merge Check #4617: Commit 164fcc1 pushed by copybara-service bot
1h 9m 1s main
main
1h 9m 1s
Reverts e2a4c1a57a6942d4432fbb7ca9ddb0d9377cb49c ROCm GPU Post-Merge Check #4616: Commit d2391fc pushed by copybara-service bot
1h 27m 51s main
main
1h 27m 51s
Remove jax.experimental.shard_map docstring ROCm GPU Post-Merge Check #4615: Commit 6e85aa4 pushed by copybara-service bot
2h 5m 3s main
main
2h 5m 3s
Merge pull request #32527 from jax-ml:shmap ROCm GPU Post-Merge Check #4614: Commit c15d107 pushed by copybara-service bot
34m 12s main
main
34m 12s
[bazel] add visibility hook for legacy jax.experimental.shard_map ROCm GPU Post-Merge Check #4613: Commit 39183b4 pushed by copybara-service bot
3m 8s main
main
3m 8s
Merge pull request #32522 from jakevdp:shard-map-docs ROCm GPU Post-Merge Check #4612: Commit 0a42953 pushed by copybara-service bot
26m 23s main
main
26m 23s
Update pip and uv default index with mirror artifacts repository … ROCm GPU Post-Merge Check #4611: Commit f99c3e6 pushed by copybara-service bot
30m 39s main
main
30m 39s
Document that negative low/high padding is allowed for lax.pad. ROCm GPU Post-Merge Check #4610: Commit 6c64f37 pushed by copybara-service bot
1h 22m 26s main
main
1h 22m 26s
Switch JAX to use an FFI call for cholesky decompositions rather than… ROCm GPU Post-Merge Check #4609: Commit cf9a352 pushed by copybara-service bot
15m 34s main
main
15m 34s
[Mosaic GPU] Support splitting of minor dimension for collective copi… ROCm GPU Post-Merge Check #4608: Commit 83d8304 pushed by copybara-service bot
1h 12m 28s main
main
1h 12m 28s
Merge pull request #32499 from mattjj:issue32479 ROCm GPU Post-Merge Check #4607: Commit c4cfff9 pushed by copybara-service bot
4m 53s main
main
4m 53s
Merge pull request #32493 from mattjj:newstyle-hiprimitive2 ROCm GPU Post-Merge Check #4606: Commit 05b4ac8 pushed by copybara-service bot
2m 20s main
main
2m 20s
[Pallas:MGPU] Implement a basic reduce-scatter collective using multi… ROCm GPU Post-Merge Check #4605: Commit 97d2e20 pushed by copybara-service bot
3m 24s main
main
3m 24s
[Mosaic GPU][NFC] Save 5 LOC. ROCm GPU Post-Merge Check #4604: Commit 2fb239d pushed by copybara-service bot
2m 34s main
main
2m 34s
[Pallas:MGPU] Add a collective matmul tutorial ROCm GPU Post-Merge Check #4603: Commit dfdb0dc pushed by copybara-service bot
3m 30s main
main
3m 30s