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
5,590 workflow runs
5,590 workflow runs

Filter by Event

Filter by Status

Filter by Branch

Filter by Actor

[Mosaic GPU] Add support for .cta_group::2 MMA with n=512 on Blackwell CI - Cloud TPU (presubmit) #5685: Pull request #27055 synchronize by copybara-service bot
March 11, 2025 16:20 8m 40s test_735699619
test_735699619
March 11, 2025 16:20 8m 40s
Test new LibTPU dependency. CI - Cloud TPU (presubmit) #5684: Pull request #27046 synchronize by copybara-service bot
March 11, 2025 16:13 8m 57s test_735534881
test_735534881
March 11, 2025 16:13 8m 57s
Test new LibTPU dependency. CI - Cloud TPU (presubmit) #5683: Pull request #27046 synchronize by copybara-service bot
March 11, 2025 15:52 17m 14s test_735534881
test_735534881
March 11, 2025 15:52 17m 14s
Opaque pointers for callbacks. CI - Cloud TPU (presubmit) #5682: Pull request #27050 synchronize by copybara-service bot
March 11, 2025 15:51 16m 27s test_735615230
test_735615230
March 11, 2025 15:51 16m 27s
[ROCm] Add gfx1101 Target and deprecate gfx940, gfx941 CI - Cloud TPU (presubmit) #5681: Pull request #26882 synchronize by JehandadKhan
March 11, 2025 15:46 22m 4s ROCm:jd-gfx1101-upstream
ROCm:jd-gfx1101-upstream
March 11, 2025 15:46 22m 4s
[ROCm] Add gfx1101 Target and deprecate gfx940, gfx941 CI - Cloud TPU (presubmit) #5680: Pull request #26882 synchronize by JehandadKhan
March 11, 2025 15:43 3m 15s ROCm:jd-gfx1101-upstream
ROCm:jd-gfx1101-upstream
March 11, 2025 15:43 3m 15s
Disable running thread-unsafe hypothesis tests in state_test.py CI - Cloud TPU (presubmit) #5679: Pull request #27058 opened by vfdev-5
March 11, 2025 15:36 7m 32s vfdev-5:ft-adapt-state-test
vfdev-5:ft-adapt-state-test
March 11, 2025 15:36 7m 32s
Update :build_jaxlib flag to control whether we should add py_import dependencies to the test targets. CI - Cloud TPU (presubmit) #5678: Pull request #26945 synchronize by copybara-service bot
March 11, 2025 15:31 7m 38s test_733777754
test_733777754
March 11, 2025 15:31 7m 38s
Update :build_jaxlib flag to control whether we should add `py_impo… CI - Cloud TPU (presubmit) #5677: Commit 1aca76f pushed by copybara-service bot
March 11, 2025 15:31 7m 26s main
main
March 11, 2025 15:31 7m 26s
Test new LibTPU dependency. CI - Cloud TPU (presubmit) #5676: Pull request #27046 synchronize by copybara-service bot
March 11, 2025 15:19 8m 43s test_735534881
test_735534881
March 11, 2025 15:19 8m 43s
[Mosaic GPU] Add TiledLayout to the mlir dialect and use it in layout inference, instead of WGMMAFragLayout. CI - Cloud TPU (presubmit) #5675: Pull request #27057 synchronize by copybara-service bot
March 11, 2025 15:17 8m 23s test_735760879
test_735760879
March 11, 2025 15:17 8m 23s
[Mosaic GPU] Add TiledLayout to the mlir dialect and use it in layout inference, instead of WGMMAFragLayout. CI - Cloud TPU (presubmit) #5674: Pull request #27057 opened by copybara-service bot
March 11, 2025 15:16 1m 9s test_735760879
test_735760879
March 11, 2025 15:16 1m 9s
Update :build_jaxlib flag to control whether we should add py_import dependencies to the test targets. CI - Cloud TPU (presubmit) #5673: Pull request #26945 synchronize by copybara-service bot
March 11, 2025 15:03 7m 20s test_733777754
test_733777754
March 11, 2025 15:03 7m 20s
[Mosaic GPU] Check in WIP persistent matmul kernel CI - Cloud TPU (presubmit) #5672: Pull request #26997 synchronize by andportnoy
March 11, 2025 15:00 Action required andportnoy:fdgg
andportnoy:fdgg
March 11, 2025 15:00 Action required
[Mosaic GPU] Check in WIP persistent matmul kernel CI - Cloud TPU (presubmit) #5671: Pull request #26997 synchronize by andportnoy
March 11, 2025 14:58 Action required andportnoy:fdgg
andportnoy:fdgg
March 11, 2025 14:58 Action required
[ROCm] Fix ROCm builds not finding numa library CI - Cloud TPU (presubmit) #5670: Pull request #26775 synchronize by charleshofer
March 11, 2025 14:58 Action required ROCm:rocm-fix-numalib
ROCm:rocm-fix-numalib
March 11, 2025 14:58 Action required
Proof-of-concept: A JAX callback that operates directly on device buffers. CI - Cloud TPU (presubmit) #5669: Pull request #26850 synchronize by copybara-service bot
March 11, 2025 14:50 3m 4s test_728322827
test_728322827
March 11, 2025 14:50 3m 4s
[Pallas/Mosaic GPU] Explicitly disable ops_test on Mosaic GPU pre-H… CI - Cloud TPU (presubmit) #5668: Commit 7fd32ec pushed by copybara-service bot
March 11, 2025 14:11 7m 24s main
main
March 11, 2025 14:11 7m 24s
[Pallas/Mosaic GPU] Explicitly disable ops_test on Mosaic GPU pre-Hopper. CI - Cloud TPU (presubmit) #5667: Pull request #27056 opened by copybara-service bot
March 11, 2025 13:27 9m 46s test_735734074
test_735734074
March 11, 2025 13:27 9m 46s
Support error checking in explicit mode CI - Cloud TPU (presubmit) #5666: Pull request #26549 synchronize by copybara-service bot
March 11, 2025 12:54 8m 18s test_726939419
test_726939419
March 11, 2025 12:54 8m 18s
Support error checking in explicit mode CI - Cloud TPU (presubmit) #5665: Pull request #26549 synchronize by copybara-service bot
March 11, 2025 11:53 7m 8s test_726939419
test_726939419
March 11, 2025 11:53 7m 8s
Update XLA dependency to use revision CI - Cloud TPU (presubmit) #5664: Commit b6da46e pushed by copybara-service bot
March 11, 2025 11:42 7m 31s main
main
March 11, 2025 11:42 7m 31s
Opaque pointers for callbacks. CI - Cloud TPU (presubmit) #5663: Pull request #27050 synchronize by copybara-service bot
March 11, 2025 11:11 7m 27s test_735615230
test_735615230
March 11, 2025 11:11 7m 27s
[Mosaic GPU] Add support for .cta_group::2 MMA with n=512 on Blackwell CI - Cloud TPU (presubmit) #5662: Pull request #27055 opened by copybara-service bot
March 11, 2025 11:00 7m 16s test_735699619
test_735699619
March 11, 2025 11:00 7m 16s
Add support for scratch buffers in jax_triton. CI - Cloud TPU (presubmit) #5661: Pull request #27025 synchronize by copybara-service bot
March 11, 2025 10:46 8m 0s test_735226186
test_735226186
March 11, 2025 10:46 8m 0s