Skip to content
CI

Unify JAX and Pallas unit test workflows to overcome the 20-workflow limit in CI #1890

Sign in to view logs

Unify JAX and Pallas unit test workflows to overcome the 20-workflow limit in CI

Unify JAX and Pallas unit test workflows to overcome the 20-workflow limit in CI #1890

Run time

Learn about OS pricing on GitHub Actions
Job Run time
metadata
1s
metadata
1s
amd64 / test-distribution (extra-only-distribution.sh)
9s
amd64 / test-distribution (extra-only-distribution.sh)
9s
arm64 / test-distribution (extra-only-distribution.sh)
5s
arm64 / test-distribution (extra-only-distribution.sh)
5s
amd64 / test-distribution (mirror-only-distribution.sh)
6s
amd64 / test-distribution (mirror-only-distribution.sh)
6s
arm64 / test-distribution (mirror-only-distribution.sh)
10s
arm64 / test-distribution (mirror-only-distribution.sh)
10s
amd64 / test-distribution (upstream-only-distribution.sh)
4s
amd64 / test-distribution (upstream-only-distribution.sh)
4s
arm64 / test-distribution (upstream-only-distribution.sh)
5s
arm64 / test-distribution (upstream-only-distribution.sh)
5s
amd64 / build-base / bump-manifest
7s
amd64 / build-base / bump-manifest
7s
arm64 / build-base / bump-manifest
4s
arm64 / build-base / bump-manifest
4s
arm64 / build-base / build-base
5m 43s
arm64 / build-base / build-base
5m 43s
amd64 / build-base / build-base
5m 29s
amd64 / build-base / build-base
5m 29s
amd64 / build-jax / build-jax
21m 0s
amd64 / build-jax / build-jax
8m 55s
amd64 / build-maxtext
0s
amd64 / build-pallas
0s
amd64 / build-t5x
0s
amd64 / build-levanter
0s
amd64 / build-pax
0s
amd64 / test-jax
0s
amd64 / build-rosetta-t5x
0s
amd64 / test-pallas
0s
amd64 / test-upstream-t5x
0s
amd64 / test-te
0s
amd64 / build-rosetta-pax
0s
amd64 / test-upstream-pax
0s
amd64 / test-rosetta-t5x
0s
amd64 / collect-docker-tags
1s
amd64 / collect-docker-tags
1s
amd64 / test-rosetta-pax
0s
arm64 / build-jax / build-jax
12m 29s
arm64 / build-jax / build-jax
12m 29s
arm64 / build-maxtext
0s
arm64 / build-maxtext
0s
arm64 / build-pallas
0s
arm64 / build-pallas
0s
arm64 / test-jax
0s
arm64 / test-jax
0s
arm64 / build-levanter / build-levanter
15m 32s
arm64 / build-levanter / build-levanter
15m 32s
arm64 / build-t5x / build-upstream-t5x
32m 28s
arm64 / build-t5x / build-upstream-t5x
32m 28s
arm64 / build-pax / build-upstream-pax
25m 25s
arm64 / build-pax / build-upstream-pax
25m 25s
arm64 / test-pallas
0s
arm64 / test-pallas
0s
arm64 / test-te
0s
arm64 / test-te
0s
arm64 / test-upstream-pax
0s
arm64 / test-upstream-pax
0s
arm64 / build-rosetta-pax / build-rosetta
16m 31s
arm64 / build-rosetta-pax / build-rosetta
16m 31s
arm64 / test-upstream-t5x
0s
arm64 / test-upstream-t5x
0s
arm64 / build-rosetta-t5x / build-rosetta
20m 23s
arm64 / build-rosetta-t5x / build-rosetta
20m 23s
arm64 / test-rosetta-pax
0s
arm64 / test-rosetta-pax
0s
arm64 / collect-docker-tags
1s
arm64 / collect-docker-tags
1s
arm64 / test-rosetta-t5x
0s
arm64 / test-rosetta-t5x
0s
make-publish-configs
3s
make-publish-configs
5s
finalize / upload-badge
7s
finalize / upload-badge
1m 43s
finalize / report
4s
finalize / report
15s
publish (ghcr.io/nvidia/jax-toolbox-internal:7825968984-upstream-t5x-arm64, mock-jax, type=raw,va... / publish
10s
publish (ghcr.io/nvidia/jax-toolbox-internal:7825968984-jax-amd64, ghcr.io/nvidia/jax-toolbox-int... / publish
27s
publish (ghcr.io/nvidia/jax-toolbox-internal:7825968984-upstream-pax-arm64, mock-jax, type=raw,va... / publish
17s
publish (ghcr.io/nvidia/jax-toolbox-internal:7825968984-pallas-amd64, mock-jax, type=raw,value=pa... / publish
29s
publish (ghcr.io/nvidia/jax-toolbox-internal:7825968984-pax-arm64, mock-jax, type=raw,value=roset... / publish
14s
publish (ghcr.io/nvidia/jax-toolbox-internal:7825968984-levanter-amd64, ghcr.io/nvidia/jax-toolbo... / publish
16s
publish (ghcr.io/nvidia/jax-toolbox-internal:7825968984-t5x-arm64, mock-jax, type=raw,value=roset... / publish
19s
publish (ghcr.io/nvidia/jax-toolbox-internal:7825968984-maxtext-amd64, mock-jax, type=raw,value=m... / publish
14s
publish (ghcr.io/nvidia/jax-toolbox-internal:7825968984-levanter-arm64-mealkit, mock-jax-mealkit,... / publish
10s
publish (ghcr.io/nvidia/jax-toolbox-internal:7825968984-pallas-amd64-mealkit, mock-jax-mealkit, t... / publish
12s
publish (ghcr.io/nvidia/jax-toolbox-internal:7825968984-upstream-pax-arm64-mealkit, mock-jax-meal... / publish
10s
publish (ghcr.io/nvidia/jax-toolbox-internal:7825968984-levanter-amd64-mealkit, ghcr.io/nvidia/ja... / publish
24s
publish (ghcr.io/nvidia/jax-toolbox-internal:7825968984-upstream-t5x-arm64-mealkit, mock-jax-meal... / publish
15s
publish (ghcr.io/nvidia/jax-toolbox-internal:7825968984-maxtext-amd64-mealkit, mock-jax-mealkit, ... / publish
16s
publish (ghcr.io/nvidia/jax-toolbox-internal:7825968984-t5x-arm64-mealkit, mock-jax-mealkit, type... / publish
20s
publish (ghcr.io/nvidia/jax-toolbox-internal:7825968984-upstream-t5x-amd64-mealkit, ghcr.io/nvidi... / publish
22s
publish (ghcr.io/nvidia/jax-toolbox-internal:7825968984-pax-arm64-mealkit, mock-jax-mealkit, type... / publish
16s
publish (ghcr.io/nvidia/jax-toolbox-internal:7825968984-upstream-pax-amd64-mealkit, ghcr.io/nvidi... / publish
14s
publish (ghcr.io/nvidia/jax-toolbox-internal:7825968984-base-amd64, ghcr.io/nvidia/jax-toolbox-in... / publish
19s
publish (ghcr.io/nvidia/jax-toolbox-internal:7825968984-t5x-amd64-mealkit, ghcr.io/nvidia/jax-too... / publish
17s
publish (ghcr.io/nvidia/jax-toolbox-internal:7825968984-jax-arm64, mock-jax, type=raw,value=jax,p... / publish
10s
publish (ghcr.io/nvidia/jax-toolbox-internal:7825968984-pax-amd64-mealkit, ghcr.io/nvidia/jax-too... / publish
23s
publish (ghcr.io/nvidia/jax-toolbox-internal:7825968984-levanter-arm64, mock-jax, type=raw,value=... / publish
12s
publish (ghcr.io/nvidia/jax-toolbox-internal:7825968984-base-amd64, ghcr.io/nvidia/jax-toolbox-in... / publish
24s
publish (ghcr.io/nvidia/jax-toolbox-internal:7825968984-jax-arm64-mealkit, mock-jax-mealkit, type... / publish
19s
publish (ghcr.io/nvidia/jax-toolbox-internal:7825968984-jax-amd64-mealkit, ghcr.io/nvidia/jax-too... / publish
13s
finalize / publish-badge
0s
finalize / publish-badge
0s
amd64 / build-maxtext / build-maxtext
5m 8s
amd64 / build-levanter / build-levanter
4m 56s
amd64 / build-t5x / build-upstream-t5x
8m 10s
amd64 / build-pax / build-upstream-pax
6m 59s
amd64 / test-jax / jax-unit-test (V100)
25m 48s
amd64 / test-jax / jax-unit-test (A100)
8m 41s
amd64 / test-jax / pallas-unit-test
0s
amd64 / build-pallas / build-pallas
21m 51s
amd64 / test-jax / runner / launch-slurm-runner
3h 53m 52s
amd64 / build-rosetta-pax / build-rosetta
6m 53s
amd64 / test-te / te-multi-gpu (1)
32m 30s
amd64 / test-te / te-unit-tests
22m 4s
amd64 / test-te / te-multi-gpu (2)
43m 34s
amd64 / test-upstream-pax / single-process-multi-device (1, 8, 1, 1)
3h 57m 57s
amd64 / test-upstream-pax / single-process-multi-device (1, 1, 2, 4)
4h 2m 11s
amd64 / test-upstream-pax / pax-multi-node (1, 1, 1, 1)
4h 5m 13s
amd64 / test-upstream-pax / pax-multi-node (1, 8, 1, 1)
3h 40m 30s
amd64 / test-upstream-pax / pax-multi-node (1, 1, 8, 1)
3h 34m 54s
amd64 / test-upstream-pax / pax-multi-node (4, 2, 1, 1)
4h 6m 46s
amd64 / test-upstream-pax / single-process-evaluation (1, 8, 1, 1)
3h 29m 44s
amd64 / test-upstream-pax / pax-multi-node (1, 4, 1, 2)
3h 59m 52s
amd64 / test-upstream-pax / pax-multi-node (1, 16, 1, 1)
3h 54m 28s
amd64 / test-upstream-pax / pax-multi-node (4, 2, 1, 2)
3h 49m 18s
amd64 / build-rosetta-t5x / build-rosetta
11m 3s
amd64 / test-upstream-t5x / t5x-multi-gpu (1)
4h 9m 25s
amd64 / test-upstream-t5x / t5x-multi-gpu (2)
4h 6m 32s
amd64 / test-upstream-t5x / t5x-multi-gpu (4)
4h 10m 48s
amd64 / test-upstream-t5x / t5x-multi-gpu (8)
4h 28m 30s
amd64 / test-upstream-t5x / t5x-multi-node (1, 1)
4h 7m 57s
amd64 / test-upstream-t5x / t5x-multi-node (1, 2)
4h 25m 40s
amd64 / test-upstream-t5x / t5x-multi-node (2, 1)
4h 12m 2s
amd64 / test-upstream-t5x / t5x-multi-node (2, 2)
4h 14m 59s
amd64 / test-upstream-t5x / t5x-multi-node (4, 1)
4h 22m 22s
amd64 / test-upstream-t5x / t5x-multi-node (4, 2)
4h 31m 40s
amd64 / test-upstream-t5x / t5x-multi-node (8, 1)
4h 36m 2s
amd64 / test-upstream-t5x / t5x-multi-node (8, 2)
4h 19m 38s
amd64 / test-rosetta-pax / single-process-multi-device-te (1, 8, 1, 1)
4h 42m 2s
amd64 / test-rosetta-pax / rosetta-pax-multi-node-te (1, 1, 1, 1)
4h 29m 17s
amd64 / test-rosetta-pax / single-process-multi-device-te (1, 1, 2, 4)
5h 1m 36s
amd64 / test-rosetta-pax / rosetta-pax-multi-node (1, 8, 1, 1)
4h 33m 53s
amd64 / test-rosetta-pax / rosetta-pax-multi-node-te (1, 8, 1, 1)
4h 51m 13s
amd64 / test-rosetta-pax / rosetta-pax-multi-node (1, 4, 1, 2)
4h 44m 0s
amd64 / test-rosetta-pax / rosetta-pax-multi-node-te (1, 1, 8, 1)
4h 48m 25s
amd64 / test-rosetta-pax / rosetta-pax-multi-node (4, 2, 1, 1)
5h 6m 9s
amd64 / test-rosetta-pax / rosetta-pax-multi-node-te (1, 4, 1, 2)
5h 6m 13s
amd64 / test-rosetta-pax / rosetta-pax-single-node-dropout-te (1, 8, 1, 1)
4h 46m 49s
amd64 / test-rosetta-pax / rosetta-pax-multi-node (4, 2, 1, 2)
4h 58m 3s
amd64 / test-rosetta-pax / rosetta-pax-multi-node-te (1, 16, 1, 1)
4h 38m 47s
amd64 / test-rosetta-pax / single-process-evaluation-te (1, 8, 1, 1)
5h 0m 35s
amd64 / test-rosetta-t5x / multi-gpu-multi-node (1N1G-te-1, 1, 1, --gin.train/utils.DatasetConfig.pack=False --gin.train_eva...
5h 3m 20s
amd64 / test-rosetta-t5x / multi-gpu-multi-node (1N8G-te-1, 8, 1, --gin.train/utils.DatasetConfig.pack=False --gin.train_eva...
5h 24m 45s
amd64 / test-rosetta-t5x / multi-gpu-multi-node (2N8G-te-1, 8, 2, --gin.train/utils.DatasetConfig.pack=False --gin.train_eva...
5h 40m 39s
amd64 / test-rosetta-t5x / vit-multi-gpu-multi-node (1, 1)
5h 4m 39s
amd64 / test-rosetta-t5x / vit-multi-gpu-multi-node (1, 2)
5h 44m 17s
amd64 / test-rosetta-t5x / vit-single-process-multi-device (8)
5h 28m 56s
amd64 / test-rosetta-t5x / multi-gpu-multi-node (2N2G_te-0, 2, 2, --enable-te 0)
5h 19m 50s
amd64 / test-rosetta-t5x / vit-multi-gpu-multi-node (8, 1)
5h 16m 54s
amd64 / test-rosetta-t5x / vit-multi-gpu-multi-node (8, 2)
5h 11m 13s
amd64 / test-rosetta-t5x / single-process-multi-device (1P1G_te-1, 1, --gin.train/utils.DatasetConfig.pack=False --gin.train...
5h 47m 0s
amd64 / test-rosetta-t5x / single-process-multi-device (1P1G_te-0, 1, --enable-te 0)
5h 27m 31s
amd64 / test-rosetta-t5x / single-process-multi-device (1P8G_te-1, 8, --gin.train/utils.DatasetConfig.pack=False --gin.train...
5h 47m 40s
amd64 / test-pallas / pallas-unit-test (V100)
2m 31s
amd64 / test-pallas / pallas-unit-test (A100)
2m 55s
amd64 / test-pallas / jax-unit-test
0s
amd64 / test-pallas / runner / launch-slurm-runner
5h 48m 33s
amd64 / test-upstream-pax / summary
1s
amd64 / test-upstream-pax / metrics
20s
amd64 / test-upstream-pax / sitrep / sitrep
8s
amd64 / test-upstream-pax / outcome
1s
amd64 / test-upstream-t5x / metrics
37s
amd64 / test-upstream-t5x / summary
1s
amd64 / test-upstream-t5x / sitrep / sitrep
50s
amd64 / test-upstream-t5x / outcome
1s
amd64 / test-rosetta-pax / summary
1s
amd64 / test-rosetta-pax / metrics
1m 23s
amd64 / test-rosetta-pax / publish-test / publish
12s
amd64 / test-rosetta-pax / outcome
1s
amd64 / test-rosetta-t5x / summary
1s
amd64 / test-rosetta-t5x / publish-test / publish
14s
amd64 / test-rosetta-t5x / outcome
1s
publish (ghcr.io/nvidia/jax-toolbox-internal:7825968984-upstream-t5x-amd64, ghcr.io/nvidia/jax-to... / publish
12s
publish (ghcr.io/nvidia/jax-toolbox-internal:7825968984-pax-amd64, ghcr.io/nvidia/jax-toolbox-int... / publish
17s
publish (ghcr.io/nvidia/jax-toolbox-internal:7825968984-t5x-amd64, ghcr.io/nvidia/jax-toolbox-int... / publish
25s
publish (ghcr.io/nvidia/jax-toolbox-internal:7825968984-upstream-pax-amd64, ghcr.io/nvidia/jax-to... / publish
24s
9d 20h 50m 12s