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
520 workflow runs
520 workflow runs

Filter by Event

Filter by Status

Filter by Branch

Filter by Actor

Merge pull request #26977 from jakevdp:fix-expn ROCm GPU Post-Merge Check #470: Commit c942b0f pushed by copybara-service bot
March 10, 2025 21:09 15m 52s main
main
March 10, 2025 21:09 15m 52s
[Pallas/Fuser] Add debug option to fuser.fuse that prints out jaxpr ROCm GPU Post-Merge Check #469: Commit 87272fb pushed by copybara-service bot
March 10, 2025 21:07 9m 15s main
main
March 10, 2025 21:07 9m 15s
Rename dot_with_no_batch_dims_saveable to `dots_with_no_batch_dims_… ROCm GPU Post-Merge Check #468: Commit affe2e7 pushed by copybara-service bot
March 10, 2025 20:04 9m 25s main
main
March 10, 2025 20:04 9m 25s
Define lax.ragged_dot_general and express lax.ragged_dot in terms of it. ROCm GPU Post-Merge Check #467: Commit b6d4fe5 pushed by copybara-service bot
March 10, 2025 19:25 9m 15s main
main
March 10, 2025 19:25 9m 15s
Merge pull request #27035 from vfdev-5:add-file-zip-to-tsan-ci-jobs ROCm GPU Post-Merge Check #466: Commit 64beebb pushed by copybara-service bot
March 10, 2025 19:12 9m 9s main
main
March 10, 2025 19:12 9m 9s
Merge pull request #26525 from wenscarl:e2m1fn ROCm GPU Post-Merge Check #465: Commit 18f2f19 pushed by copybara-service bot
March 10, 2025 18:46 17m 9s main
main
March 10, 2025 18:46 17m 9s
[Pallas] Small fix to TPU interpret mode (input_output_aliases + scal… ROCm GPU Post-Merge Check #464: Commit 73d20cd pushed by copybara-service bot
March 10, 2025 18:40 14m 3s main
main
March 10, 2025 18:40 14m 3s
Remove version limit for setuptools dependency. ROCm GPU Post-Merge Check #463: Commit 007fc7a pushed by copybara-service bot
March 10, 2025 18:36 9m 3s main
main
March 10, 2025 18:36 9m 3s
Modify version test to consider "rc" versions as well ROCm GPU Post-Merge Check #462: Commit d41e968 pushed by copybara-service bot
March 10, 2025 18:10 9m 11s main
main
March 10, 2025 18:10 9m 11s
Warn the user if transparent huge pages aren't enabled. ROCm GPU Post-Merge Check #461: Commit 5cb2994 pushed by copybara-service bot
March 10, 2025 17:38 9m 13s main
main
March 10, 2025 17:38 9m 13s
Merge pull request #27032 from dfm:lax-dtype ROCm GPU Post-Merge Check #460: Commit 14b215f pushed by copybara-service bot
March 10, 2025 17:19 9m 25s main
main
March 10, 2025 17:19 9m 25s
Merge pull request #26811 from dfm:direct-lin ROCm GPU Post-Merge Check #459: Commit ab0ce8a pushed by copybara-service bot
March 10, 2025 15:39 9m 6s main
main
March 10, 2025 15:39 9m 6s
[Mosaic GPU] Test the wgmma_op lowering when a is in registers. ROCm GPU Post-Merge Check #458: Commit d2bf034 pushed by copybara-service bot
March 10, 2025 15:25 10m 48s main
main
March 10, 2025 15:25 10m 48s
Merge pull request #27026 from garymm:patch-3 ROCm GPU Post-Merge Check #457: Commit 5a7ef40 pushed by copybara-service bot
March 10, 2025 15:20 5m 15s main
main
March 10, 2025 15:20 5m 15s
Move (most) jaxlib linalg custom call registration into JAX. ROCm GPU Post-Merge Check #456: Commit 21884d4 pushed by copybara-service bot
March 10, 2025 15:17 9m 24s main
main
March 10, 2025 15:17 9m 24s
[pallas:mosaic_gpu] Added support for math functions to the WG lowering ROCm GPU Post-Merge Check #455: Commit 91340ea pushed by copybara-service bot
March 10, 2025 12:08 9m 8s main
main
March 10, 2025 12:08 9m 8s
Update XLA dependency to use revision ROCm GPU Post-Merge Check #454: Commit f906d2b pushed by copybara-service bot
March 10, 2025 11:14 9m 23s main
main
March 10, 2025 11:14 9m 23s
[Pallas/Mosaic GPU] Add lowerings/layout inference for all the necess… ROCm GPU Post-Merge Check #453: Commit 75d8702 pushed by copybara-service bot
March 10, 2025 09:14 9m 12s main
main
March 10, 2025 09:14 9m 12s
Update XLA dependency to use revision ROCm GPU Post-Merge Check #452: Commit b9fb69d pushed by copybara-service bot
March 9, 2025 10:55 24m 6s main
main
March 9, 2025 10:55 24m 6s
Merge pull request #27006 from dfm:more-direct-lin-debug-info ROCm GPU Post-Merge Check #451: Commit 922935a pushed by copybara-service bot
March 8, 2025 14:46 9m 24s main
main
March 8, 2025 14:46 9m 24s
Update XLA dependency to use revision ROCm GPU Post-Merge Check #450: Commit 04696b4 pushed by copybara-service bot
March 8, 2025 11:27 9m 24s main
main
March 8, 2025 11:27 9m 24s
[Mosaic TPU][Pallas] Add pl.reciprocal ROCm GPU Post-Merge Check #449: Commit 0f0636a pushed by copybara-service bot
March 8, 2025 02:29 9m 15s main
main
March 8, 2025 02:29 9m 15s
Merge pull request #27010 from mattjj:direct-linearize-fixes-3 ROCm GPU Post-Merge Check #448: Commit 4988adc pushed by copybara-service bot
March 8, 2025 02:15 9m 32s main
main
March 8, 2025 02:15 9m 32s
Merge pull request #27005 from mattjj:direct-linearize-fixes-2 ROCm GPU Post-Merge Check #447: Commit 4660d7b pushed by copybara-service bot
March 8, 2025 01:17 9m 8s main
main
March 8, 2025 01:17 9m 8s
Support MHA in ragged paged attention for packed type ROCm GPU Post-Merge Check #446: Commit 041f575 pushed by copybara-service bot
March 7, 2025 22:47 9m 11s main
main
March 7, 2025 22:47 9m 11s