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

Filter by Event

Filter by Status

Filter by Branch

Filter by Actor

Allow multiple meshes for avals but in that case, just use empty_abstract_mesh instead of enabling computation follows data only for **Auto mode**. CI - Bazel CUDA tests (RBE) #554: Pull request #26197 synchronize by copybara-service bot
January 30, 2025 04:32 9m 3s test_721086072
test_721086072
January 30, 2025 04:32 9m 3s
Return arrays from ArrayImpl._check_and_rearrange. CI - Bazel CUDA tests (RBE) #553: Pull request #26166 synchronize by copybara-service bot
January 30, 2025 04:23 8m 7s test_720692284
test_720692284
January 30, 2025 04:23 8m 7s
Return arrays from ArrayImpl._check_and_rearrange. CI - Bazel CUDA tests (RBE) #552: Pull request #26166 synchronize by copybara-service bot
January 30, 2025 03:43 8m 54s test_720692284
test_720692284
January 30, 2025 03:43 8m 54s
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) #551: Pull request #25918 synchronize by copybara-service bot
January 30, 2025 03:21 9m 50s test_716042460
test_716042460
January 30, 2025 03:21 9m 50s
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) #550: Pull request #25918 synchronize by copybara-service bot
January 30, 2025 03:20 1m 0s test_716042460
test_716042460
January 30, 2025 03:20 1m 0s
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) #549: Pull request #25918 synchronize by copybara-service bot
January 30, 2025 03:10 10m 24s test_716042460
test_716042460
January 30, 2025 03:10 10m 24s
Return arrays from ArrayImpl._check_and_rearrange. CI - Bazel CUDA tests (RBE) #548: Pull request #26166 synchronize by copybara-service bot
January 30, 2025 02:37 8m 44s test_720692284
test_720692284
January 30, 2025 02:37 8m 44s
Return arrays from ArrayImpl._check_and_rearrange. CI - Bazel CUDA tests (RBE) #547: Pull request #26166 synchronize by copybara-service bot
January 30, 2025 02:36 53s test_720692284
test_720692284
January 30, 2025 02:36 53s
Return arrays from ArrayImpl._check_and_rearrange. CI - Bazel CUDA tests (RBE) #546: Pull request #26166 synchronize by copybara-service bot
January 30, 2025 01:53 10m 59s test_720692284
test_720692284
January 30, 2025 01:53 10m 59s
Return arrays from ArrayImpl._check_and_rearrange. CI - Bazel CUDA tests (RBE) #545: Pull request #26166 synchronize by copybara-service bot
January 30, 2025 01:03 8m 56s test_720692284
test_720692284
January 30, 2025 01:03 8m 56s
Refactor JAX wheel build rules to control the wheel filename and maintain reproducible wheel content and filename results. CI - Bazel CUDA tests (RBE) #544: Pull request #25126 synchronize by copybara-service bot
January 30, 2025 00:47 10m 32s test_699315679
test_699315679
January 30, 2025 00:47 10m 32s
Return arrays from ArrayImpl._check_and_rearrange. CI - Bazel CUDA tests (RBE) #543: Pull request #26166 synchronize by copybara-service bot
January 30, 2025 00:19 10m 20s test_720692284
test_720692284
January 30, 2025 00:19 10m 20s
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) #542: Pull request #25918 synchronize by copybara-service bot
January 29, 2025 23:53 11m 2s test_716042460
test_716042460
January 29, 2025 23:53 11m 2s
Add skeleton for a multi-pass source mapper for Jaxprs/HLO to jax.experimental. CI - Bazel CUDA tests (RBE) #541: Pull request #26193 synchronize by copybara-service bot
January 29, 2025 23:01 5m 19s test_718108389
test_718108389
January 29, 2025 23:01 5m 19s
Lower np.ndarray to DenseElementsAttr instead of ArrayAttr. CI - Bazel CUDA tests (RBE) #540: Pull request #26199 synchronize by copybara-service bot
January 29, 2025 23:01 9m 18s test_721117955
test_721117955
January 29, 2025 23:01 9m 18s
Lower np.ndarray to DenseElementsAttr instead of ArrayAttr. CI - Bazel CUDA tests (RBE) #539: Pull request #26199 opened by copybara-service bot
January 29, 2025 23:00 1m 6s test_721117955
test_721117955
January 29, 2025 23:00 1m 6s
Fix lax.ragged_all_to_all degenerate case CI - Bazel CUDA tests (RBE) #538: Pull request #26198 opened by copybara-service bot
January 29, 2025 22:53 10m 9s test_720669122
test_720669122
January 29, 2025 22:53 10m 9s
Add skeleton for a multi-pass source mapper for Jaxprs/HLO to jax.experimental. CI - Bazel CUDA tests (RBE) #537: Pull request #26193 synchronize by copybara-service bot
January 29, 2025 22:47 11m 37s test_718108389
test_718108389
January 29, 2025 22:47 11m 37s
Refactor JAX wheel build rules to control the wheel filename and maintain reproducible wheel content and filename results. CI - Bazel CUDA tests (RBE) #536: Pull request #25126 synchronize by copybara-service bot
January 29, 2025 22:42 10m 41s test_699315679
test_699315679
January 29, 2025 22:42 10m 41s
Add skeleton for a multi-pass source mapper for Jaxprs/HLO to jax.experimental. CI - Bazel CUDA tests (RBE) #535: Pull request #26193 synchronize by copybara-service bot
January 29, 2025 22:32 8m 16s test_718108389
test_718108389
January 29, 2025 22:32 8m 16s
add jax.distributed.is_initialized CI - Bazel CUDA tests (RBE) #534: Pull request #26172 synchronize by ZacCranko
January 29, 2025 22:22 10m 21s ZacCranko:is-distributed-init
ZacCranko:is-distributed-init
January 29, 2025 22:22 10m 21s
Enable pivoted QR on GPU via MAGMA. CI - Bazel CUDA tests (RBE) #533: Pull request #25955 synchronize by tttc3
January 29, 2025 22:14 13m 13s tttc3:magma_qr
tttc3:magma_qr
January 29, 2025 22:14 13m 13s
Refactor JAX wheel build rules to control the wheel filename and maintain reproducible wheel content and filename results. CI - Bazel CUDA tests (RBE) #532: Pull request #25126 synchronize by copybara-service bot
January 29, 2025 22:14 8m 17s test_699315679
test_699315679
January 29, 2025 22:14 8m 17s
Allow multiple meshes for avals but in that case, just use empty_abstract_mesh instead of enabling computation follows data only for **Auto mode**. CI - Bazel CUDA tests (RBE) #531: Pull request #26197 opened by copybara-service bot
January 29, 2025 22:09 8m 30s test_721086072
test_721086072
January 29, 2025 22:09 8m 30s
Add skeleton for a multi-pass source mapper for Jaxprs/HLO to jax.experimental. CI - Bazel CUDA tests (RBE) #530: Pull request #26193 synchronize by copybara-service bot
January 29, 2025 22:08 11m 31s test_718108389
test_718108389
January 29, 2025 22:08 11m 31s
ProTip! You can narrow down the results and go further in time using created:<2025-01-29 or the other filters available.