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
4,058 workflow runs
4,058 workflow runs

Filter by Event

Filter by Status

Filter by Branch

Filter by Actor

Internal compatibility change CI - Cloud TPU (presubmit) #4122: Pull request #26637 synchronize by copybara-service bot
February 21, 2025 09:22 8m 29s test_729144579
test_729144579
February 21, 2025 09:22 8m 29s
Internal compatibility change CI - Cloud TPU (presubmit) #4121: Commit b510127 pushed by copybara-service bot
February 21, 2025 09:22 8m 34s main
main
February 21, 2025 09:22 8m 34s
[Mosaic GPU] Use Index instead of I32 for the indices of async_load and async_store. CI - Cloud TPU (presubmit) #4120: Pull request #26655 opened by copybara-service bot
February 21, 2025 08:55 9m 4s test_729420704
test_729420704
February 21, 2025 08:55 9m 4s
[better_errors] Port the Pallas debug info mechanisms to the new JAX debugging info CI - Cloud TPU (presubmit) #4119: Pull request #26058 synchronize by gnecula
February 21, 2025 08:27 7m 50s gnecula:debug_info_pallas
gnecula:debug_info_pallas
February 21, 2025 08:27 7m 50s
Internal change for visibility CI - Cloud TPU (presubmit) #4118: Pull request #26654 opened by copybara-service bot
February 21, 2025 05:38 8m 31s test_729367507
test_729367507
February 21, 2025 05:38 8m 31s
Create JAX wheel build target. CI - Cloud TPU (presubmit) #4117: Pull request #26378 synchronize by copybara-service bot
February 21, 2025 05:27 8m 13s test_724102315
test_724102315
February 21, 2025 05:27 8m 13s
Test paged attention kernel on TPU v6e CI - Cloud TPU (presubmit) #4116: Pull request #26653 synchronize by copybara-service bot
February 21, 2025 05:18 8m 17s test_729360043
test_729360043
February 21, 2025 05:18 8m 17s
Test paged attention kernel on TPU v6e CI - Cloud TPU (presubmit) #4115: Pull request #26653 synchronize by copybara-service bot
February 21, 2025 05:09 8m 47s test_729360043
test_729360043
February 21, 2025 05:09 8m 47s
Create JAX wheel build target. CI - Cloud TPU (presubmit) #4114: Pull request #26378 synchronize by copybara-service bot
February 21, 2025 05:07 8m 24s test_724102315
test_724102315
February 21, 2025 05:07 8m 24s
Test paged attention kernel on TPU v6e CI - Cloud TPU (presubmit) #4113: Pull request #26653 opened by copybara-service bot
February 21, 2025 05:04 5m 51s test_729360043
test_729360043
February 21, 2025 05:04 5m 51s
Set the mesh of tangent.aval when we are creating zeros_like_aval because when you close over an array which is unused, we error out during canonicalization CI - Cloud TPU (presubmit) #4112: Pull request #26652 synchronize by copybara-service bot
February 21, 2025 03:32 8m 16s test_729299263
test_729299263
February 21, 2025 03:32 8m 16s
Set the mesh of tangent.aval when we are creating zeros_like_aval… CI - Cloud TPU (presubmit) #4111: Commit bcd4048 pushed by copybara-service bot
February 21, 2025 03:32 8m 51s main
main
February 21, 2025 03:32 8m 51s
Set the mesh of tangent.aval when we are creating zeros_like_aval because when you close over an array which is unused, we error out during canonicalization CI - Cloud TPU (presubmit) #4110: Pull request #26652 synchronize by copybara-service bot
February 21, 2025 03:17 9m 36s test_729299263
test_729299263
February 21, 2025 03:17 9m 36s
Set the mesh of tangent.aval when we are creating zeros_like_aval because when you close over an array which is unused, we error out during canonicalization CI - Cloud TPU (presubmit) #4109: Pull request #26652 synchronize by copybara-service bot
February 21, 2025 03:11 7m 14s test_729299263
test_729299263
February 21, 2025 03:11 7m 14s
Set the mesh of tangent.aval when we are creating zeros_like_aval because when you close over an array which is unused, we error out during canonicalization CI - Cloud TPU (presubmit) #4108: Pull request #26652 synchronize by copybara-service bot
February 21, 2025 02:52 8m 15s test_729299263
test_729299263
February 21, 2025 02:52 8m 15s
Set the mesh of tangent.aval when we are creating zeros_like_aval because when you close over an array which is unused, we error out during canonicalization CI - Cloud TPU (presubmit) #4107: Pull request #26652 synchronize by copybara-service bot
February 21, 2025 02:51 1m 33s test_729299263
test_729299263
February 21, 2025 02:51 1m 33s
Set the mesh of tangent.aval when we are creating zeros_like_aval because when you close over an array which is unused, we error out during canonicalization CI - Cloud TPU (presubmit) #4106: Pull request #26652 synchronize by copybara-service bot
February 21, 2025 02:38 8m 30s test_729299263
test_729299263
February 21, 2025 02:38 8m 30s
Set the mesh of tangent.aval when we are creating zeros_like_aval because when you close over an array which is unused, we error out during canonicalization CI - Cloud TPU (presubmit) #4105: Pull request #26652 opened by copybara-service bot
February 21, 2025 02:36 1m 53s test_729299263
test_729299263
February 21, 2025 02:36 1m 53s
Use the mesh of out_aval when converting GSPMDSharding to NamedSharding. This makes sure that the axis types of the corresponding output is correct. CI - Cloud TPU (presubmit) #4104: Pull request #26651 synchronize by copybara-service bot
February 21, 2025 01:13 8m 21s test_729195322
test_729195322
February 21, 2025 01:13 8m 21s
Use the mesh of out_aval when converting GSPMDSharding to NamedShar… CI - Cloud TPU (presubmit) #4103: Commit 250e2ee pushed by copybara-service bot
February 21, 2025 01:13 8m 22s main
main
February 21, 2025 01:13 8m 22s
Use the mesh of out_aval when converting GSPMDSharding to NamedSharding. This makes sure that the axis types of the corresponding output is correct. CI - Cloud TPU (presubmit) #4102: Pull request #26651 synchronize by copybara-service bot
February 21, 2025 00:56 8m 27s test_729195322
test_729195322
February 21, 2025 00:56 8m 27s
Create JAX wheel build target. CI - Cloud TPU (presubmit) #4101: Pull request #26378 synchronize by copybara-service bot
February 21, 2025 00:37 8m 16s test_724102315
test_724102315
February 21, 2025 00:37 8m 16s
Use the mesh of out_aval when converting GSPMDSharding to NamedSharding. This makes sure that the axis types of the corresponding output is correct. CI - Cloud TPU (presubmit) #4100: Pull request #26651 opened by copybara-service bot
February 21, 2025 00:23 8m 22s test_729195322
test_729195322
February 21, 2025 00:23 8m 22s
[Pallas] Add option for async DMAs in the new TPU interpret mode CI - Cloud TPU (presubmit) #4099: Pull request #26650 opened by copybara-service bot
February 20, 2025 23:41 8m 4s test_729245019
test_729245019
February 20, 2025 23:41 8m 4s
roofline: Handle ClosedJaxpr instances. CI - Cloud TPU (presubmit) #4098: Pull request #26648 synchronize by copybara-service bot
February 20, 2025 23:20 9m 18s test_729230803
test_729230803
February 20, 2025 23:20 9m 18s