Skip to content

Actions: jax-ml/jax

Actions

CI - Cloud TPU (presubmit)

Actions

Loading...
Loading

Show workflow options

Create status badge

Loading
cloud-tpu-ci-presubmit.yml
1,856 workflow runs
1,856 workflow runs

Filter by Event

Filter by Status

Filter by Branch

Filter by Actor

[Mosaic GPU] Add support for converting all fragmented layouts to ir and back. CI - Cloud TPU (presubmit) #1797: Pull request #26003 synchronize by copybara-service bot
January 21, 2025 11:27 9m 12s test_717820872
test_717820872
January 21, 2025 11:27 9m 12s
[mosaic_gpu] a function for bitwidth as well as bytewidht. CI - Cloud TPU (presubmit) #1796: Pull request #26006 opened by copybara-service bot
January 21, 2025 11:26 4m 49s test_717826998
test_717826998
January 21, 2025 11:26 4m 49s
[Mosaic TPU] Add optimized casts for bf16->s4 in TPUv6 CI - Cloud TPU (presubmit) #1795: Pull request #26005 opened by copybara-service bot
January 21, 2025 11:13 6m 38s test_717829264
test_717829264
January 21, 2025 11:13 6m 38s
[Mosaic GPU] Add support for converting all fragmented layouts to ir and back. CI - Cloud TPU (presubmit) #1794: Pull request #26003 synchronize by copybara-service bot
January 21, 2025 11:13 7m 17s test_717820872
test_717820872
January 21, 2025 11:13 7m 17s
[Mosaic GPU] Add support for converting all fragmented layouts to ir and back. CI - Cloud TPU (presubmit) #1793: Pull request #26003 opened by copybara-service bot
January 21, 2025 10:41 6m 45s test_717820872
test_717820872
January 21, 2025 10:41 6m 45s
[Pallas] Improve testing for casts from narrow types + test int4 CI - Cloud TPU (presubmit) #1792: Pull request #26002 opened by copybara-service bot
January 21, 2025 10:39 6m 21s test_717821169
test_717821169
January 21, 2025 10:39 6m 21s
[better_errors] Consolidate TracingDebugInfo and JaxprDebugInfo CI - Cloud TPU (presubmit) #1791: Pull request #26001 synchronize by gnecula
January 21, 2025 09:30 6m 15s gnecula:debug_info_6
gnecula:debug_info_6
January 21, 2025 09:30 6m 15s
[better_errors] Ensure debug_info.arg_names is never None. CI - Cloud TPU (presubmit) #1790: Pull request #25992 synchronize by gnecula
January 21, 2025 09:10 6m 14s gnecula:debug_info_arg_names
gnecula:debug_info_arg_names
January 21, 2025 09:10 6m 14s
[better_errors] Consolidate TracingDebugInfo and JaxprDebugInfo CI - Cloud TPU (presubmit) #1789: Pull request #26001 opened by gnecula
January 21, 2025 09:03 6m 11s gnecula:debug_info_6
gnecula:debug_info_6
January 21, 2025 09:03 6m 11s
[better_errors] Ensure debug_info.arg_names is never None. CI - Cloud TPU (presubmit) #1788: Pull request #25992 synchronize by gnecula
January 21, 2025 08:58 6m 8s gnecula:debug_info_arg_names
gnecula:debug_info_arg_names
January 21, 2025 08:58 6m 8s
[Mosaic GPU] Do not use mgpu in wgmma.py CI - Cloud TPU (presubmit) #1787: Pull request #25999 synchronize by copybara-service bot
January 21, 2025 08:52 7m 19s test_717784498
test_717784498
January 21, 2025 08:52 7m 19s
[Mosaic GPU] Do not use mgpu in wgmma.py CI - Cloud TPU (presubmit) #1786: Pull request #25999 opened by copybara-service bot
January 21, 2025 08:48 4m 28s test_717784498
test_717784498
January 21, 2025 08:48 4m 28s
[better_errors] Ensure debug_info.arg_names is never None. CI - Cloud TPU (presubmit) #1785: Pull request #25992 synchronize by gnecula
January 21, 2025 08:16 6m 0s gnecula:debug_info_arg_names
gnecula:debug_info_arg_names
January 21, 2025 08:16 6m 0s
[better_errors] Ensure debug_info.arg_names is never None. CI - Cloud TPU (presubmit) #1784: Pull request #25992 synchronize by gnecula
January 21, 2025 08:03 6m 15s gnecula:debug_info_arg_names
gnecula:debug_info_arg_names
January 21, 2025 08:03 6m 15s
[better_errors] Ensure debug_info.arg_names is never None. CI - Cloud TPU (presubmit) #1783: Pull request #25992 synchronize by gnecula
January 21, 2025 07:58 4m 57s gnecula:debug_info_arg_names
gnecula:debug_info_arg_names
January 21, 2025 07:58 4m 57s
Make make_mesh take visible_axes, hidden_axes and collective_axes as parameters instead of axis_types to make it a more cleaner API. CI - Cloud TPU (presubmit) #1782: Pull request #25997 synchronize by copybara-service bot
January 21, 2025 07:52 7m 11s test_717718333
test_717718333
January 21, 2025 07:52 7m 11s
Make make_mesh take visible_axes, hidden_axes and collective_axes as parameters instead of axis_types to make it a more cleaner API. CI - Cloud TPU (presubmit) #1781: Pull request #25997 opened by copybara-service bot
January 21, 2025 07:42 6m 6s test_717718333
test_717718333
January 21, 2025 07:42 6m 6s
Make sure reshard and mesh_cast behave properly under eager mode CI - Cloud TPU (presubmit) #1780: Pull request #25996 synchronize by copybara-service bot
January 21, 2025 04:19 7m 11s test_717702155
test_717702155
January 21, 2025 04:19 7m 11s
Make sure reshard and mesh_cast behave properly under eager mode CI - Cloud TPU (presubmit) #1779: Pull request #25996 synchronize by copybara-service bot
January 21, 2025 04:04 7m 22s test_717702155
test_717702155
January 21, 2025 04:04 7m 22s
Make sure reshard and mesh_cast behave properly under eager mode CI - Cloud TPU (presubmit) #1778: Pull request #25996 opened by copybara-service bot
January 21, 2025 04:03 1m 0s test_717702155
test_717702155
January 21, 2025 04:03 1m 0s
[sharding_in_types] Move the calculation of new_mesh inside the decorator so that functions can be decorated with hidden_axes and visible_axes at the top level CI - Cloud TPU (presubmit) #1777: Pull request #25995 synchronize by copybara-service bot
January 21, 2025 03:41 6m 16s test_717691707
test_717691707
January 21, 2025 03:41 6m 16s
[sharding_in_types] Move the calculation of new_mesh inside the decorator so that functions can be decorated with hidden_axes and visible_axes at the top level CI - Cloud TPU (presubmit) #1776: Pull request #25995 synchronize by copybara-service bot
January 21, 2025 03:26 6m 18s test_717691707
test_717691707
January 21, 2025 03:26 6m 18s
[sharding_in_types] Move the calculation of new_mesh inside the decorator so that functions can be decorated with hidden_axes and visible_axes at the top level CI - Cloud TPU (presubmit) #1775: Pull request #25995 opened by copybara-service bot
January 21, 2025 03:15 6m 36s test_717691707
test_717691707
January 21, 2025 03:15 6m 36s
Fix attention backwards pass CI - Cloud TPU (presubmit) #1774: Pull request #25647 synchronize by Rifur13
January 21, 2025 02:51 7m 21s Rifur13:bwd_pass
Rifur13:bwd_pass
January 21, 2025 02:51 7m 21s
Clarify type hints for wrapped functions CI - Cloud TPU (presubmit) #1773: Pull request #25994 opened by ncoghlan
January 21, 2025 01:45 6m 52s ncoghlan:patch-1
ncoghlan:patch-1
January 21, 2025 01:45 6m 52s