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
3,435 workflow runs
3,435 workflow runs

Filter by Event

Filter by Status

Filter by Branch

Filter by Actor

[Mosaic GPU] Remove old jaxlib version guards. CI - Cloud TPU (presubmit) #3487: Commit 837418c pushed by copybara-service bot
February 12, 2025 16:49 In progress main
main
February 12, 2025 16:49 In progress
[sharding_in_types] Make the typing checks and sharding rule checks a little bit less strict when the current or aval mesh is empty/unset. Also some more changes as listed below: CI - Cloud TPU (presubmit) #3486: Pull request #26477 synchronize by copybara-service bot
February 12, 2025 16:34 11m 2s test_723699680
test_723699680
February 12, 2025 16:34 11m 2s
[Mosaic GPU] Remove old jaxlib version guards. CI - Cloud TPU (presubmit) #3485: Pull request #26490 synchronize by copybara-service bot
February 12, 2025 16:34 11m 10s test_726049813
test_726049813
February 12, 2025 16:34 11m 10s
[sharding_in_types] When caching mesh with axis_types, make sure the data structure is (axis_size, axis_names, tuple(axis_types)) CI - Cloud TPU (presubmit) #3484: Pull request #26473 synchronize by copybara-service bot
February 12, 2025 16:23 8m 52s test_725697451
test_725697451
February 12, 2025 16:23 8m 52s
[sharding_in_types] When caching mesh with axis_types, make sure the … CI - Cloud TPU (presubmit) #3483: Commit b4b4a98 pushed by copybara-service bot
February 12, 2025 16:23 9m 23s main
main
February 12, 2025 16:23 9m 23s
[Mosaic GPU] Put all inline asm output constraints before input constraints in optimization barrier CI - Cloud TPU (presubmit) #3482: Pull request #26495 opened by copybara-service bot
February 12, 2025 16:16 12m 45s test_726061404
test_726061404
February 12, 2025 16:16 12m 45s
Make sure we take libTPU version into account in the Pallas lowering CI - Cloud TPU (presubmit) #3481: Pull request #26487 synchronize by copybara-service bot
February 12, 2025 16:16 13m 12s test_726024766
test_726024766
February 12, 2025 16:16 13m 12s
Make sure we take libTPU version into account in the Pallas lowering CI - Cloud TPU (presubmit) #3480: Commit f1ab751 pushed by copybara-service bot
February 12, 2025 16:16 8m 6s main
main
February 12, 2025 16:16 8m 6s
Enable pivoted QR on GPU via MAGMA. CI - Cloud TPU (presubmit) #3479: Pull request #25955 synchronize by tttc3
February 12, 2025 16:13 8m 58s tttc3:magma_qr
tttc3:magma_qr
February 12, 2025 16:13 8m 58s
[Mosaic GPU] Simplify the collective barrier test to avoid GMEM atomics CI - Cloud TPU (presubmit) #3478: Pull request #26494 opened by copybara-service bot
February 12, 2025 16:06 7m 6s test_726058177
test_726058177
February 12, 2025 16:06 7m 6s
[sharding_in_types] When caching mesh with axis_types, make sure the data structure is (axis_size, axis_names, tuple(axis_types)) CI - Cloud TPU (presubmit) #3477: Pull request #26473 synchronize by copybara-service bot
February 12, 2025 16:05 6m 21s test_725697451
test_725697451
February 12, 2025 16:05 6m 21s
Make sure we take libTPU version into account in the Pallas lowering CI - Cloud TPU (presubmit) #3476: Pull request #26487 synchronize by copybara-service bot
February 12, 2025 16:02 8m 7s test_726024766
test_726024766
February 12, 2025 16:02 8m 7s
[Mosaic GPU] Reorganize tests to make sure WGMMA tests are skipped on Blackwell CI - Cloud TPU (presubmit) #3475: Pull request #26493 opened by copybara-service bot
February 12, 2025 15:57 6m 29s test_726054872
test_726054872
February 12, 2025 15:57 6m 29s
Codebeaver/test 723699680 26477 CI - Cloud TPU (presubmit) #3474: Pull request #26492 opened by CodeBeaverAI
February 12, 2025 15:43 Action required CodeBeaverAI:codebeaver/test_723699680-26477
CodeBeaverAI:codebeaver/test_723699680-26477
February 12, 2025 15:43 Action required
Make sure we take libTPU version into account in the Pallas lowering CI - Cloud TPU (presubmit) #3473: Pull request #26487 synchronize by copybara-service bot
February 12, 2025 15:42 8m 5s test_726024766
test_726024766
February 12, 2025 15:42 8m 5s
[Mosaic GPU] Remove old jaxlib version guards. CI - Cloud TPU (presubmit) #3472: Pull request #26490 synchronize by copybara-service bot
February 12, 2025 15:41 6m 26s test_726049813
test_726049813
February 12, 2025 15:41 6m 26s
Part 1 of a new autodidax based on "stackless" CI - Cloud TPU (presubmit) #3471: Pull request #26373 synchronize by dougalm
February 12, 2025 15:41 6m 26s autodidax-stackless
autodidax-stackless
February 12, 2025 15:41 6m 26s
[Mosaic GPU] Define TMEMLayout without referring to the PTX guide CI - Cloud TPU (presubmit) #3470: Pull request #26491 opened by copybara-service bot
February 12, 2025 15:40 6m 17s test_726049815
test_726049815
February 12, 2025 15:40 6m 17s
[Mosaic GPU] Remove old jaxlib version guards. CI - Cloud TPU (presubmit) #3469: Pull request #26490 opened by copybara-service bot
February 12, 2025 15:38 3m 4s test_726049813
test_726049813
February 12, 2025 15:38 3m 4s
Download and use jax wheels from GCS bucket for nightly/release test workflows CI - Cloud TPU (presubmit) #3468: Pull request #26489 synchronize by copybara-service bot
February 12, 2025 15:29 6m 43s test_726037038
test_726037038
February 12, 2025 15:29 6m 43s
Merge pull request #26447 from jakevdp:refactor-contractions CI - Cloud TPU (presubmit) #3467: Commit e14466a pushed by copybara-service bot
February 12, 2025 15:14 6m 21s main
main
February 12, 2025 15:14 6m 21s
Add JAX error checking support CI - Cloud TPU (presubmit) #3466: Pull request #26469 synchronize by copybara-service bot
February 12, 2025 15:13 6m 32s test_725633039
test_725633039
February 12, 2025 15:13 6m 32s
Make sure we take libTPU version into account in the Pallas lowering CI - Cloud TPU (presubmit) #3465: Pull request #26487 synchronize by copybara-service bot
February 12, 2025 15:02 8m 0s test_726024766
test_726024766
February 12, 2025 15:02 8m 0s
Fix the string array test to not assume that there will always be exactly 2 CPU devices CI - Cloud TPU (presubmit) #3464: Pull request #26488 synchronize by copybara-service bot
February 12, 2025 14:55 6m 16s test_726032202
test_726032202
February 12, 2025 14:55 6m 16s
Fix the string array test to not assume that there will always be exa… CI - Cloud TPU (presubmit) #3463: Commit 6662ea9 pushed by copybara-service bot
February 12, 2025 14:55 6m 37s main
main
February 12, 2025 14:55 6m 37s