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

Filter by Event

Filter by Status

Filter by Branch

Filter by Actor

Do not check device_count constraint for requested users. CI - Bazel GPU tests (RBE) #2162: Pull request #26025 opened by copybara-service bot
January 22, 2025 00:36 8m 5s test_716346724
test_716346724
January 22, 2025 00:36 8m 5s
Added jax.experimental.multihost_utils.live_devices API. CI - Bazel GPU tests (RBE) #2161: Pull request #25291 synchronize by copybara-service bot
January 22, 2025 00:28 9m 39s test_703178517
test_703178517
January 22, 2025 00:28 9m 39s
Remove device_context from trace_context because we don't need it there. We can get compilation cache misses (and tracing/lowering cache hit) naturally without putting concrete devices into trace_context. CI - Bazel GPU tests (RBE) #2160: Pull request #26023 synchronize by copybara-service bot
January 22, 2025 00:21 5m 25s test_717320942
test_717320942
January 22, 2025 00:21 5m 25s
[Pallas TPU] Remove obsolete skip condition CI - Bazel GPU tests (RBE) #2159: Pull request #25910 synchronize by copybara-service bot
January 22, 2025 00:15 5m 30s test_715853352
test_715853352
January 22, 2025 00:15 5m 30s
[xla:cpu] FFI: Implement Jax callbacks for CPU/GPU devices using XLA FFI. CI - Bazel GPU tests (RBE) #2158: Pull request #25815 synchronize by copybara-service bot
January 22, 2025 00:11 10m 52s test_713295376
test_713295376
January 22, 2025 00:11 10m 52s
[xla:cpu] FFI: Implement Jax callbacks for CPU/GPU devices using XLA FFI. CI - Bazel GPU tests (RBE) #2157: Pull request #25815 synchronize by copybara-service bot
January 22, 2025 00:10 1m 34s test_713295376
test_713295376
January 22, 2025 00:10 1m 34s
jax.numpy setops: use ensure_arraylike & avoid asarray CI - Bazel GPU tests (RBE) #2156: Pull request #26024 opened by jakevdp
January 22, 2025 00:06 11m 28s jakevdp:setops-args
jakevdp:setops-args
January 22, 2025 00:06 11m 28s
Remove device_context from trace_context because we don't need it there. We can get compilation cache misses (and tracing/lowering cache hit) naturally without putting concrete devices into trace_context. CI - Bazel GPU tests (RBE) #2155: Pull request #26023 synchronize by copybara-service bot
January 22, 2025 00:06 5m 13s test_717320942
test_717320942
January 22, 2025 00:06 5m 13s
Added jax.experimental.multihost_utils.live_devices API. CI - Bazel GPU tests (RBE) #2154: Pull request #25291 synchronize by copybara-service bot
January 21, 2025 23:59 5m 15s test_703178517
test_703178517
January 21, 2025 23:59 5m 15s
Remove device_context from trace_context because we don't need it there. We can get compilation cache misses (and tracing/lowering cache hit) naturally without putting concrete devices into trace_context. CI - Bazel GPU tests (RBE) #2153: Pull request #26023 opened by copybara-service bot
January 21, 2025 23:43 9m 25s test_717320942
test_717320942
January 21, 2025 23:43 9m 25s
Fix cache init when JAX Array is created early (#25768) CI - Bazel GPU tests (RBE) #2152: Pull request #25889 synchronize by Stella-S-Yan
January 21, 2025 23:40 9m 12s Stella-S-Yan:cache_reset
Stella-S-Yan:cache_reset
January 21, 2025 23:40 9m 12s
Added jax.experimental.multihost_utils.live_devices API. CI - Bazel GPU tests (RBE) #2151: Pull request #25291 synchronize by copybara-service bot
January 21, 2025 23:07 8m 16s test_703178517
test_703178517
January 21, 2025 23:07 8m 16s
Error out if contracting dimensions are sharded and ask the user to provide the output sharding CI - Bazel GPU tests (RBE) #2150: Pull request #26018 synchronize by copybara-service bot
January 21, 2025 23:07 7m 10s test_718048897
test_718048897
January 21, 2025 23:07 7m 10s
Added jax.experimental.multihost_utils.live_devices API. CI - Bazel GPU tests (RBE) #2149: Pull request #25291 synchronize by copybara-service bot
January 21, 2025 23:06 1m 33s test_703178517
test_703178517
January 21, 2025 23:06 1m 33s
Error out if contracting dimensions are sharded and ask the user to provide the output sharding CI - Bazel GPU tests (RBE) #2148: Pull request #26018 synchronize by copybara-service bot
January 21, 2025 22:53 5m 23s test_718048897
test_718048897
January 21, 2025 22:53 5m 23s
jax.tree.map: preserve dict key order CI - Bazel GPU tests (RBE) #2147: Pull request #26017 synchronize by copybara-service bot
January 21, 2025 22:45 6m 54s test_718051012
test_718051012
January 21, 2025 22:45 6m 54s
[Mosaic TPU] Emulate converting x16 vector to mask if mask packing is supported. CI - Bazel GPU tests (RBE) #2146: Pull request #25937 synchronize by copybara-service bot
January 21, 2025 22:27 5m 4s test_716395352
test_716395352
January 21, 2025 22:27 5m 4s
Refactor JAX wheel build rules to control the wheel filename and maintain reproducible wheel content and filename results. CI - Bazel GPU tests (RBE) #2145: Pull request #25126 synchronize by copybara-service bot
January 21, 2025 22:12 1m 12s test_699315679
test_699315679
January 21, 2025 22:12 1m 12s
Refactor JAX wheel build rules to control the wheel filename and maintain reproducible wheel content and filename results. CI - Bazel GPU tests (RBE) #2144: Pull request #25126 synchronize by copybara-service bot
January 21, 2025 22:08 1m 8s test_699315679
test_699315679
January 21, 2025 22:08 1m 8s
Error out if contracting dimensions are sharded and ask the user to provide the output sharding CI - Bazel GPU tests (RBE) #2143: Pull request #26018 opened by copybara-service bot
January 21, 2025 22:07 8m 18s test_718048897
test_718048897
January 21, 2025 22:07 8m 18s
jax.tree.map: preserve dict key order CI - Bazel GPU tests (RBE) #2142: Pull request #26017 synchronize by copybara-service bot
January 21, 2025 22:03 9m 19s test_718051012
test_718051012
January 21, 2025 22:03 9m 19s
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 GPU tests (RBE) #2141: Pull request #25918 synchronize by copybara-service bot
January 21, 2025 21:47 11m 1s test_716042460
test_716042460
January 21, 2025 21:47 11m 1s
jax.tree.map: preserve dict key order CI - Bazel GPU tests (RBE) #2140: Pull request #26017 opened by copybara-service bot
January 21, 2025 21:44 9m 20s test_718051012
test_718051012
January 21, 2025 21:44 9m 20s
Add support for constants in the decomposition of lax.composites. CI - Bazel GPU tests (RBE) #2139: Pull request #25921 synchronize by copybara-service bot
January 21, 2025 21:11 10m 37s test_716148466
test_716148466
January 21, 2025 21:11 10m 37s
Remove unnecessary use of xla_client.OpMetadata class. CI - Bazel GPU tests (RBE) #2138: Pull request #25975 synchronize by copybara-service bot
January 21, 2025 20:24 10m 56s test_716844471
test_716844471
January 21, 2025 20:24 10m 56s