Skip to content

Actions: jax-ml/jax

Actions

CI

Actions

Loading...
Loading

Show workflow options

Create status badge

Loading
ci-build.yaml
25,924 workflow runs
25,924 workflow runs

Filter by Event

Filter by Status

Filter by Branch

Filter by Actor

[Pallas/MGPU] Fix a race in the pipelining code CI #62474: Pull request #23939 synchronize by copybara-service bot
September 26, 2024 13:48 9m 37s test_679102765
test_679102765
September 26, 2024 13:48 9m 37s
[Pallas/Mosaic GPU] Implement a more comprehensive matmul kernel to see what we're still missing CI #62473: Pull request #23937 synchronize by copybara-service bot
September 26, 2024 13:41 9m 18s test_679076014
test_679076014
September 26, 2024 13:41 9m 18s
#sdy add JAX Shardy support for shard_map. CI #62472: Pull request #23799 synchronize by copybara-service bot
September 26, 2024 13:14 9m 20s test_672993069
test_672993069
September 26, 2024 13:14 9m 20s
[Pallas/Mosaic GPU] Implement a more comprehensive matmul kernel to see what we're still missing CI #62471: Pull request #23937 synchronize by copybara-service bot
September 26, 2024 13:10 11m 31s test_679076014
test_679076014
September 26, 2024 13:10 11m 31s
[Pallas/Mosaic GPU] Disable inference of sequential axis shapes CI #62470: Commit 5788773 pushed by copybara-service bot
September 26, 2024 12:53 10m 22s main
main
September 26, 2024 12:53 10m 22s
Merge pull request #23928 from nvcastet:fix_mosaic_barrier CI #62469: Commit a6b4648 pushed by copybara-service bot
September 26, 2024 12:51 2m 11s main
main
September 26, 2024 12:51 2m 11s
[Pallas/Mosaic GPU] Disable inference of sequential axis shapes CI #62468: Pull request #23938 synchronize by copybara-service bot
September 26, 2024 12:41 9m 37s test_679076013
test_679076013
September 26, 2024 12:41 9m 37s
[Pallas/MGPU] Skip outgoing TMA when the output is being revisited CI #62467: Pull request #23942 opened by copybara-service bot
September 26, 2024 12:40 10m 38s test_679111363
test_679111363
September 26, 2024 12:40 10m 38s
Raise ValueError when axis1==axis2 for jnp.trace CI #62466: Pull request #23921 synchronize by rajasekharporeddy
September 26, 2024 12:29 Action required rajasekharporeddy:testbranch4
rajasekharporeddy:testbranch4
September 26, 2024 12:29 Action required
[Pallas/MGPU] Implement block spec evaluation correctly CI #62465: Pull request #23941 opened by copybara-service bot
September 26, 2024 12:11 10m 32s test_679102766
test_679102766
September 26, 2024 12:11 10m 32s
Improve typing of jax.jit CI #62464: Pull request #23720 synchronize by lebrice
September 26, 2024 12:10 10m 9s lebrice:improve-jit-typing
lebrice:improve-jit-typing
September 26, 2024 12:10 10m 9s
Improve documentation for jax.jacobian CI #62463: Pull request #23940 opened by jakevdp
September 26, 2024 12:10 9m 59s jakevdp:jacobian-doc
jakevdp:jacobian-doc
September 26, 2024 12:10 9m 59s
[Pallas/MGPU] Fix a race in the pipelining code CI #62462: Pull request #23939 opened by copybara-service bot
September 26, 2024 12:10 9m 38s test_679102765
test_679102765
September 26, 2024 12:10 9m 38s
[Pallas/Mosaic GPU] Replace tiling/transpose fields of GPUBlockSpec w… CI #62461: Commit 3c25da2 pushed by copybara-service bot
September 26, 2024 10:41 10m 6s main
main
September 26, 2024 10:41 10m 6s
[Pallas/Mosaic GPU] Disable inference of sequential axis shapes CI #62460: Pull request #23938 opened by copybara-service bot
September 26, 2024 10:30 10m 38s test_679076013
test_679076013
September 26, 2024 10:30 10m 38s
[Pallas/Mosaic GPU] Implement a more comprehensive matmul kernel to see what we're still missing CI #62459: Pull request #23937 opened by copybara-service bot
September 26, 2024 10:30 10m 57s test_679076014
test_679076014
September 26, 2024 10:30 10m 57s
[Pallas/Mosaic GPU] Replace tiling/transpose fields of GPUBlockSpec with a transform list CI #62458: Pull request #23935 synchronize by copybara-service bot
September 26, 2024 10:29 10m 7s test_679045307
test_679045307
September 26, 2024 10:29 10m 7s
#sdy add JAX Shardy support for shard_map. CI #62457: Pull request #23799 synchronize by copybara-service bot
September 26, 2024 10:16 10m 14s test_672993069
test_672993069
September 26, 2024 10:16 10m 14s
Added support for lax.cond_p to Pallas Mosaic GPU lowering CI #62456: Pull request #23920 synchronize by copybara-service bot
September 26, 2024 10:10 9m 19s test_676133222
test_676133222
September 26, 2024 10:10 9m 19s
[pallas::mosaic_gpu] Turn the accumulator into a reference CI #62455: Commit b6d668e pushed by copybara-service bot
September 26, 2024 09:18 10m 46s main
main
September 26, 2024 09:18 10m 46s
[pallas::mosaic_gpu] Turn the accumulator into a reference CI #62454: Pull request #23793 synchronize by copybara-service bot
September 26, 2024 09:02 10m 43s test_676801466
test_676801466
September 26, 2024 09:02 10m 43s
[Pallas/Mosaic GPU] Replace tiling/transpose fields of GPUBlockSpec with a transform list CI #62453: Pull request #23935 synchronize by copybara-service bot
September 26, 2024 08:59 9m 57s test_679045307
test_679045307
September 26, 2024 08:59 9m 57s
[pallas::mosaic_gpu] Turn the accumulator into a reference CI #62452: Pull request #23793 synchronize by copybara-service bot
September 26, 2024 08:48 9m 53s test_676801466
test_676801466
September 26, 2024 08:48 9m 53s
[Pallas/Mosaic GPU] Replace tiling/transpose fields of GPUBlockSpec with a transform list CI #62451: Pull request #23935 opened by copybara-service bot
September 26, 2024 08:42 11m 35s test_679045307
test_679045307
September 26, 2024 08:42 11m 35s
Raise ValueError when axis1==axis2 for jnp.trace CI #62450: Pull request #23921 synchronize by rajasekharporeddy
September 26, 2024 08:32 Action required rajasekharporeddy:testbranch4
rajasekharporeddy:testbranch4
September 26, 2024 08:32 Action required