Skip to content

Actions: jax-ml/jax

CI - Bazel GPU tests (RBE)

Actions

Loading...
Loading

Show workflow options

Create status badge

Loading
1,685 workflow runs
1,685 workflow runs

Filter by Event

Filter by Status

Filter by Branch

Filter by Actor

In progress experimention. Add StringDType to JAX's supported types.
CI - Bazel GPU tests (RBE) #1625: Pull request #25865 opened by copybara-service bot
January 13, 2025 22:53 9m 54s test_715114024
January 13, 2025 22:53 9m 54s
More linearization fixes
CI - Bazel GPU tests (RBE) #1624: Pull request #25864 opened by dougalm
January 13, 2025 22:06 9m 41s yet-more-linearization-fixes
January 13, 2025 22:06 9m 41s
Jax: Stop returning a list of cost-analyses.
CI - Bazel GPU tests (RBE) #1623: Pull request #25542 synchronize by copybara-service bot
January 13, 2025 21:37 11m 4s test_707170952
January 13, 2025 21:37 11m 4s
[Pallas TPU] Add vector support to pl.debug_print
CI - Bazel GPU tests (RBE) #1622: Pull request #25099 synchronize by copybara-service bot
January 13, 2025 21:22 10m 34s test_699920381
January 13, 2025 21:22 10m 34s
Jax: Stop returning a list of cost-analyses.
CI - Bazel GPU tests (RBE) #1621: Pull request #25542 synchronize by copybara-service bot
January 13, 2025 21:09 10m 10s test_707170952
January 13, 2025 21:09 10m 10s
[Mosaic TPU] Add a faster implementation for packing b16 to s8 in TPUv6
CI - Bazel GPU tests (RBE) #1620: Pull request #25829 synchronize by copybara-service bot
January 13, 2025 21:07 12m 5s test_713997968
January 13, 2025 21:07 12m 5s
[Mosaic GPU] Fix layout API bugs.
CI - Bazel GPU tests (RBE) #1619: Pull request #25860 synchronize by copybara-service bot
January 13, 2025 20:59 5m 15s test_715015735
January 13, 2025 20:59 5m 15s
[Mosaic GPU] Fix layout API bugs.
CI - Bazel GPU tests (RBE) #1618: Pull request #25860 synchronize by copybara-service bot
January 13, 2025 20:45 7m 18s test_715015735
January 13, 2025 20:45 7m 18s
Start a new TPU interpret mode for Pallas
CI - Bazel GPU tests (RBE) #1617: Pull request #25097 synchronize by jburnim
January 13, 2025 20:35 8m 5s jburnim:jburnim_pallas_interpret_mode
January 13, 2025 20:35 8m 5s
More linearize fixes
CI - Bazel GPU tests (RBE) #1616: Pull request #25833 synchronize by dougalm
January 13, 2025 20:30 31m 33s more-linearize-fixes
January 13, 2025 20:30 31m 33s
[Pallas TPU] Add vector support to pl.debug_print
CI - Bazel GPU tests (RBE) #1615: Pull request #25099 synchronize by copybara-service bot
January 13, 2025 20:25 6m 52s test_699920381
January 13, 2025 20:25 6m 52s
Jax: Stop returning a list of cost-analyses.
CI - Bazel GPU tests (RBE) #1614: Pull request #25542 synchronize by copybara-service bot
January 13, 2025 20:05 6m 29s test_707170952
January 13, 2025 20:05 6m 29s
[Pallas TPU] Add vector support to pl.debug_print
CI - Bazel GPU tests (RBE) #1613: Pull request #25099 synchronize by copybara-service bot
January 13, 2025 19:58 10m 55s test_699920381
January 13, 2025 19:58 10m 55s
More linearize fixes
CI - Bazel GPU tests (RBE) #1612: Pull request #25833 synchronize by dougalm
January 13, 2025 19:56 7m 50s more-linearize-fixes
January 13, 2025 19:56 7m 50s
[Docs] Remove --xla_gpu_enable_triton_softmax_fusion from docs
CI - Bazel GPU tests (RBE) #1611: Pull request #25863 opened by copybara-service bot
January 13, 2025 19:43 7m 46s test_715051174
January 13, 2025 19:43 7m 46s
[Mosaic] Allow passing ApplyVectorLayoutCtx to tpu.apply_layout_op.
CI - Bazel GPU tests (RBE) #1610: Pull request #25862 opened by copybara-service bot
January 13, 2025 19:24 9m 35s test_715042538
January 13, 2025 19:24 9m 35s
In progress experimention. Add StringDType to JAX's supported types.
CI - Bazel GPU tests (RBE) #1609: Pull request #25820 synchronize by copybara-service bot
January 13, 2025 19:20 9m 9s test_713874040
January 13, 2025 19:20 9m 9s
In progress experimention. Add StringDType to JAX's supported types.
CI - Bazel GPU tests (RBE) #1608: Pull request #25820 synchronize by copybara-service bot
January 13, 2025 19:19 1m 5s test_713874040
January 13, 2025 19:19 1m 5s
[Mosaic TPU] Append dump id to timestamp to make dump list ordered
CI - Bazel GPU tests (RBE) #1607: Pull request #25793 synchronize by copybara-service bot
January 13, 2025 18:56 8m 24s test_713450623
January 13, 2025 18:56 8m 24s
Jax: Stop returning a list of cost-analyses.
CI - Bazel GPU tests (RBE) #1606: Pull request #25542 synchronize by copybara-service bot
January 13, 2025 18:33 10m 11s test_707170952
January 13, 2025 18:33 10m 11s
[Pallas TPU] Add vector support to pl.debug_print
CI - Bazel GPU tests (RBE) #1605: Pull request #25099 synchronize by copybara-service bot
January 13, 2025 18:25 6m 37s test_699920381
January 13, 2025 18:25 6m 37s
Remove dead codepaths now that MemorySpaceDescription works in OSS
CI - Bazel GPU tests (RBE) #1604: Pull request #25621 synchronize by copybara-service bot
January 13, 2025 18:23 8m 56s test_708016987
January 13, 2025 18:23 8m 56s
[Mosaic GPU] Fix layout API bugs.
CI - Bazel GPU tests (RBE) #1603: Pull request #25860 synchronize by copybara-service bot
January 13, 2025 18:20 5m 52s test_715015735
January 13, 2025 18:20 5m 52s
[Pallas TPU] Add vector support to pl.debug_print
CI - Bazel GPU tests (RBE) #1602: Pull request #25099 synchronize by copybara-service bot
January 13, 2025 18:20 4m 41s test_699920381
January 13, 2025 18:20 4m 41s
[Mosaic GPU] Fix layout API bugs.
CI - Bazel GPU tests (RBE) #1601: Pull request #25860 opened by copybara-service bot
January 13, 2025 18:18 3m 0s test_715015735
January 13, 2025 18:18 3m 0s