Skip to content

Actions: jax-ml/jax

Actions

CI - Bazel CPU tests (RBE)

Actions

Loading...
Loading

Show workflow options

Create status badge

Loading
bazel_cpu_rbe.yml
1,659 workflow runs
1,659 workflow runs

Filter by Event

Filter by Status

Filter by Branch

Filter by Actor

Debug info 3 WIP CI - Bazel CPU tests (RBE) #1674: Pull request #25876 opened by gnecula
January 14, 2025 14:56 Queued gnecula:debug_info_3
gnecula:debug_info_3
January 14, 2025 14:56 Queued
[pallas/pallas_mgpu] Discharging run_scoped should not be discharging the intermediates CI - Bazel CPU tests (RBE) #1673: Pull request #25639 synchronize by copybara-service bot
January 14, 2025 14:20 8m 14s test_703334972
test_703334972
January 14, 2025 14:20 8m 14s
More linearization fixes CI - Bazel CPU tests (RBE) #1672: Pull request #25864 synchronize by dougalm
January 14, 2025 13:54 7m 5s yet-more-linearization-fixes
yet-more-linearization-fixes
January 14, 2025 13:54 7m 5s
[pallas] Support DMA start partial discharge and run_scoped() does its own partial discharge. CI - Bazel CPU tests (RBE) #1671: Pull request #25775 synchronize by copybara-service bot
January 14, 2025 13:54 5m 54s test_713280422
test_713280422
January 14, 2025 13:54 5m 54s
Mention expected tangent aval in error message, see #25517. CI - Bazel CPU tests (RBE) #1670: Pull request #25875 opened by dougalm
January 14, 2025 13:51 7m 53s issue-25517
issue-25517
January 14, 2025 13:51 7m 53s
[better_errors] Add debug_info to DynamicJaxprTrace and JaxprStackFrame CI - Bazel CPU tests (RBE) #1669: Pull request #25827 synchronize by gnecula
January 14, 2025 13:49 9m 47s gnecula:debug_info_2
gnecula:debug_info_2
January 14, 2025 13:49 9m 47s
[jax2tf] Fix bitrot in docs CI - Bazel CPU tests (RBE) #1668: Pull request #25872 opened by gnecula
January 14, 2025 11:36 6m 24s gnecula:jax2tf_doc
gnecula:jax2tf_doc
January 14, 2025 11:36 6m 24s
[Pallas] Fix integer array indexing CI - Bazel CPU tests (RBE) #1667: Pull request #23758 synchronize by copybara-service bot
January 14, 2025 11:06 18m 15s test_676368116
test_676368116
January 14, 2025 11:06 18m 15s
#sdy dynamically choose which custom_partitioning API to use based on the current CI - Bazel CPU tests (RBE) #1666: Pull request #25834 synchronize by copybara-service bot
January 14, 2025 10:12 4m 24s test_714054637
test_714054637
January 14, 2025 10:12 4m 24s
[better_errors] Add debug_info to DynamicJaxprTrace and JaxprStackFrame CI - Bazel CPU tests (RBE) #1665: Pull request #25827 synchronize by gnecula
January 14, 2025 10:10 8m 29s gnecula:debug_info_2
gnecula:debug_info_2
January 14, 2025 10:10 8m 29s
Automated Code Change CI - Bazel CPU tests (RBE) #1664: Pull request #25871 opened by copybara-service bot
January 14, 2025 10:09 1m 29s test_715178155
test_715178155
January 14, 2025 10:09 1m 29s
#sdy dynamically choose which custom_partitioning API to use based on the current CI - Bazel CPU tests (RBE) #1663: Pull request #25834 synchronize by copybara-service bot
January 14, 2025 09:43 19m 48s test_714054637
test_714054637
January 14, 2025 09:43 19m 48s
[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 CPU tests (RBE) #1662: Pull request #25849 synchronize by copybara-service bot
January 14, 2025 07:37 5m 36s test_706880226
test_706880226
January 14, 2025 07:37 5m 36s
[better_errors] Add debug_info to DynamicJaxprTrace and JaxprStackFrame CI - Bazel CPU tests (RBE) #1661: Pull request #25827 synchronize by gnecula
January 14, 2025 07:08 4m 48s gnecula:debug_info_2
gnecula:debug_info_2
January 14, 2025 07:08 4m 48s
enable partitionable threefry by default CI - Bazel CPU tests (RBE) #1660: Pull request #25363 synchronize by copybara-service bot
January 14, 2025 06:46 4m 14s test_702832099
test_702832099
January 14, 2025 06:46 4m 14s
enable partitionable threefry by default CI - Bazel CPU tests (RBE) #1659: Pull request #25363 synchronize by copybara-service bot
January 14, 2025 06:32 3m 59s test_702832099
test_702832099
January 14, 2025 06:32 3m 59s
Pin actions/checkout to a commit CI - Bazel CPU tests (RBE) #1658: Pull request #25870 opened by copybara-service bot
January 14, 2025 06:23 5m 32s test_715236865
test_715236865
January 14, 2025 06:23 5m 32s
[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 CPU tests (RBE) #1657: Pull request #25869 synchronize by copybara-service bot
January 14, 2025 06:11 6m 18s test_715221163
test_715221163
January 14, 2025 06:11 6m 18s
[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 CPU tests (RBE) #1656: Pull request #25869 synchronize by copybara-service bot
January 14, 2025 06:10 1m 8s test_715221163
test_715221163
January 14, 2025 06:10 1m 8s
enable partitionable threefry by default CI - Bazel CPU tests (RBE) #1655: Pull request #25363 synchronize by copybara-service bot
January 14, 2025 05:46 7m 9s test_702832099
test_702832099
January 14, 2025 05:46 7m 9s
[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 CPU tests (RBE) #1654: Pull request #25869 opened by copybara-service bot
January 14, 2025 05:41 6m 44s test_715221163
test_715221163
January 14, 2025 05:41 6m 44s
More linearization fixes CI - Bazel CPU tests (RBE) #1653: Pull request #25864 synchronize by dougalm
January 14, 2025 03:55 4m 33s yet-more-linearization-fixes
yet-more-linearization-fixes
January 14, 2025 03:55 4m 33s
More linearization fixes CI - Bazel CPU tests (RBE) #1652: Pull request #25864 synchronize by dougalm
January 14, 2025 03:16 6m 22s yet-more-linearization-fixes
yet-more-linearization-fixes
January 14, 2025 03:16 6m 22s
More linearization fixes CI - Bazel CPU tests (RBE) #1651: Pull request #25864 synchronize by dougalm
January 14, 2025 01:45 6m 8s yet-more-linearization-fixes
yet-more-linearization-fixes
January 14, 2025 01:45 6m 8s
Fix remat bug on primitives with multiple outputs. CI - Bazel CPU tests (RBE) #1650: Pull request #25866 synchronize by copybara-service bot
January 14, 2025 01:25 3m 56s test_715125226
test_715125226
January 14, 2025 01:25 3m 56s