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,261 workflow runs
1,261 workflow runs

Filter by Event

Filter by Status

Filter by Branch

Filter by Actor

[sharding_in_types] Aval sharding under full auto mode should contain None and not UNCONSTRAINED because axis_types + pspec give the full picture. CI - Bazel GPU tests (RBE) #1239: Pull request #25764 opened by copybara-service bot
January 7, 2025 23:48 8m 20s test_713066768
test_713066768
January 7, 2025 23:48 8m 20s
Increase the minimum SciPy version to 1.11.1. CI - Bazel GPU tests (RBE) #1238: Pull request #25744 synchronize by copybara-service bot
January 7, 2025 23:48 9m 31s test_712614146
test_712614146
January 7, 2025 23:48 9m 31s
[Pallas] Improvements to core_map CI - Bazel GPU tests (RBE) #1237: Pull request #25721 synchronize by copybara-service bot
January 7, 2025 23:35 9m 58s test_711864273
test_711864273
January 7, 2025 23:35 9m 58s
[Mosaic] Fix inferMemRefLayout to error out if shape not aligned to tiling. CI - Bazel GPU tests (RBE) #1236: Pull request #25758 synchronize by copybara-service bot
January 7, 2025 23:31 16m 9s test_712999303
test_712999303
January 7, 2025 23:31 16m 9s
[Mosaic] Create a stub for TPUExtDialect. CI - Bazel GPU tests (RBE) #1235: Pull request #25719 synchronize by copybara-service bot
January 7, 2025 23:27 23m 30s test_711533286
test_711533286
January 7, 2025 23:27 23m 30s
[Pallas] Improvements to core_map CI - Bazel GPU tests (RBE) #1234: Pull request #25721 synchronize by copybara-service bot
January 7, 2025 23:02 10m 22s test_711864273
test_711864273
January 7, 2025 23:02 10m 22s
[Mosaic] Create a stub for TPUExtDialect. CI - Bazel GPU tests (RBE) #1233: Pull request #25719 synchronize by copybara-service bot
January 7, 2025 22:56 7m 36s test_711533286
test_711533286
January 7, 2025 22:56 7m 36s
[Mosaic GPU] Allow multiple gmem indexers on copies. CI - Bazel GPU tests (RBE) #1232: Pull request #25763 opened by copybara-service bot
January 7, 2025 22:45 7m 9s test_713029565
test_713029565
January 7, 2025 22:45 7m 9s
In progress experimention. Add StringDType to JAX's supported types. CI - Bazel GPU tests (RBE) #1231: Pull request #25592 synchronize by copybara-service bot
January 7, 2025 22:44 10m 14s test_707662268
test_707662268
January 7, 2025 22:44 10m 14s
[Mosaic GPU] Enable loop carries in the pipeline emitter. CI - Bazel GPU tests (RBE) #1230: Pull request #25762 opened by copybara-service bot
January 7, 2025 22:44 6m 39s test_713029567
test_713029567
January 7, 2025 22:44 6m 39s
[array api] update test suite to latest commit CI - Bazel GPU tests (RBE) #1229: Pull request #25761 opened by jakevdp
January 7, 2025 21:58 13m 1s jakevdp:array-api-update
jakevdp:array-api-update
January 7, 2025 21:58 13m 1s
In progress experimention. Add StringDType to JAX's supported types. CI - Bazel GPU tests (RBE) #1228: Pull request #25592 synchronize by copybara-service bot
January 7, 2025 21:36 11m 51s test_707662268
test_707662268
January 7, 2025 21:36 11m 51s
[Mosaic] Fix inferMemRefLayout to error out if shape not aligned to tiling. CI - Bazel GPU tests (RBE) #1227: Pull request #25758 opened by copybara-service bot
January 7, 2025 20:19 10m 28s test_712999303
test_712999303
January 7, 2025 20:19 10m 28s
Introduce jax.shard_map, without requiring mesh arg CI - Bazel GPU tests (RBE) #1226: Pull request #25757 opened by copybara-service bot
January 7, 2025 20:06 9m 8s test_712995474
test_712995474
January 7, 2025 20:06 9m 8s
[Mosaic] Create a stub for TPUExtDialect. CI - Bazel GPU tests (RBE) #1225: Pull request #25719 synchronize by copybara-service bot
January 7, 2025 20:02 8m 46s test_711533286
test_711533286
January 7, 2025 20:02 8m 46s
[Mosaic] Create a stub for TPUExtDialect. CI - Bazel GPU tests (RBE) #1224: Pull request #25719 synchronize by copybara-service bot
January 7, 2025 19:29 8m 35s test_711533286
test_711533286
January 7, 2025 19:29 8m 35s
In progress experimention. Add StringDType to JAX's supported types. CI - Bazel GPU tests (RBE) #1223: Pull request #25592 synchronize by copybara-service bot
January 7, 2025 19:21 9m 11s test_707662268
test_707662268
January 7, 2025 19:21 9m 11s
Sort manual axes when lowering jax.shard_map to sdy.manual_computation, which ensures the determinism in the generated sdy.manual_computation. CI - Bazel GPU tests (RBE) #1222: Pull request #25742 synchronize by copybara-service bot
January 7, 2025 19:03 6m 26s test_712664646
test_712664646
January 7, 2025 19:03 6m 26s
[Mosaic] Create a stub for TPUExtDialect. CI - Bazel GPU tests (RBE) #1221: Pull request #25719 synchronize by copybara-service bot
January 7, 2025 18:51 10m 46s test_711533286
test_711533286
January 7, 2025 18:51 10m 46s
Sort manual axes when lowering jax.shard_map to sdy.manual_computation, which ensures the determinism in the generated sdy.manual_computation. CI - Bazel GPU tests (RBE) #1220: Pull request #25742 synchronize by copybara-service bot
January 7, 2025 18:45 14m 13s test_712664646
test_712664646
January 7, 2025 18:45 14m 13s
Add an example demonstrating input-output aliasing with the FFI CI - Bazel GPU tests (RBE) #1219: Pull request #25042 synchronize by dfm
January 7, 2025 18:22 11m 41s dfm:ffi-example-input-output-alias
dfm:ffi-example-input-output-alias
January 7, 2025 18:22 11m 41s
[ROCm] Implement RNN support CI - Bazel GPU tests (RBE) #1218: Pull request #25755 opened by Ruturaj4
January 7, 2025 17:47 3m 27s ROCm:ci_rnn_final-upstream
ROCm:ci_rnn_final-upstream
January 7, 2025 17:47 3m 27s
[Pallas] Fix pallas_call lowering mutating compiler params during Triton lowering. CI - Bazel GPU tests (RBE) #1217: Pull request #25735 synchronize by copybara-service bot
January 7, 2025 17:01 7m 24s test_712566709
test_712566709
January 7, 2025 17:01 7m 24s
[AutoPGLE] FIx PGLE kokoro test failures. CI - Bazel GPU tests (RBE) #1216: Pull request #25746 synchronize by copybara-service bot
January 7, 2025 17:00 9m 14s test_712777161
test_712777161
January 7, 2025 17:00 9m 14s
[Pallas] Fix pallas_call lowering mutating compiler params during Triton lowering. CI - Bazel GPU tests (RBE) #1215: Pull request #25735 synchronize by copybara-service bot
January 7, 2025 16:46 7m 45s test_712566709
test_712566709
January 7, 2025 16:46 7m 45s