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

Allow P.UNCONSTRAINED in out_shardings at top level jit. This is required for sharding in types to work properly when out_avals contain UNCONSTRAINED specs. CI - Cloud TPU (presubmit) #109: Pull request #25524 opened by copybara-service bot
December 17, 2024 07:25 6m 19s test_706966253
test_706966253
December 17, 2024 07:25 6m 19s
Update jnp.floor_divide to make it consistent with np.floor_divide for division by zero CI - Cloud TPU (presubmit) #108: Pull request #25032 synchronize by rajasekharporeddy
December 17, 2024 05:56 6m 26s rajasekharporeddy:floor_divide
rajasekharporeddy:floor_divide
December 17, 2024 05:56 6m 26s
Put abstract_mesh on every eqn so that we can preserve it during eval_jaxpr and check_jaxpr roundtrip. CI - Cloud TPU (presubmit) #107: Pull request #25522 synchronize by copybara-service bot
December 17, 2024 05:30 6m 17s test_704907411
test_704907411
December 17, 2024 05:30 6m 17s
[Mosaic] Extend macros to handle parentheses. CI - Cloud TPU (presubmit) #106: Pull request #25523 synchronize by copybara-service bot
December 17, 2024 05:07 10m 30s test_706937151
test_706937151
December 17, 2024 05:07 10m 30s
[Mosaic] Extend macros to handle parentheses. CI - Cloud TPU (presubmit) #105: Pull request #25523 opened by copybara-service bot
December 17, 2024 05:07 56s test_706937151
test_706937151
December 17, 2024 05:07 56s
In progress experimentation for supporting JAX Arrays with variable-width strings (i.e., with dtype = StringDType). CI - Cloud TPU (presubmit) #104: Pull request #25322 synchronize by copybara-service bot
December 17, 2024 04:20 7m 9s test_703603535
test_703603535
December 17, 2024 04:20 7m 9s
In progress experimentation for supporting JAX Arrays with variable-width strings (i.e., with dtype = StringDType). CI - Cloud TPU (presubmit) #103: Pull request #25322 synchronize by copybara-service bot
December 17, 2024 04:08 7m 26s test_703603535
test_703603535
December 17, 2024 04:08 7m 26s
Put abstract_mesh on every eqn so that we can preserve it during eval_jaxpr and check_jaxpr roundtrip. CI - Cloud TPU (presubmit) #102: Pull request #25522 opened by copybara-service bot
December 17, 2024 04:04 6m 15s test_704907411
test_704907411
December 17, 2024 04:04 6m 15s
In progress experimentation for supporting JAX Arrays with variable-width strings (i.e., with dtype = StringDType). CI - Cloud TPU (presubmit) #101: Pull request #25322 synchronize by copybara-service bot
December 17, 2024 03:52 6m 11s test_703603535
test_703603535
December 17, 2024 03:52 6m 11s
Add lax.composite primitive CI - Cloud TPU (presubmit) #100: Pull request #25104 synchronize by copybara-service bot
December 17, 2024 03:22 6m 2s test_698570003
test_698570003
December 17, 2024 03:22 6m 2s
Add lax.composite primitive CI - Cloud TPU (presubmit) #99: Pull request #25104 synchronize by copybara-service bot
December 17, 2024 02:43 7m 0s test_698570003
test_698570003
December 17, 2024 02:43 7m 0s
In progress experimentation for supporting JAX Arrays with variable-width strings (i.e., with dtype = StringDType). CI - Cloud TPU (presubmit) #98: Pull request #25322 synchronize by copybara-service bot
December 17, 2024 01:17 6m 14s test_703603535
test_703603535
December 17, 2024 01:17 6m 14s
Added CI job with TSAN and free-threading CI - Cloud TPU (presubmit) #97: Pull request #24898 synchronize by vfdev-5
December 17, 2024 01:13 6m 11s vfdev-5:add-tsan-ft-ci-job
vfdev-5:add-tsan-ft-ci-job
December 17, 2024 01:13 6m 11s
Fix save_from_both_policies in presence of save_and_offload_only_these_names by comparing the enum CI - Cloud TPU (presubmit) #96: Pull request #25509 synchronize by copybara-service bot
December 17, 2024 00:41 7m 10s test_706720875
test_706720875
December 17, 2024 00:41 7m 10s
Fix save_from_both_policies in presence of save_and_offload_only_these_names by comparing the enum CI - Cloud TPU (presubmit) #95: Pull request #25509 synchronize by copybara-service bot
December 17, 2024 00:06 7m 6s test_706720875
test_706720875
December 17, 2024 00:06 7m 6s
Cleanup: toward merging core.concrete_aval & xla.abstractify CI - Cloud TPU (presubmit) #94: Pull request #25456 synchronize by jakevdp
December 17, 2024 00:04 7m 10s jakevdp:xla-abstractify
jakevdp:xla-abstractify
December 17, 2024 00:04 7m 10s
In progress experimentation for supporting JAX Arrays with variable-width strings (i.e., with dtype = StringDType). CI - Cloud TPU (presubmit) #93: Pull request #25322 synchronize by copybara-service bot
December 16, 2024 23:15 6m 5s test_703603535
test_703603535
December 16, 2024 23:15 6m 5s
Cleanup: toward merging core.concrete_aval & xla.abstractify CI - Cloud TPU (presubmit) #92: Pull request #25456 synchronize by jakevdp
December 16, 2024 22:46 6m 19s jakevdp:xla-abstractify
jakevdp:xla-abstractify
December 16, 2024 22:46 6m 19s
[Pallas TPU] Use vector.broadcast instead of vector.BroadcastOp to fix type check failure CI - Cloud TPU (presubmit) #91: Pull request #25520 synchronize by copybara-service bot
December 16, 2024 22:44 7m 10s test_706839044
test_706839044
December 16, 2024 22:44 7m 10s
[Pallas TPU] Use vector.broadcast instead of vector.BroadcastOp to fix type check failure CI - Cloud TPU (presubmit) #90: Pull request #25520 synchronize by copybara-service bot
December 16, 2024 22:43 1m 31s test_706839044
test_706839044
December 16, 2024 22:43 1m 31s
[Pallas TPU] Use vector.broadcast instead of vector.BroadcastOp to fix type check failure CI - Cloud TPU (presubmit) #89: Pull request #25520 opened by copybara-service bot
December 16, 2024 22:38 4m 22s test_706839044
test_706839044
December 16, 2024 22:38 4m 22s
[Pallas TPU] Add vector support to pl.debug_print CI - Cloud TPU (presubmit) #88: Pull request #25099 synchronize by copybara-service bot
December 16, 2024 22:23 6m 41s test_699920381
test_699920381
December 16, 2024 22:23 6m 41s
[Pallas TPU] Add vector support to pl.debug_print CI - Cloud TPU (presubmit) #87: Pull request #25099 synchronize by copybara-service bot
December 16, 2024 22:22 1m 7s test_699920381
test_699920381
December 16, 2024 22:22 1m 7s
[Jax][Pallas][Mosaic] Implement platform dependent diag, with branch selection driven by constant prop in mosaic lowering. CI - Cloud TPU (presubmit) #86: Pull request #25324 synchronize by copybara-service bot
December 16, 2024 21:19 8m 52s test_693594640
test_693594640
December 16, 2024 21:19 8m 52s
Re-deprecate a number of symbols from jax.core CI - Cloud TPU (presubmit) #85: Pull request #25508 synchronize by jakevdp
December 16, 2024 21:04 6m 22s jakevdp:dep-core
jakevdp:dep-core
December 16, 2024 21:04 6m 22s
ProTip! You can narrow down the results and go further in time using created:<2024-12-16 or the other filters available.