Skip to content
CI

Add FMHA T5x test #1950

Sign in to view logs

Add FMHA T5x test

Add FMHA T5x test #1950

Run time

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