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

Filter by Event

Filter by Status

Filter by Branch

Filter by Actor

Merge pull request #27044 from carlosgmartin:add_breadcrumbs_to_docs ROCm GPU Post-Merge Check #517: Commit ba367cd pushed by copybara-service bot
March 12, 2025 22:14 9m 11s main
main
March 12, 2025 22:14 9m 11s
[Mosaic TPU][NFC] Throw NYI error instead of crash when squeeze ref t… ROCm GPU Post-Merge Check #516: Commit 12c0987 pushed by copybara-service bot
March 12, 2025 21:18 12m 39s main
main
March 12, 2025 21:18 12m 39s
Add a set_mesh API to jax.sharding. set_mesh sets the sharding an… ROCm GPU Post-Merge Check #515: Commit 47480b4 pushed by copybara-service bot
March 12, 2025 21:12 9m 12s main
main
March 12, 2025 21:12 9m 12s
[sharding_in_types] Make reshard work with np.array. ROCm GPU Post-Merge Check #514: Commit 8674495 pushed by copybara-service bot
March 12, 2025 20:41 9m 3s main
main
March 12, 2025 20:41 9m 3s
[Pallas] Plumb compiler flags through source mapper. ROCm GPU Post-Merge Check #513: Commit 6978f35 pushed by copybara-service bot
March 12, 2025 18:20 9m 25s main
main
March 12, 2025 18:20 9m 25s
[mosaic_gpu/pallas:mgpu] Eradicate wgmma_layout ROCm GPU Post-Merge Check #512: Commit b34f56b pushed by copybara-service bot
March 12, 2025 17:47 9m 21s main
main
March 12, 2025 17:47 9m 21s
Merge pull request #27092 from pearu:pearu/gammainc-bug-fix ROCm GPU Post-Merge Check #511: Commit 3de7ecf pushed by copybara-service bot
March 12, 2025 17:20 9m 14s main
main
March 12, 2025 17:20 9m 14s
Merge pull request #27041 from carlosgmartin:fix_binomial_value_error ROCm GPU Post-Merge Check #510: Commit e7d10a2 pushed by copybara-service bot
March 12, 2025 17:05 9m 13s main
main
March 12, 2025 17:05 9m 13s
[sharding_in_types] Initial commit to add `varying_manual_axes: froze… ROCm GPU Post-Merge Check #509: Commit abcc7fd pushed by copybara-service bot
March 12, 2025 15:29 16m 35s main
main
March 12, 2025 15:29 16m 35s
Fix integer overflow in workspace size computations for experimental.… ROCm GPU Post-Merge Check #508: Commit 8b7cfcb pushed by copybara-service bot
March 12, 2025 15:22 15m 0s main
main
March 12, 2025 15:22 15m 0s
[pallas:mosaic_gpu] Added support for reductions to the WG lowering ROCm GPU Post-Merge Check #507: Commit e33f3fc pushed by copybara-service bot
March 12, 2025 15:18 9m 15s main
main
March 12, 2025 15:18 9m 15s
Fix matrix exclude syntax in TPU tests block ROCm GPU Post-Merge Check #506: Commit d89835a pushed by copybara-service bot
March 12, 2025 14:12 9m 9s main
main
March 12, 2025 14:12 9m 9s
Ignore Pallas TPU tests when testing with the oldest supported libtpu ROCm GPU Post-Merge Check #505: Commit a6ab6bb pushed by copybara-service bot
March 12, 2025 12:20 9m 21s main
main
March 12, 2025 12:20 9m 21s
Update XLA dependency to use revision ROCm GPU Post-Merge Check #504: Commit 61ba2b2 pushed by copybara-service bot
March 12, 2025 11:52 9m 20s main
main
March 12, 2025 11:52 9m 20s
Add support for scratch buffers in jax_triton. ROCm GPU Post-Merge Check #503: Commit 74b4d86 pushed by copybara-service bot
March 12, 2025 03:49 9m 18s main
main
March 12, 2025 03:49 9m 18s
Run single python version for v4-8 and min & max for v5e-8 for TPU te… ROCm GPU Post-Merge Check #502: Commit ff751ec pushed by copybara-service bot
March 12, 2025 03:03 9m 8s main
main
March 12, 2025 03:03 9m 8s
add autodiff rules for jax.lax.ragged_all_to_all collective ROCm GPU Post-Merge Check #501: Commit 66a6eb2 pushed by copybara-service bot
March 12, 2025 01:22 9m 24s main
main
March 12, 2025 01:22 9m 24s
Rename get_ty to typeof which is an alias of get_aval ROCm GPU Post-Merge Check #500: Commit 3a26804 pushed by copybara-service bot
March 12, 2025 00:34 9m 21s main
main
March 12, 2025 00:34 9m 21s
[Pallas/Fuser] Add custom evaluate to allow/disallow transposes ROCm GPU Post-Merge Check #499: Commit c6b164d pushed by copybara-service bot
March 11, 2025 23:35 8m 59s main
main
March 11, 2025 23:35 8m 59s
Fix a bug where full and use_mesh outside jit did not work becaus… ROCm GPU Post-Merge Check #498: Commit f45cbf3 pushed by copybara-service bot
March 11, 2025 22:25 8m 52s main
main
March 11, 2025 22:25 8m 52s
[Pallas TPU] Fix preferred_element_type propagation in dot_general wi… ROCm GPU Post-Merge Check #497: Commit 29bfd00 pushed by copybara-service bot
March 11, 2025 22:06 9m 27s main
main
March 11, 2025 22:06 9m 27s
Upgrade ml-dtypes version in py3.10-py3.13 hermetic python lock… ROCm GPU Post-Merge Check #496: Commit 13eb8d3 pushed by copybara-service bot
March 11, 2025 21:41 9m 15s main
main
March 11, 2025 21:41 9m 15s
Remove unsupported mac x86 CI build options ROCm GPU Post-Merge Check #495: Commit 4df691e pushed by copybara-service bot
March 11, 2025 21:12 9m 14s main
main
March 11, 2025 21:12 9m 14s
Merge pull request #20699 from pearu:pearu/gammainc ROCm GPU Post-Merge Check #494: Commit 7ac088c pushed by copybara-service bot
March 11, 2025 20:53 17m 17s main
main
March 11, 2025 20:53 17m 17s
[Mosaic GPU] Replace WGMMAFragLayout with TiledLayout in the mlir… ROCm GPU Post-Merge Check #493: Commit 99c9106 pushed by copybara-service bot
March 11, 2025 20:50 10m 39s main
main
March 11, 2025 20:50 10m 39s