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] Removed debug prints in jax._src.lib.mosaic_gpu CI - Cloud TPU (presubmit) #1822: Pull request #26010 synchronize by copybara-service bot
January 21, 2025 17:07 5m 33s test_717913893
test_717913893
January 21, 2025 17:07 5m 33s
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) #1821: Pull request #25997 synchronize by copybara-service bot
January 21, 2025 17:01 6m 20s test_717718333
test_717718333
January 21, 2025 17:01 6m 20s
[jax:custom_partitioning] Support SdyShardingRule with multiple leading CI - Cloud TPU (presubmit) #1820: Pull request #25886 synchronize by copybara-service bot
January 21, 2025 16:57 6m 23s test_715515514
test_715515514
January 21, 2025 16:57 6m 23s
[jax:custom_partitioning] Support SdyShardingRule with multiple leading CI - Cloud TPU (presubmit) #1819: Pull request #25886 synchronize by copybara-service bot
January 21, 2025 16:49 7m 19s test_715515514
test_715515514
January 21, 2025 16:49 7m 19s
[jax:custom_partitioning] Support SdyShardingRule with multiple leading CI - Cloud TPU (presubmit) #1818: Pull request #25886 synchronize by copybara-service bot
January 21, 2025 16:37 7m 18s test_715515514
test_715515514
January 21, 2025 16:37 7m 18s
Fix cache init when JAX Array is created early (#25768) CI - Cloud TPU (presubmit) #1817: Pull request #25889 synchronize by Stella-S-Yan
January 21, 2025 16:34 8m 12s Stella-S-Yan:cache_reset
Stella-S-Yan:cache_reset
January 21, 2025 16:34 8m 12s
Implement gaussian_filter1d and gaussian_filter CI - Cloud TPU (presubmit) #1816: Pull request #26011 opened by adonath
January 21, 2025 16:30 Action required adonath:gaussian_filter
adonath:gaussian_filter
January 21, 2025 16:30 Action required
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) #1815: Pull request #25997 synchronize by copybara-service bot
January 21, 2025 16:28 13m 47s test_717718333
test_717718333
January 21, 2025 16:28 13m 47s
Fix cache init when JAX Array is created early (#25768) CI - Cloud TPU (presubmit) #1814: Pull request #25889 synchronize by Stella-S-Yan
January 21, 2025 16:25 9m 2s Stella-S-Yan:cache_reset
Stella-S-Yan:cache_reset
January 21, 2025 16:25 9m 2s
[mosaic_gpu] Removed debug prints in jax._src.lib.mosaic_gpu CI - Cloud TPU (presubmit) #1813: Pull request #26010 synchronize by copybara-service bot
January 21, 2025 16:24 18m 28s test_717913893
test_717913893
January 21, 2025 16:24 18m 28s
Use ensure_arraylike utility in jax.numpy sources CI - Cloud TPU (presubmit) #1812: Pull request #25966 synchronize by copybara-service bot
January 21, 2025 16:10 6m 31s test_716762925
test_716762925
January 21, 2025 16:10 6m 31s
Refactor JAX wheel build rules to control the wheel filename and maintain reproducible wheel content and filename results. CI - Cloud TPU (presubmit) #1811: Pull request #25126 synchronize by copybara-service bot
January 21, 2025 16:01 1m 57s test_699315679
test_699315679
January 21, 2025 16:01 1m 57s
[mosaic_gpu] Removed debug prints in jax._src.lib.mosaic_gpu CI - Cloud TPU (presubmit) #1810: Pull request #26010 opened by copybara-service bot
January 21, 2025 15:50 6m 21s test_717913893
test_717913893
January 21, 2025 15:50 6m 21s
Remove meaningless template keywords. CI - Cloud TPU (presubmit) #1809: Pull request #26008 opened by copybara-service bot
January 21, 2025 15:42 7m 17s test_717904295
test_717904295
January 21, 2025 15:42 7m 17s
[jax:custom_partitioning] Support SdyShardingRule with multiple leading CI - Cloud TPU (presubmit) #1808: Pull request #25886 synchronize by copybara-service bot
January 21, 2025 15:36 7m 11s test_715515514
test_715515514
January 21, 2025 15:36 7m 11s
[mosaic_gpu] a function for bitwidth as well as bytewidht. CI - Cloud TPU (presubmit) #1807: Pull request #26006 synchronize by copybara-service bot
January 21, 2025 14:38 8m 45s test_717826998
test_717826998
January 21, 2025 14:38 8m 45s
[Pallas] Improve testing for casts from narrow types + test int4 CI - Cloud TPU (presubmit) #1806: Pull request #26002 synchronize by copybara-service bot
January 21, 2025 13:56 6m 19s test_717821169
test_717821169
January 21, 2025 13:56 6m 19s
[Pallas] Improve testing for casts from narrow types + test int4 CI - Cloud TPU (presubmit) #1805: Pull request #26002 synchronize by copybara-service bot
January 21, 2025 13:43 8m 50s test_717821169
test_717821169
January 21, 2025 13:43 8m 50s
[Mosaic GPU] Implement basic WGMMAFragLayout inference and propagation CI - Cloud TPU (presubmit) #1804: Pull request #26007 opened by copybara-service bot
January 21, 2025 12:43 6m 10s test_717844490
test_717844490
January 21, 2025 12:43 6m 10s
[better_errors] Ensure debug_info.arg_names is never None. CI - Cloud TPU (presubmit) #1803: Pull request #25992 synchronize by gnecula
January 21, 2025 12:38 6m 27s gnecula:debug_info_arg_names
gnecula:debug_info_arg_names
January 21, 2025 12:38 6m 27s
[mosaic_gpu] a function for bitwidth as well as bytewidht. CI - Cloud TPU (presubmit) #1802: Pull request #26006 synchronize by copybara-service bot
January 21, 2025 12:11 6m 3s test_717826998
test_717826998
January 21, 2025 12:11 6m 3s
[mosaic_gpu] a function for bitwidth as well as bytewidht. CI - Cloud TPU (presubmit) #1801: Pull request #26006 synchronize by copybara-service bot
January 21, 2025 11:53 6m 29s test_717826998
test_717826998
January 21, 2025 11:53 6m 29s
[mosaic_gpu] a function for bitwidth as well as bytewidht. CI - Cloud TPU (presubmit) #1800: Pull request #26006 synchronize by copybara-service bot
January 21, 2025 11:51 1m 57s test_717826998
test_717826998
January 21, 2025 11:51 1m 57s
[Pallas] Improve testing for casts from narrow types + test int4 CI - Cloud TPU (presubmit) #1799: Pull request #26002 synchronize by copybara-service bot
January 21, 2025 11:45 7m 44s test_717821169
test_717821169
January 21, 2025 11:45 7m 44s
[mosaic_gpu] a function for bitwidth as well as bytewidht. CI - Cloud TPU (presubmit) #1798: Pull request #26006 synchronize by copybara-service bot
January 21, 2025 11:31 7m 26s test_717826998
test_717826998
January 21, 2025 11:31 7m 26s