Skip to content

Actions: jax-ml/jax

Actions

CI - Bazel CUDA tests (RBE)

Actions

Loading...
Loading

Show workflow options

Create status badge

Loading
bazel_cuda_rbe.yml
2,028 workflow runs
2,028 workflow runs

Filter by Event

Filter by Status

Filter by Branch

Filter by Actor

Re-enable Pallas bf16 exp2 tests on TPU CI - Bazel CUDA tests (RBE) #354: Pull request #26146 opened by copybara-service bot
January 28, 2025 10:27 9m 53s test_720493419
test_720493419
January 28, 2025 10:27 9m 53s
[sharding_in_types] Add clamp_p sharding rule. CI - Bazel CUDA tests (RBE) #353: Pull request #26142 synchronize by copybara-service bot
January 28, 2025 05:45 6m 10s test_720421161
test_720421161
January 28, 2025 05:45 6m 10s
[sharding_in_types] Add clamp_p sharding rule. CI - Bazel CUDA tests (RBE) #352: Pull request #26142 opened by copybara-service bot
January 28, 2025 05:27 8m 18s test_720421161
test_720421161
January 28, 2025 05:27 8m 18s
[sharding_in_types] Add support for svd_p CI - Bazel CUDA tests (RBE) #351: Pull request #26141 synchronize by copybara-service bot
January 28, 2025 04:32 6m 33s test_720368073
test_720368073
January 28, 2025 04:32 6m 33s
[sharding_in_types] Add support for svd_p CI - Bazel CUDA tests (RBE) #350: Pull request #26141 synchronize by copybara-service bot
January 28, 2025 04:04 9m 41s test_720368073
test_720368073
January 28, 2025 04:04 9m 41s
[sharding_in_types] Add support for svd_p CI - Bazel CUDA tests (RBE) #349: Pull request #26141 synchronize by copybara-service bot
January 28, 2025 03:56 8m 20s test_720368073
test_720368073
January 28, 2025 03:56 8m 20s
[sharding_in_types] Add support for svd_p CI - Bazel CUDA tests (RBE) #348: Pull request #26141 synchronize by copybara-service bot
January 28, 2025 02:23 10m 0s test_720368073
test_720368073
January 28, 2025 02:23 10m 0s
[sharding_in_types] Add support for svd_p CI - Bazel CUDA tests (RBE) #347: Pull request #26141 opened by copybara-service bot
January 28, 2025 02:22 1m 3s test_720368073
test_720368073
January 28, 2025 02:22 1m 3s
Adds a minimal but viable implementation of string arrays (with numpy.dtypes.StringDType) in JAX. Currently this only supports making of a string array by means of either jax.numpy.asarray or jax.device_put and reading it back with jax.device_get. CI - Bazel CUDA tests (RBE) #346: Pull request #25918 synchronize by copybara-service bot
January 28, 2025 02:16 7m 58s test_716042460
test_716042460
January 28, 2025 02:16 7m 58s
[Pallas TPU] Support jnp.take_along_axis for 32-bit vreg-sized vector. CI - Bazel CUDA tests (RBE) #345: Pull request #26140 opened by copybara-service bot
January 28, 2025 02:15 12m 46s test_720348801
test_720348801
January 28, 2025 02:15 12m 46s
[Pallas] Refactor Pallas HLO interpret mode to a standalone file. CI - Bazel CUDA tests (RBE) #344: Pull request #26136 synchronize by copybara-service bot
January 28, 2025 01:52 5m 13s test_713419621
test_713419621
January 28, 2025 01:52 5m 13s
[Pallas] Refactor Pallas HLO interpret mode to a standalone file. CI - Bazel CUDA tests (RBE) #343: Pull request #26136 synchronize by copybara-service bot
January 28, 2025 01:38 5m 50s test_713419621
test_713419621
January 28, 2025 01:38 5m 50s
Add win32 guard to fix imports on Windows CI - Bazel CUDA tests (RBE) #342: Pull request #26138 synchronize by copybara-service bot
January 28, 2025 01:28 6m 28s test_720339950
test_720339950
January 28, 2025 01:28 6m 28s
Add win32 guard to fix imports on Windows CI - Bazel CUDA tests (RBE) #341: Pull request #26138 synchronize by copybara-service bot
January 28, 2025 01:13 5m 9s test_720339950
test_720339950
January 28, 2025 01:13 5m 9s
Add win32 guard to fix imports on Windows CI - Bazel CUDA tests (RBE) #340: Pull request #26138 synchronize by copybara-service bot
January 28, 2025 00:53 5m 10s test_720339950
test_720339950
January 28, 2025 00:53 5m 10s
Add a retry logic to avoid flakes when downloading Bazel CI - Bazel CUDA tests (RBE) #339: Pull request #26139 synchronize by copybara-service bot
January 28, 2025 00:36 5m 35s test_720341312
test_720341312
January 28, 2025 00:36 5m 35s
[Pallas] Refactor Pallas HLO interpret mode to a standalone file. CI - Bazel CUDA tests (RBE) #338: Pull request #26136 synchronize by copybara-service bot
January 28, 2025 00:33 6m 34s test_713419621
test_713419621
January 28, 2025 00:33 6m 34s
[Pallas] Refactor Pallas HLO interpret mode to a standalone file. CI - Bazel CUDA tests (RBE) #337: Pull request #26136 synchronize by copybara-service bot
January 28, 2025 00:32 1m 5s test_713419621
test_713419621
January 28, 2025 00:32 1m 5s
Add a retry logic to avoid flakes when downloading Bazel CI - Bazel CUDA tests (RBE) #336: Pull request #26139 synchronize by copybara-service bot
January 28, 2025 00:21 6m 36s test_720341312
test_720341312
January 28, 2025 00:21 6m 36s
Fix import for Windows platforms CI - Bazel CUDA tests (RBE) #335: Pull request #26066 synchronize by copybara-service bot
January 28, 2025 00:19 7m 19s test_718982994
test_718982994
January 28, 2025 00:19 7m 19s
Add a retry logic to avoid flakes when downloading Bazel CI - Bazel CUDA tests (RBE) #334: Pull request #26139 synchronize by copybara-service bot
January 28, 2025 00:19 2m 10s test_720341312
test_720341312
January 28, 2025 00:19 2m 10s
Disable pytorch_interoperability_test under asan on all backends. CI - Bazel CUDA tests (RBE) #333: Pull request #26137 synchronize by copybara-service bot
January 28, 2025 00:18 5m 20s test_720339169
test_720339169
January 28, 2025 00:18 5m 20s
Support dynamic masks in splash attention CI - Bazel CUDA tests (RBE) #332: Pull request #25213 synchronize by Rifur13
January 28, 2025 00:15 7m 31s Rifur13:dynamic_mask
Rifur13:dynamic_mask
January 28, 2025 00:15 7m 31s
Add a retry logic to avoid flakes when downloading Bazel CI - Bazel CUDA tests (RBE) #331: Pull request #26139 opened by copybara-service bot
January 28, 2025 00:09 6m 36s test_720341312
test_720341312
January 28, 2025 00:09 6m 36s
Disable pytorch_interoperability_test under asan on all backends. CI - Bazel CUDA tests (RBE) #330: Pull request #26137 synchronize by copybara-service bot
January 28, 2025 00:06 6m 35s test_720339169
test_720339169
January 28, 2025 00:06 6m 35s
ProTip! You can narrow down the results and go further in time using created:<2025-01-28 or the other filters available.