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

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
Increase the number of iterations in a test that compares rolled vers… ROCm GPU Post-Merge Check #492: Commit 67aa997 pushed by copybara-service bot
March 11, 2025 20:45 5m 23s main
main
March 11, 2025 20:45 5m 23s
Remove installation of NVIDIA wheels for CPU tests ROCm GPU Post-Merge Check #491: Commit e0545a7 pushed by copybara-service bot
March 11, 2025 20:43 9m 14s main
main
March 11, 2025 20:43 9m 14s
Fix the assumption that pages_per_seq is already a multiple of num_kv… ROCm GPU Post-Merge Check #490: Commit eff612a pushed by copybara-service bot
March 11, 2025 19:36 15m 53s main
main
March 11, 2025 19:36 15m 53s
Add NVIDIA wheel requirements only for Linux builds. ROCm GPU Post-Merge Check #489: Commit 0db14aa pushed by copybara-service bot
March 11, 2025 19:34 9m 11s main
main
March 11, 2025 19:34 9m 11s
Add TPU test jobs to the new CI continuous and nightly/release test w… ROCm GPU Post-Merge Check #488: Commit 7ac6355 pushed by copybara-service bot
March 11, 2025 18:42 9m 22s main
main
March 11, 2025 18:42 9m 22s
Merge pull request #27059 from jakevdp:fix-while-loop ROCm GPU Post-Merge Check #487: Commit c2c68c0 pushed by copybara-service bot
March 11, 2025 18:32 9m 9s main
main
March 11, 2025 18:32 9m 9s
Fix syntax error and typos for composite primitive docstring. ROCm GPU Post-Merge Check #486: Commit d191927 pushed by copybara-service bot
March 11, 2025 17:37 11m 38s main
main
March 11, 2025 17:37 11m 38s
Skip ASAN tests for the big Mosaic GPU tests ROCm GPU Post-Merge Check #485: Commit 6f7ce9d pushed by copybara-service bot
March 11, 2025 17:30 9m 24s main
main
March 11, 2025 17:30 9m 24s
[Mosaic GPU] Add support for .cta_group::2 MMA with n=512 on Blackwell ROCm GPU Post-Merge Check #484: Commit 30a9e1b pushed by copybara-service bot
March 11, 2025 16:53 9m 22s main
main
March 11, 2025 16:53 9m 22s
Update :build_jaxlib flag to control whether we should add `py_impo… ROCm GPU Post-Merge Check #483: Commit 1aca76f pushed by copybara-service bot
March 11, 2025 15:31 9m 28s main
main
March 11, 2025 15:31 9m 28s
[Pallas/Mosaic GPU] Explicitly disable ops_test on Mosaic GPU pre-H… ROCm GPU Post-Merge Check #482: Commit 7fd32ec pushed by copybara-service bot
March 11, 2025 14:11 9m 22s main
main
March 11, 2025 14:11 9m 22s
Update XLA dependency to use revision ROCm GPU Post-Merge Check #481: Commit b6da46e pushed by copybara-service bot
March 11, 2025 11:42 9m 16s main
main
March 11, 2025 11:42 9m 16s
PR #22800: Change the default value of print_operand_shape_ to false … ROCm GPU Post-Merge Check #480: Commit cb2eb15 pushed by copybara-service bot
March 11, 2025 10:17 9m 36s main
main
March 11, 2025 10:17 9m 36s
Under pjit the with mesh: context will use use_mesh(mesh): jit in… ROCm GPU Post-Merge Check #479: Commit 76dec38 pushed by copybara-service bot
March 11, 2025 03:21 9m 28s main
main
March 11, 2025 03:21 9m 28s
[Pallas TPU] Remove next_slot SMEM tensor from pipeline emitter ROCm GPU Post-Merge Check #478: Commit 02505fa pushed by copybara-service bot
March 11, 2025 00:19 9m 18s main
main
March 11, 2025 00:19 9m 18s
Better error message when raise_if_error() is called within a trace… ROCm GPU Post-Merge Check #477: Commit 988a120 pushed by copybara-service bot
March 10, 2025 23:55 9m 11s main
main
March 10, 2025 23:55 9m 11s
[Pallas] Enable skipping of floating-point operations when interpreti… ROCm GPU Post-Merge Check #476: Commit aceae84 pushed by copybara-service bot
March 10, 2025 22:14 9m 0s main
main
March 10, 2025 22:14 9m 0s
[Pallas] Increase tolerance in PallasOutOfBoundsInterpretTest. ROCm GPU Post-Merge Check #475: Commit 802cb33 pushed by copybara-service bot
March 10, 2025 21:49 9m 25s main
main
March 10, 2025 21:49 9m 25s
Merge pull request #26840 from rajasekharporeddy:testbranch1 ROCm GPU Post-Merge Check #474: Commit d558797 pushed by copybara-service bot
March 10, 2025 21:33 10m 6s main
main
March 10, 2025 21:33 10m 6s
Merge pull request #26839 from Sai-Suraj-27:fix_jax.debug.print ROCm GPU Post-Merge Check #473: Commit b859081 pushed by copybara-service bot
March 10, 2025 21:26 6m 30s main
main
March 10, 2025 21:26 6m 30s
[Pallas/Fuser] Add select_n push rule ROCm GPU Post-Merge Check #472: Commit 81dde22 pushed by copybara-service bot
March 10, 2025 21:23 11m 25s main
main
March 10, 2025 21:23 11m 25s
Merge pull request #27038 from jakevdp:vmap-sentinel ROCm GPU Post-Merge Check #471: Commit 261e6e5 pushed by copybara-service bot
March 10, 2025 21:21 1m 51s main
main
March 10, 2025 21:21 1m 51s