CI: run MaxText tests on AWS with NGC release candidate images #3724

Run time

Learn about OS pricing on GitHub Actions

Job	Run time
metadata	1s
bump-manifest	38s
arm64 / test-distribution (extra-only-distribution.sh)	10s
amd64 / test-distribution (extra-only-distribution.sh)	6s
amd64 / test-distribution (mirror-only-distribution.sh)	7s
arm64 / test-distribution (mirror-only-distribution.sh)	7s
amd64 / test-distribution (upstream-only-distribution.sh)	5s
arm64 / test-distribution (upstream-only-distribution.sh)	8s
amd64 / test-distribution (local-patch-distribution.sh)	5s
arm64 / test-distribution (local-patch-distribution.sh)	7s
arm64 / build-base / build-base	4m 38s
amd64 / build-base / build-base	4m 49s
arm64 / build-jax / build-jax	18m 11s
amd64 / build-jax / build-jax	10m 43s
amd64 / test-nsys-jax-eks	5m 19s
amd64 / build-equinox / build-equinox	3m 40s
amd64 / build-triton / build-triton	19m 44s
amd64 / build-levanter / build-levanter	5m 8s
amd64 / build-maxtext / build-maxtext	13m 36s
amd64 / build-upstream-pax / build-upstream-pax	9m 21s
amd64 / build-upstream-t5x / build-upstream-t5x	13m 4s
amd64 / build-gemma / build-gemma	11m 48s
amd64 / test-jax / jax-V100-unit-test	34m 56s
amd64 / test-jax / jax-A100-unit-test	17m 9s
amd64 / test-nsys-jax / nsys-jax-V100-unit-test	7m 11s
amd64 / test-nsys-jax / nsys-jax-A100-unit-test	7m 47s
amd64 / test-jax / runner / launch-slurm-runner	19m 54s
amd64 / test-nsys-jax / runner / launch-slurm-runner	28m 41s
amd64 / test-levanter / levanter-V100-unit-test	12m 42s
amd64 / test-levanter / levanter-A100-unit-test	10m 46s
amd64 / test-levanter / runner / launch-slurm-runner	34m 58s
arm64 / build-triton	0s
arm64 / build-gemma	0s
arm64 / test-jax	0s
arm64 / test-nsys-jax	0s
arm64 / test-nsys-jax-eks	0s
arm64 / build-levanter / build-levanter	7m 0s
arm64 / build-maxtext / build-maxtext	16m 23s
arm64 / build-equinox / build-equinox	5m 12s
arm64 / build-upstream-pax / build-upstream-pax	10m 54s
arm64 / build-upstream-t5x / build-upstream-t5x	15m 54s
arm64 / test-triton	0s
arm64 / test-gemma	0s
arm64 / test-nsys-jax-archive	0s
amd64 / test-te	0s
amd64 / test-upstream-pax	0s
amd64 / test-te-multigpu	0s
amd64 / build-rosetta-pax	0s
amd64 / test-rosetta-pax	0s
amd64 / test-gemma / gemma-V100-unit-test	4m 25s
amd64 / test-gemma / gemma-A100-unit-test	4m 41s
amd64 / test-gemma / runner / launch-slurm-runner	39m 37s
arm64 / test-levanter	0s
amd64 / build-rosetta-t5x / build-rosetta	15m 32s
amd64 / test-upstream-t5x / t5x-multi-node (8G2N, 8, 2)	6h 0m 11s
amd64 / test-upstream-t5x / t5x-multi-node (8G2N_fmha, 8, 2, --enable-fmha 1)	2m 4s
amd64 / test-upstream-t5x / t5x-multi-gpu (1P8G, 8)	42m 54s
amd64 / test-upstream-t5x / t5x-multi-gpu (1P2G_fmha, 2, --enable-fmha 1)	28m 54s
amd64 / test-maxtext / single-process-multi-device (1, 1, 2, 4)	46m 57s
amd64 / test-maxtext / maxtext-multinode (1, 1, 1, 1)	45m 4s
amd64 / test-maxtext / maxtext-multinode (1, 1, 8, 1)	28m 21s
amd64 / test-maxtext / maxtext-multinode (1, 1, 1, 8)	1h 14m 39s
amd64 / test-maxtext / maxtext-multinode (1, 1, 4, 2)	2h 24m 49s
amd64 / test-maxtext / maxtext-multinode (1, 2, 2, 2)	43m 9s
amd64 / test-maxtext / maxtext-multinode (1, 4, 2, 2)	5m 1s
arm64 / build-rosetta-pax	0s
arm64 / test-te	0s
arm64 / test-te-multigpu	0s
arm64 / test-upstream-pax	0s
arm64 / test-rosetta-pax	0s
amd64 / test-triton / triton-V100-unit-test	3m 14s
amd64 / test-triton / triton-A100-unit-test	3m 56s
amd64 / test-triton / runner / launch-slurm-runner	43m 43s
arm64 / test-upstream-t5x	0s
arm64 / build-rosetta-t5x / build-rosetta	17m 8s
arm64 / test-maxtext	0s
amd64 / collect-docker-tags	1s
amd64 / test-rosetta-t5x / multi-gpu-multi-node (2N8G-te-1, 8, 2, --gin.train/utils.DatasetConfig.pack=False --gin.train_eva...	6h 0m 10s
amd64 / test-rosetta-t5x / single-process-multi-device (1P1G_te-0, 1, --enable-te 0)	50m 16s
amd64 / test-rosetta-t5x / single-process-multi-device (1P8G_te-1, 8, --gin.train/utils.DatasetConfig.pack=False --gin.train...	1h 15m 31s
amd64 / test-rosetta-t5x / multi-gpu-multi-node (2N2G_te-0, 2, 2, --enable-te 0)	6h 0m 15s
amd64 / test-rosetta-t5x / vit-multi-gpu-multi-node (1, 1)	54m 38s
amd64 / test-rosetta-t5x / vit-multi-gpu-multi-node (1, 2)	6h 0m 12s
amd64 / test-rosetta-t5x / vit-single-process-multi-device (8)	48m 22s
amd64 / test-rosetta-t5x / vit-multi-gpu-multi-node (8, 1)	8m 3s
amd64 / test-rosetta-t5x / vit-multi-gpu-multi-node (8, 2)	6h 0m 13s
arm64 / collect-docker-tags	1s
arm64 / test-rosetta-t5x	0s
amd64 / test-nsys-jax-archive (ubuntu-22.04)	2m 41s
amd64 / test-nsys-jax-archive (ubuntu-24.04)	2m 43s
amd64 / test-nsys-jax-archive (macOS-latest)	3m 14s
amd64 / test-maxtext / test-maxtext-summary	1s
amd64 / test-maxtext / test-maxtext-metrics	0s
amd64 / test-maxtext / test-maxtext-sitrep / sitrep	5m 3s
amd64 / test-maxtext / test-maxtext-outcome	1s
amd64 / test-upstream-t5x / test-upstream-t5x-summary	1s
amd64 / test-upstream-t5x / test-upstream-t5x-metrics	0s
amd64 / test-upstream-t5x / test-upstream-t5x-sitrep / sitrep	3m 17s
amd64 / test-upstream-t5x / test-upstream-t5x-outcome	1s
amd64 / test-rosetta-t5x / test-t5x-rosetta-summary	1s
amd64 / test-rosetta-t5x / test-t5x-rosetta-metrics	0s
amd64 / test-rosetta-t5x / test-t5x-rosetta-sitrep / sitrep	11s
amd64 / test-rosetta-t5x / test-t5x-rosetta-outcome	1s
make-publish-configs	4s
merge-new-manifest	0s
publish-containers (equinox, mock-jax-mealkit, 500, ghcr.io/nvidia/jax-toolbox-internal:127528243... / publish	16s
publish-containers (equinox, mock-jax, 900, ghcr.io/nvidia/jax-toolbox-internal:12752824313-equin... / publish	17s
publish-containers (gemma, mock-jax, 900, ghcr.io/nvidia/jax-toolbox-internal:12752824313-gemma-a... / publish	24s
publish-containers (gemma, mock-jax-mealkit, 500, ghcr.io/nvidia/jax-toolbox-internal:12752824313... / publish	10s
publish-containers (base, mock-jax, 800, ghcr.io/nvidia/jax-toolbox-internal:12752824313-base-amd... / publish	13s
publish-containers (jax, mock-jax, 1000, ghcr.io/nvidia/jax-toolbox-internal:12752824313-jax-amd6... / publish	14s
publish-containers (levanter, mock-jax-mealkit, 500, ghcr.io/nvidia/jax-toolbox-internal:12752824... / publish	13s
publish-containers (jax, mock-jax-mealkit, 500, ghcr.io/nvidia/jax-toolbox-internal:12752824313-j... / publish	12s
publish-containers (levanter, mock-jax, 900, ghcr.io/nvidia/jax-toolbox-internal:12752824313-leva... / publish	25s
publish-containers (maxtext, mock-jax, 900, ghcr.io/nvidia/jax-toolbox-internal:12752824313-maxte... / publish	13s
publish-containers (t5x, mock-jax-mealkit, 500, ghcr.io/nvidia/jax-toolbox-internal:12752824313-t... / publish	13s
publish-containers (t5x, mock-jax, 900, ghcr.io/nvidia/jax-toolbox-internal:12752824313-t5x-amd64... / publish	13s
publish-containers (maxtext, mock-jax-mealkit, 500, ghcr.io/nvidia/jax-toolbox-internal:127528243... / publish	14s
publish-containers (triton, mock-jax, 900, ghcr.io/nvidia/jax-toolbox-internal:12752824313-triton... / publish	17s
publish-containers (triton, mock-jax-mealkit, 500, ghcr.io/nvidia/jax-toolbox-internal:1275282431... / publish	14s
publish-containers (upstream-t5x, mock-jax, 900, ghcr.io/nvidia/jax-toolbox-internal:12752824313-... / publish	12s
publish-containers (upstream-t5x, mock-jax-mealkit, 500, ghcr.io/nvidia/jax-toolbox-internal:1275... / publish	13s
finalize / report	8s
finalize / workflow-badge	4s
finalize / upload-badge	4s
finalize / publish-badge	5s
	2d 2h 4m 56s

Provide feedback

Saved searches

Use saved searches to filter your results more quickly

CI: run MaxText tests on AWS with NGC release candidate images #3724

Usage

CI: run MaxText tests on AWS with NGC release candidate images #3724

Jobs

Run details

Run time