Skip to content

Actions: jax-ml/jax

Actions

CI - Bazel GPU tests (RBE)

Actions

Loading...
Loading

Show workflow options

Create status badge

Loading
bazel_gpu_rbe.yml
1,111 workflow runs
1,111 workflow runs

Filter by Event

Filter by Status

Filter by Branch

Filter by Actor

[Mosaic TPU][NFC] Remove redundant num_subelems attribute from CreateSubelementMaskOp CI - Bazel GPU tests (RBE) #1113: Pull request #25729 synchronize by copybara-service bot
January 6, 2025 07:20 9m 35s test_712408747
test_712408747
January 6, 2025 07:20 9m 35s
[Mosaic TPU][NFC] Remove redundant num_subelems attribute from CreateSubelementMaskOp CI - Bazel GPU tests (RBE) #1112: Pull request #25729 synchronize by copybara-service bot
January 6, 2025 07:19 1m 6s test_712408747
test_712408747
January 6, 2025 07:19 1m 6s
[Mosaic TPU] Enable non-sublane-aligned 2D int8 load/stores CI - Bazel GPU tests (RBE) #1111: Pull request #25727 synchronize by copybara-service bot
January 6, 2025 07:17 9m 51s test_712402456
test_712402456
January 6, 2025 07:17 9m 51s
[Mosaic TPU][NFC] Remove redundant num_subelems attribute from CreateSubelementMaskOp CI - Bazel GPU tests (RBE) #1110: Pull request #25729 opened by copybara-service bot
January 6, 2025 07:17 2m 3s test_712408747
test_712408747
January 6, 2025 07:17 2m 3s
Fix the doc error: module 'scipy.misc' has no attribute 'face' CI - Bazel GPU tests (RBE) #1109: Pull request #25728 opened by zhenying-liu
January 6, 2025 07:05 Action required zhenying-liu:scipy.misc
zhenying-liu:scipy.misc
January 6, 2025 07:05 Action required
[Mosaic TPU] Enable non-sublane-aligned 2D int8 load/stores CI - Bazel GPU tests (RBE) #1108: Pull request #25727 synchronize by copybara-service bot
January 6, 2025 07:00 5m 27s test_712402456
test_712402456
January 6, 2025 07:00 5m 27s
[Mosaic TPU] Enable non-sublane-aligned 2D int8 load/stores CI - Bazel GPU tests (RBE) #1107: Pull request #25727 opened by copybara-service bot
January 6, 2025 06:51 8m 48s test_712402456
test_712402456
January 6, 2025 06:51 8m 48s
Fixes tril/triu comments (they were flipped) CI - Bazel GPU tests (RBE) #1106: Pull request #25723 synchronize by copybara-service bot
January 6, 2025 06:41 9m 5s test_711889610
test_711889610
January 6, 2025 06:41 9m 5s
[Mosaic TPU] Enable unaligned bf16 2D load/stores for earlier TPU gens CI - Bazel GPU tests (RBE) #1105: Pull request #25726 opened by copybara-service bot
January 6, 2025 06:20 16m 21s test_712397610
test_712397610
January 6, 2025 06:20 16m 21s
Add jax.random.multinomial. CI - Bazel GPU tests (RBE) #1104: Pull request #25688 synchronize by carlosgmartin
January 6, 2025 05:05 Action required carlosgmartin:random_multinomial
carlosgmartin:random_multinomial
January 6, 2025 05:05 Action required
Automated Code Change CI - Bazel GPU tests (RBE) #1103: Pull request #25725 opened by copybara-service bot
January 6, 2025 04:12 4m 38s test_712350110
test_712350110
January 6, 2025 04:12 4m 38s
Implement column-pivoted QR via geqp3 (CPU lowering only) CI - Bazel GPU tests (RBE) #1102: Pull request #20282 synchronize by tttc3
January 5, 2025 11:51 Action required tttc3:pivoted-qr
tttc3:pivoted-qr
January 5, 2025 11:51 Action required
Add SMEM as a supported Pallas output memory space. CI - Bazel GPU tests (RBE) #1101: Pull request #25689 synchronize by copybara-service bot
January 5, 2025 03:33 6m 53s test_707794510
test_707794510
January 5, 2025 03:33 6m 53s
Add SMEM as a supported Pallas output memory space. CI - Bazel GPU tests (RBE) #1100: Pull request #25689 synchronize by copybara-service bot
January 5, 2025 03:32 1m 20s test_707794510
test_707794510
January 5, 2025 03:32 1m 20s
Add SMEM as a supported Pallas output memory space. CI - Bazel GPU tests (RBE) #1099: Pull request #25689 synchronize by copybara-service bot
January 5, 2025 01:30 18m 5s test_707794510
test_707794510
January 5, 2025 01:30 18m 5s
Add examples for jax.checkpoint activation offloading policies CI - Bazel GPU tests (RBE) #1098: Pull request #25594 synchronize by zhenying-liu
January 5, 2025 00:36 Action required zhenying-liu:activation-offloading-doc
zhenying-liu:activation-offloading-doc
January 5, 2025 00:36 Action required
Fixes tril/triu comments (they were flipped) CI - Bazel GPU tests (RBE) #1097: Pull request #25723 synchronize by copybara-service bot
January 4, 2025 15:39 9m 59s test_711889610
test_711889610
January 4, 2025 15:39 9m 59s
Fixes tril/triu comments (they were flipped) CI - Bazel GPU tests (RBE) #1096: Pull request #25723 synchronize by copybara-service bot
January 4, 2025 05:47 9m 27s test_711889610
test_711889610
January 4, 2025 05:47 9m 27s
Fixes tril/triu comments (they were flipped) CI - Bazel GPU tests (RBE) #1095: Pull request #25723 opened by copybara-service bot
January 4, 2025 01:03 10m 29s test_711889610
test_711889610
January 4, 2025 01:03 10m 29s
[Pallas] Add empty/empty_like helper functions CI - Bazel GPU tests (RBE) #1094: Pull request #25722 opened by copybara-service bot
January 3, 2025 23:57 7m 42s test_711874730
test_711874730
January 3, 2025 23:57 7m 42s
[Pallas] Improvements to core_map CI - Bazel GPU tests (RBE) #1093: Pull request #25721 opened by copybara-service bot
January 3, 2025 23:45 13m 20s test_711864273
test_711864273
January 3, 2025 23:45 13m 20s
[Mosaic TPU] Add some elementwise canonicalizations CI - Bazel GPU tests (RBE) #1092: Pull request #25720 opened by copybara-service bot
January 3, 2025 23:31 15m 17s test_711869914
test_711869914
January 3, 2025 23:31 15m 17s
[Mosaic] Create a stub for TPUExtDialect. CI - Bazel GPU tests (RBE) #1091: Pull request #25719 opened by copybara-service bot
January 3, 2025 21:59 5m 22s test_711533286
test_711533286
January 3, 2025 21:59 5m 22s
Disable avxvnniint8 when building with Clang version < 19, or GCC < 13. CI - Bazel GPU tests (RBE) #1090: Pull request #25712 synchronize by copybara-service bot
January 3, 2025 20:05 6m 35s test_711455380
test_711455380
January 3, 2025 20:05 6m 35s
Disable avxvnniint8 when building with Clang version < 19, or GCC < 13. CI - Bazel GPU tests (RBE) #1089: Pull request #25712 synchronize by copybara-service bot
January 3, 2025 19:55 8m 32s test_711455380
test_711455380
January 3, 2025 19:55 8m 32s