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

Adds a minimal but viable implementation of string arrays (with numpy.dtypes.StringDType) in JAX. Currently this only supports making of a string array by means of either jax.numpy.asarray or jax.device_put and reading it back with jax.device_get. CI - Cloud TPU (presubmit) #1847: Pull request #25918 synchronize by copybara-service bot
January 21, 2025 20:22 7m 30s test_716042460
test_716042460
January 21, 2025 20:22 7m 30s
Remove unnecessary use of xla_client.OpMetadata class. CI - Cloud TPU (presubmit) #1846: Pull request #25975 synchronize by copybara-service bot
January 21, 2025 20:10 7m 17s test_716844471
test_716844471
January 21, 2025 20:10 7m 17s
Refactor JAX wheel build rules to control the wheel filename and maintain reproducible wheel content and filename results. CI - Cloud TPU (presubmit) #1845: Pull request #25126 synchronize by copybara-service bot
January 21, 2025 20:02 1m 57s test_699315679
test_699315679
January 21, 2025 20:02 1m 57s
[Mosaic GPU] Add manual consumed barrier handling to WS pipeline. CI - Cloud TPU (presubmit) #1844: Pull request #25964 synchronize by copybara-service bot
January 21, 2025 19:29 6m 19s test_716735225
test_716735225
January 21, 2025 19:29 6m 19s
Experiment: preserve dict key order in tree.map CI - Cloud TPU (presubmit) #1843: Pull request #26014 synchronize by jakevdp
January 21, 2025 19:28 6m 12s jakevdp:tree-map
jakevdp:tree-map
January 21, 2025 19:28 6m 12s
[Mosaic TPU] Emulate converting x16 vector to mask if mask packing is supported. CI - Cloud TPU (presubmit) #1842: Pull request #25937 synchronize by copybara-service bot
January 21, 2025 19:24 6m 29s test_716395352
test_716395352
January 21, 2025 19:24 6m 29s
Experiment: preserve dict key order in tree.map CI - Cloud TPU (presubmit) #1841: Pull request #26014 synchronize by jakevdp
January 21, 2025 19:18 7m 33s jakevdp:tree-map
jakevdp:tree-map
January 21, 2025 19:18 7m 33s
Experiment: preserve dict key order in tree.map CI - Cloud TPU (presubmit) #1840: Pull request #26014 opened by jakevdp
January 21, 2025 19:15 3m 26s jakevdp:tree-map
jakevdp:tree-map
January 21, 2025 19:15 3m 26s
Adds a minimal but viable implementation of string arrays (with numpy.dtypes.StringDType) in JAX. Currently this only supports making of a string array by means of either jax.numpy.asarray or jax.device_put and reading it back with jax.device_get. CI - Cloud TPU (presubmit) #1839: Pull request #25918 synchronize by copybara-service bot
January 21, 2025 19:02 6m 20s test_716042460
test_716042460
January 21, 2025 19:02 6m 20s
Adds a minimal but viable implementation of string arrays (with numpy.dtypes.StringDType) in JAX. Currently this only supports making of a string array by means of either jax.numpy.asarray or jax.device_put and reading it back with jax.device_get. CI - Cloud TPU (presubmit) #1838: Pull request #25918 synchronize by copybara-service bot
January 21, 2025 18:38 7m 32s test_716042460
test_716042460
January 21, 2025 18:38 7m 32s
Add an option to simplify keystr output and use a custom separator. CI - Cloud TPU (presubmit) #1837: Pull request #25985 synchronize by copybara-service bot
January 21, 2025 18:18 6m 24s test_717455550
test_717455550
January 21, 2025 18:18 6m 24s
Add part (non-quantized K/V pages) of paged_attention_kernel tests back for TPU v6. CI - Cloud TPU (presubmit) #1836: Pull request #25934 synchronize by copybara-service bot
January 21, 2025 18:13 7m 15s test_716375593
test_716375593
January 21, 2025 18:13 7m 15s
Refactor JAX wheel build rules to control the wheel filename and maintain reproducible wheel content and filename results. CI - Cloud TPU (presubmit) #1835: Pull request #25126 synchronize by copybara-service bot
January 21, 2025 18:12 3m 3s test_699315679
test_699315679
January 21, 2025 18:12 3m 3s
[Mosaic] Remove hardcoded TARGET_SHAPE and align Python/C++ APIs. CI - Cloud TPU (presubmit) #1834: Pull request #25939 synchronize by copybara-service bot
January 21, 2025 18:10 7m 12s test_716383597
test_716383597
January 21, 2025 18:10 7m 12s
Add an option to simplify keystr output and use a custom separator. CI - Cloud TPU (presubmit) #1833: Pull request #25985 synchronize by copybara-service bot
January 21, 2025 17:49 6m 41s test_717455550
test_717455550
January 21, 2025 17:49 6m 41s
Add part (non-quantized K/V pages) of paged_attention_kernel tests back for TPU v6. CI - Cloud TPU (presubmit) #1832: Pull request #25934 synchronize by copybara-service bot
January 21, 2025 17:46 6m 28s test_716375593
test_716375593
January 21, 2025 17:46 6m 28s
Use ensure_arraylike utility in jax.numpy sources CI - Cloud TPU (presubmit) #1831: Pull request #25966 synchronize by copybara-service bot
January 21, 2025 17:44 7m 15s test_716762925
test_716762925
January 21, 2025 17:44 7m 15s
[mosaic_gpu] Removed debug prints in jax._src.lib.mosaic_gpu CI - Cloud TPU (presubmit) #1830: Pull request #26010 synchronize by copybara-service bot
January 21, 2025 17:36 6m 14s test_717913893
test_717913893
January 21, 2025 17:36 6m 14s
[Mosaic] Remove hardcoded TARGET_SHAPE and align Python/C++ APIs. CI - Cloud TPU (presubmit) #1829: Pull request #25939 synchronize by copybara-service bot
January 21, 2025 17:35 6m 41s test_716383597
test_716383597
January 21, 2025 17:35 6m 41s
[Mosaic] Remove hardcoded TARGET_SHAPE and align Python/C++ APIs. CI - Cloud TPU (presubmit) #1828: Pull request #25939 synchronize by copybara-service bot
January 21, 2025 17:34 1m 6s test_716383597
test_716383597
January 21, 2025 17:34 1m 6s
[Mosaic GPU] Add basic support for TMA with sub-byte types CI - Cloud TPU (presubmit) #1827: Pull request #26012 opened by copybara-service bot
January 21, 2025 17:34 6m 38s test_717952238
test_717952238
January 21, 2025 17:34 6m 38s
[mosaic_gpu] Removed debug prints in jax._src.lib.mosaic_gpu CI - Cloud TPU (presubmit) #1826: Pull request #26010 synchronize by copybara-service bot
January 21, 2025 17:21 6m 24s test_717913893
test_717913893
January 21, 2025 17:21 6m 24s
Remove meaningless template keywords. CI - Cloud TPU (presubmit) #1825: Pull request #26008 synchronize by copybara-service bot
January 21, 2025 17:21 6m 35s test_717904295
test_717904295
January 21, 2025 17:21 6m 35s
[mosaic_gpu] Removed debug prints in jax._src.lib.mosaic_gpu CI - Cloud TPU (presubmit) #1824: Pull request #26010 synchronize by copybara-service bot
January 21, 2025 17:12 6m 39s test_717913893
test_717913893
January 21, 2025 17:12 6m 39s
Remove meaningless template keywords. CI - Cloud TPU (presubmit) #1823: Pull request #26008 synchronize by copybara-service bot
January 21, 2025 17:09 7m 23s test_717904295
test_717904295
January 21, 2025 17:09 7m 23s