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,201 workflow runs
2,201 workflow runs

Filter by Event

Filter by Status

Filter by Branch

Filter by Actor

Replace Hidden/Visible/Collective AxisTypes names with Auto/Explicit/Manual. CI - Bazel CUDA tests (RBE) #201: Pull request #26091 synchronize by copybara-service bot
January 25, 2025 06:33 6m 34s test_719497512
test_719497512
January 25, 2025 06:33 6m 34s
Replace Hidden/Visible/Collective AxisTypes names with Auto/Explicit/Manual. CI - Bazel CUDA tests (RBE) #200: Pull request #26091 synchronize by copybara-service bot
January 25, 2025 02:13 9m 32s test_719497512
test_719497512
January 25, 2025 02:13 9m 32s
Replace Hidden/Visible/Collective AxisTypes names with Auto/Explicit/Manual. CI - Bazel CUDA tests (RBE) #199: Pull request #26091 synchronize by copybara-service bot
January 25, 2025 02:12 1m 2s test_719497512
test_719497512
January 25, 2025 02:12 1m 2s
Replace Hidden/Visible/Collective AxisTypes names with Auto/Explicit/Manual. CI - Bazel CUDA tests (RBE) #198: Pull request #26091 opened by copybara-service bot
January 25, 2025 02:11 1m 23s test_719497512
test_719497512
January 25, 2025 02:11 1m 23s
Add support for axis_name and axis_index_groups to lax.ragged_all_to_all CI - Bazel CUDA tests (RBE) #197: Pull request #25660 synchronize by copybara-service bot
January 25, 2025 02:01 10m 51s test_708938150
test_708938150
January 25, 2025 02:01 10m 51s
Reject invalid None in jax.NamedSharding(spec=None). CI - Bazel CUDA tests (RBE) #196: Pull request #26090 synchronize by copybara-service bot
January 25, 2025 00:07 15m 36s test_719446269
test_719446269
January 25, 2025 00:07 15m 36s
Reject invalid None in jax.NamedSharding(spec=None). CI - Bazel CUDA tests (RBE) #195: Pull request #26090 synchronize by copybara-service bot
January 24, 2025 23:15 8m 27s test_719446269
test_719446269
January 24, 2025 23:15 8m 27s
Reject invalid None in jax.NamedSharding(spec=None). CI - Bazel CUDA tests (RBE) #194: Pull request #26090 opened by copybara-service bot
January 24, 2025 23:09 6m 41s test_719446269
test_719446269
January 24, 2025 23:09 6m 41s
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) #193: Pull request #25918 synchronize by copybara-service bot
January 24, 2025 22:35 7m 1s test_716042460
test_716042460
January 24, 2025 22:35 7m 1s
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) #192: Pull request #25918 synchronize by copybara-service bot
January 24, 2025 22:27 8m 14s test_716042460
test_716042460
January 24, 2025 22:27 8m 14s
Optimize the set_xla_metadata context manager. CI - Bazel CUDA tests (RBE) #191: Pull request #26089 synchronize by copybara-service bot
January 24, 2025 21:32 10m 1s test_719413455
test_719413455
January 24, 2025 21:32 10m 1s
Optimize the set_xla_metadata context manager. CI - Bazel CUDA tests (RBE) #190: Pull request #26089 opened by copybara-service bot
January 24, 2025 21:31 1m 13s test_719413455
test_719413455
January 24, 2025 21:31 1m 13s
Implement lax.full_like in terms of a primitive CI - Bazel CUDA tests (RBE) #189: Pull request #26045 synchronize by jakevdp
January 24, 2025 21:28 13m 12s jakevdp:full-like-p
jakevdp:full-like-p
January 24, 2025 21:28 13m 12s
Add new Bazel remote cache configs CI - Bazel CUDA tests (RBE) #188: Pull request #26070 synchronize by copybara-service bot
January 24, 2025 20:28 10m 51s test_716019045
test_716019045
January 24, 2025 20:28 10m 51s
Implement lax.full_like in terms of a primitive CI - Bazel CUDA tests (RBE) #187: Pull request #26045 synchronize by jakevdp
January 24, 2025 19:39 12m 32s jakevdp:full-like-p
jakevdp:full-like-p
January 24, 2025 19:39 12m 32s
Update JVP rule for abs to fix behavior for complex infinite inputs CI - Bazel CUDA tests (RBE) #186: Pull request #26086 synchronize by dfm
January 24, 2025 19:32 10m 45s dfm:abs-complex-grad
dfm:abs-complex-grad
January 24, 2025 19:32 10m 45s
Use backend._get_all_devices() to validate devices. CI - Bazel CUDA tests (RBE) #185: Pull request #26025 synchronize by copybara-service bot
January 24, 2025 19:09 9m 28s test_716346724
test_716346724
January 24, 2025 19:09 9m 28s
Don't computing forwarding information if we're going to inline. CI - Bazel CUDA tests (RBE) #184: Pull request #26085 synchronize by copybara-service bot
January 24, 2025 19:07 13m 56s test_719321640
test_719321640
January 24, 2025 19:07 13m 56s
Don't computing forwarding information if we're going to inline. CI - Bazel CUDA tests (RBE) #183: Pull request #26085 synchronize by copybara-service bot
January 24, 2025 18:52 9m 34s test_719321640
test_719321640
January 24, 2025 18:52 9m 34s
Use backend._get_all_devices() to validate devices. CI - Bazel CUDA tests (RBE) #182: Pull request #26025 synchronize by copybara-service bot
January 24, 2025 18:46 11m 18s test_716346724
test_716346724
January 24, 2025 18:46 11m 18s
Implement lax.full_like in terms of a primitive CI - Bazel CUDA tests (RBE) #181: Pull request #26045 synchronize by jakevdp
January 24, 2025 18:39 14m 37s jakevdp:full-like-p
jakevdp:full-like-p
January 24, 2025 18:39 14m 37s
Update JVP rule for abs to fix behavior for complex infinite inputs CI - Bazel CUDA tests (RBE) #180: Pull request #26086 opened by dfm
January 24, 2025 18:06 10m 4s dfm:abs-complex-grad
dfm:abs-complex-grad
January 24, 2025 18:06 10m 4s
docs: outdated link for index update syntax CI - Bazel CUDA tests (RBE) #179: Pull request #26082 synchronize by Saransh-cpp
January 24, 2025 17:36 8m 37s Saransh-cpp:index-update-syntax-link
Saransh-cpp:index-update-syntax-link
January 24, 2025 17:36 8m 37s
Adding GPU paged attention kernel CI - Bazel CUDA tests (RBE) #178: Pull request #25839 synchronize by Rifur13
January 24, 2025 17:15 9m 38s Rifur13:paged
Rifur13:paged
January 24, 2025 17:15 9m 38s
Don't computing forwarding information if we're going to inline. CI - Bazel CUDA tests (RBE) #177: Pull request #26085 opened by copybara-service bot
January 24, 2025 17:04 14m 33s test_719321640
test_719321640
January 24, 2025 17:04 14m 33s
ProTip! You can narrow down the results and go further in time using created:<2025-01-24 or the other filters available.