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

Filter by Event

Filter by Status

Filter by Branch

Filter by Actor

JEP: pytree preservation of dict key order CI - Bazel CUDA tests (RBE) #129: Pull request #26069 synchronize by jakevdp
January 23, 2025 22:51 9m 37s jakevdp:tree-key-order-jep
jakevdp:tree-key-order-jep
January 23, 2025 22:51 9m 37s
JEP: pytree preservation of dict key order CI - Bazel CUDA tests (RBE) #128: Pull request #26069 synchronize by jakevdp
January 23, 2025 22:46 5m 4s jakevdp:tree-key-order-jep
jakevdp:tree-key-order-jep
January 23, 2025 22:46 5m 4s
JEP: pytree preservation of dict key order CI - Bazel CUDA tests (RBE) #127: Pull request #26069 opened by jakevdp
January 23, 2025 22:44 1m 55s jakevdp:tree-key-order-jep
jakevdp:tree-key-order-jep
January 23, 2025 22:44 1m 55s
[Mosaic TPU] Use vmask pack if possible for mask's bitwidth change and introduce relayout op. CI - Bazel CUDA tests (RBE) #126: Pull request #25938 synchronize by copybara-service bot
January 23, 2025 22:06 9m 23s test_715163607
test_715163607
January 23, 2025 22:06 9m 23s
Update --config=cuda to add direct dependencies on CUDA libraries both for bazel build and bazel test phases. CI - Bazel CUDA tests (RBE) #125: Pull request #26068 opened by copybara-service bot
January 23, 2025 21:52 9m 50s test_719000544
test_719000544
January 23, 2025 21:52 9m 50s
[JAX] Optimize array shard reordering CI - Bazel CUDA tests (RBE) #124: Pull request #26042 synchronize by copybara-service bot
January 23, 2025 21:46 9m 10s test_718482553
test_718482553
January 23, 2025 21:46 9m 10s
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) #123: Pull request #25918 synchronize by copybara-service bot
January 23, 2025 21:42 8m 54s test_716042460
test_716042460
January 23, 2025 21:42 8m 54s
Move transfer python bindings into jax. CI - Bazel CUDA tests (RBE) #122: Pull request #26065 synchronize by copybara-service bot
January 23, 2025 21:29 11m 18s test_718724289
test_718724289
January 23, 2025 21:29 11m 18s
[jax:custom_partitioning] Extend the custom partitioning API to accept a CI - Bazel CUDA tests (RBE) #121: Pull request #25927 synchronize by copybara-service bot
January 23, 2025 21:29 11m 45s test_715961344
test_715961344
January 23, 2025 21:29 11m 45s
Fix import for Windows platforms CI - Bazel CUDA tests (RBE) #120: Pull request #26066 opened by copybara-service bot
January 23, 2025 21:08 6m 47s test_718982994
test_718982994
January 23, 2025 21:08 6m 47s
[jax:custom_partitioning] Extend the custom partitioning API to accept a CI - Bazel CUDA tests (RBE) #119: Pull request #25927 synchronize by copybara-service bot
January 23, 2025 21:06 8m 16s test_715961344
test_715961344
January 23, 2025 21:06 8m 16s
[JAX] Optimize array shard reordering CI - Bazel CUDA tests (RBE) #118: Pull request #26042 synchronize by copybara-service bot
January 23, 2025 21:06 5m 36s test_718482553
test_718482553
January 23, 2025 21:06 5m 36s
[JAX] Optimize array shard reordering CI - Bazel CUDA tests (RBE) #117: Pull request #26042 synchronize by copybara-service bot
January 23, 2025 20:49 9m 55s test_718482553
test_718482553
January 23, 2025 20:49 9m 55s
[Mosaic TPU] Use vmask pack if possible for mask's bitwidth change and introduce relayout op. CI - Bazel CUDA tests (RBE) #116: Pull request #25938 synchronize by copybara-service bot
January 23, 2025 20:41 11m 46s test_715163607
test_715163607
January 23, 2025 20:41 11m 46s
Move transfer python bindings into jax. CI - Bazel CUDA tests (RBE) #115: Pull request #26065 opened by copybara-service bot
January 23, 2025 20:38 6m 33s test_718724289
test_718724289
January 23, 2025 20:38 6m 33s
Add DCE rules for custom_jvp and custom_vjp CI - Bazel CUDA tests (RBE) #114: Pull request #25963 synchronize by dfm
January 23, 2025 20:23 9m 18s dfm:dce-custom-star
dfm:dce-custom-star
January 23, 2025 20:23 9m 18s
Add an experimental interface for customizing DCE behavior. CI - Bazel CUDA tests (RBE) #113: Pull request #25956 synchronize by copybara-service bot
January 23, 2025 19:38 13m 33s test_716596154
test_716596154
January 23, 2025 19:38 13m 33s
Add an experimental interface for customizing DCE behavior. CI - Bazel CUDA tests (RBE) #112: Pull request #25956 synchronize by copybara-service bot
January 23, 2025 19:12 9m 45s test_716596154
test_716596154
January 23, 2025 19:12 9m 45s
[jax:custom_partitioning] Extend the custom partitioning API to accept a CI - Bazel CUDA tests (RBE) #111: Pull request #25927 synchronize by copybara-service bot
January 23, 2025 18:41 16m 55s test_715961344
test_715961344
January 23, 2025 18:41 16m 55s
[jax:custom_partitioning] Extend the custom partitioning API to accept a CI - Bazel CUDA tests (RBE) #110: Pull request #25927 synchronize by copybara-service bot
January 23, 2025 18:29 9m 48s test_715961344
test_715961344
January 23, 2025 18:29 9m 48s
[jax:custom_partitioning] Extend the custom partitioning API to accept a CI - Bazel CUDA tests (RBE) #109: Pull request #25927 synchronize by copybara-service bot
January 23, 2025 18:28 1m 36s test_715961344
test_715961344
January 23, 2025 18:28 1m 36s
[Mosaic GPU] Add basic support for TMA with sub-byte types CI - Bazel CUDA tests (RBE) #108: Pull request #26012 synchronize by copybara-service bot
January 23, 2025 17:44 8m 57s test_717952238
test_717952238
January 23, 2025 17:44 8m 57s
[jax:custom_partitioning] Extend the custom partitioning API to accept a CI - Bazel CUDA tests (RBE) #107: Pull request #25927 synchronize by copybara-service bot
January 23, 2025 17:32 9m 46s test_715961344
test_715961344
January 23, 2025 17:32 9m 46s
[jax:custom_partitioning] Extend the custom partitioning API to accept a CI - Bazel CUDA tests (RBE) #106: Pull request #25927 synchronize by copybara-service bot
January 23, 2025 17:31 1m 3s test_715961344
test_715961344
January 23, 2025 17:31 1m 3s
#sdy unskip JAX Shardy tests that are already passing CI - Bazel CUDA tests (RBE) #105: Pull request #26055 synchronize by copybara-service bot
January 23, 2025 17:26 10m 49s test_718786627
test_718786627
January 23, 2025 17:26 10m 49s
ProTip! You can narrow down the results and go further in time using created:<2025-01-23 or the other filters available.