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

Filter by Event

Filter by Status

Filter by Branch

Filter by Actor

[Pallas] Fix integer array indexing CI - Bazel GPU tests (RBE) #1653: Pull request #23758 synchronize by copybara-service bot
January 14, 2025 11:06 In progress test_676368116
test_676368116
January 14, 2025 11:06 In progress
#sdy dynamically choose which custom_partitioning API to use based on the current CI - Bazel GPU tests (RBE) #1652: Pull request #25834 synchronize by copybara-service bot
January 14, 2025 10:12 10m 22s test_714054637
test_714054637
January 14, 2025 10:12 10m 22s
[better_errors] Add debug_info to DynamicJaxprTrace and JaxprStackFrame CI - Bazel GPU tests (RBE) #1651: Pull request #25827 synchronize by gnecula
January 14, 2025 10:10 11m 44s gnecula:debug_info_2
gnecula:debug_info_2
January 14, 2025 10:10 11m 44s
Automated Code Change CI - Bazel GPU tests (RBE) #1650: Pull request #25871 opened by copybara-service bot
January 14, 2025 10:09 4m 34s test_715178155
test_715178155
January 14, 2025 10:09 4m 34s
#sdy dynamically choose which custom_partitioning API to use based on the current CI - Bazel GPU tests (RBE) #1649: Pull request #25834 synchronize by copybara-service bot
January 14, 2025 09:43 19m 6s test_714054637
test_714054637
January 14, 2025 09:43 19m 6s
[pallas][mosaic kernel export] Add initial support for exporting a dynamic shapes (placeholder bound) kernel out of mosaic, via pallas as both MLIR and jaxpr. CI - Bazel GPU tests (RBE) #1648: Pull request #25849 synchronize by copybara-service bot
January 14, 2025 07:37 9m 38s test_706880226
test_706880226
January 14, 2025 07:37 9m 38s
[better_errors] Add debug_info to DynamicJaxprTrace and JaxprStackFrame CI - Bazel GPU tests (RBE) #1647: Pull request #25827 synchronize by gnecula
January 14, 2025 07:08 8m 52s gnecula:debug_info_2
gnecula:debug_info_2
January 14, 2025 07:08 8m 52s
enable partitionable threefry by default CI - Bazel GPU tests (RBE) #1646: Pull request #25363 synchronize by copybara-service bot
January 14, 2025 06:46 6m 29s test_702832099
test_702832099
January 14, 2025 06:46 6m 29s
enable partitionable threefry by default CI - Bazel GPU tests (RBE) #1645: Pull request #25363 synchronize by copybara-service bot
January 14, 2025 06:32 6m 33s test_702832099
test_702832099
January 14, 2025 06:32 6m 33s
Pin actions/checkout to a commit CI - Bazel GPU tests (RBE) #1644: Pull request #25870 opened by copybara-service bot
January 14, 2025 06:23 8m 48s test_715236865
test_715236865
January 14, 2025 06:23 8m 48s
[sharding_in_types] Handle ShapeDtypeStruct inputs with sharding_in_types by registering the sharding on the aval properly created by SDS in it's pytype_aval_mapping. CI - Bazel GPU tests (RBE) #1643: Pull request #25869 synchronize by copybara-service bot
January 14, 2025 06:11 9m 30s test_715221163
test_715221163
January 14, 2025 06:11 9m 30s
[sharding_in_types] Handle ShapeDtypeStruct inputs with sharding_in_types by registering the sharding on the aval properly created by SDS in it's pytype_aval_mapping. CI - Bazel GPU tests (RBE) #1642: Pull request #25869 synchronize by copybara-service bot
January 14, 2025 06:10 1m 6s test_715221163
test_715221163
January 14, 2025 06:10 1m 6s
enable partitionable threefry by default CI - Bazel GPU tests (RBE) #1641: Pull request #25363 synchronize by copybara-service bot
January 14, 2025 05:46 9m 7s test_702832099
test_702832099
January 14, 2025 05:46 9m 7s
[sharding_in_types] Handle ShapeDtypeStruct inputs with sharding_in_types by registering the sharding on the aval properly created by SDS in it's pytype_aval_mapping. CI - Bazel GPU tests (RBE) #1640: Pull request #25869 opened by copybara-service bot
January 14, 2025 05:41 13m 10s test_715221163
test_715221163
January 14, 2025 05:41 13m 10s
More linearization fixes CI - Bazel GPU tests (RBE) #1639: Pull request #25864 synchronize by dougalm
January 14, 2025 03:55 8m 14s yet-more-linearization-fixes
yet-more-linearization-fixes
January 14, 2025 03:55 8m 14s
More linearization fixes CI - Bazel GPU tests (RBE) #1638: Pull request #25864 synchronize by dougalm
January 14, 2025 03:16 10m 23s yet-more-linearization-fixes
yet-more-linearization-fixes
January 14, 2025 03:16 10m 23s
More linearization fixes CI - Bazel GPU tests (RBE) #1637: Pull request #25864 synchronize by dougalm
January 14, 2025 01:45 27s yet-more-linearization-fixes
yet-more-linearization-fixes
January 14, 2025 01:45 27s
Fix remat bug on primitives with multiple outputs. CI - Bazel GPU tests (RBE) #1636: Pull request #25866 synchronize by copybara-service bot
January 14, 2025 01:25 10m 9s test_715125226
test_715125226
January 14, 2025 01:25 10m 9s
More linearization fixes CI - Bazel GPU tests (RBE) #1635: Pull request #25864 synchronize by dougalm
January 14, 2025 01:17 9m 33s yet-more-linearization-fixes
yet-more-linearization-fixes
January 14, 2025 01:17 9m 33s
[Mosaic] Fix infer/apply extensions. CI - Bazel GPU tests (RBE) #1634: Pull request #25868 opened by copybara-service bot
January 14, 2025 00:50 9m 57s test_715154257
test_715154257
January 14, 2025 00:50 9m 57s
Fix remat bug on primitives with multiple outputs. CI - Bazel GPU tests (RBE) #1633: Pull request #25866 synchronize by copybara-service bot
January 14, 2025 00:29 10m 3s test_715125226
test_715125226
January 14, 2025 00:29 10m 3s
Fix remat bug on primitives with multiple outputs. CI - Bazel GPU tests (RBE) #1632: Pull request #25866 synchronize by copybara-service bot
January 14, 2025 00:28 59s test_715125226
test_715125226
January 14, 2025 00:28 59s
Fix remat bug on primitives with multiple outputs. CI - Bazel GPU tests (RBE) #1631: Pull request #25866 opened by copybara-service bot
January 14, 2025 00:26 1m 38s test_715125226
test_715125226
January 14, 2025 00:26 1m 38s
New Experimental JAX Check Numerics API. CI - Bazel GPU tests (RBE) #1630: Pull request #25785 synchronize by epiqueras
January 14, 2025 00:21 11m 14s epiqueras:feature/numerics
epiqueras:feature/numerics
January 14, 2025 00:21 11m 14s
New Experimental JAX Check Numerics API. CI - Bazel GPU tests (RBE) #1629: Pull request #25785 synchronize by epiqueras
January 14, 2025 00:20 34s epiqueras:feature/numerics
epiqueras:feature/numerics
January 14, 2025 00:20 34s