Triton: use CUDA 12.3 tools from the base image #2293

Run time

Learn about OS pricing on GitHub Actions

Job	Run time
metadata	1s
bump-manifest	33s
arm64 / test-distribution (extra-only-distribution.sh)	7s
arm64 / test-distribution (mirror-only-distribution.sh)	5s
arm64 / test-distribution (upstream-only-distribution.sh)	9s
amd64 / test-distribution (extra-only-distribution.sh)	7s
amd64 / test-distribution (mirror-only-distribution.sh)	6s
amd64 / test-distribution (upstream-only-distribution.sh)	5s
arm64 / build-base / build-base	5m 29s
amd64 / build-base / build-base	5m 56s
arm64 / build-jax / build-jax	15m 1s
amd64 / build-jax / build-jax	8m 0s
amd64 / build-equinox / build-equinox	3m 21s
amd64 / build-levanter / build-levanter	4m 59s
amd64 / build-maxtext / build-maxtext	5m 39s
amd64 / build-triton / build-triton	14m 8s
amd64 / build-upstream-t5x / build-upstream-t5x	7m 10s
amd64 / build-grok / build-grok	3m 8s
amd64 / build-upstream-pax / build-upstream-pax	9m 0s
amd64 / test-jax / jax-V100-unit-test	17m 5s
amd64 / test-jax / jax-A100-unit-test	30m 20s
amd64 / test-pallas / pallas-V100-unit-test	4m 42s
amd64 / test-pallas / pallas-A100-unit-test	26m 10s
amd64 / test-jax / runner / launch-slurm-runner	27m 12s
amd64 / test-pallas / runner / launch-slurm-runner	31m 33s
amd64 / test-levanter / levanter-V100-unit-test	13m 24s
amd64 / test-levanter / levanter-A100-unit-test	1h 13m 7s
amd64 / test-levanter / runner / launch-slurm-runner	1h 22m 44s
arm64 / build-maxtext	0s
arm64 / build-triton	0s
arm64 / test-jax	0s
arm64 / test-pallas	0s
arm64 / build-equinox / build-equinox	13m 32s
arm64 / build-levanter / build-levanter	14m 40s
arm64 / build-grok / build-grok	13m 41s
arm64 / build-upstream-pax / build-upstream-pax	26m 10s
arm64 / build-upstream-t5x / build-upstream-t5x	24m 22s
arm64 / test-maxtext	0s
arm64 / test-triton	0s
amd64 / test-maxtext / maxtext-multinode (1, 1, 1, 1)	36m 22s
amd64 / test-maxtext / single-process-multi-device (1, 1, 2, 4)	39m 32s
amd64 / test-maxtext / maxtext-multinode (1, 1, 8, 1)	41m 27s
amd64 / test-maxtext / maxtext-multinode (1, 1, 1, 8)	34m 19s
amd64 / test-maxtext / maxtext-multinode (1, 1, 4, 2)	38m 8s
amd64 / test-maxtext / maxtext-multinode (1, 2, 2, 2)	48m 18s
amd64 / test-maxtext / maxtext-multinode (1, 4, 2, 2)	1h 30m 8s
amd64 / build-rosetta-t5x / build-rosetta	8m 16s
amd64 / test-upstream-t5x / t5x-multi-node (1G1N, 1, 1)	2h 3m 27s
amd64 / test-upstream-t5x / t5x-multi-node (2G1N, 2, 1)	2h 18m 17s
amd64 / test-upstream-t5x / t5x-multi-gpu (1P1G, 1)	2h 3m 36s
amd64 / test-upstream-t5x / t5x-multi-node (4G1N, 4, 1)	2h 10m 15s
amd64 / test-upstream-t5x / t5x-multi-gpu (1P2G, 2)	1h 32m 5s
amd64 / test-upstream-t5x / t5x-multi-node (8G1N, 8, 1)	2h 21m 38s
amd64 / test-upstream-t5x / t5x-multi-gpu (1P4G, 4)	1h 32m 59s
amd64 / test-upstream-t5x / t5x-multi-node (1G2N, 1, 2)	2h 7m 3s
amd64 / test-upstream-t5x / t5x-multi-gpu (1P8G, 8)	2h 19m 14s
amd64 / test-upstream-t5x / t5x-multi-gpu (1P1G_fmha, 1, --enable-fmha 1)	1h 49m 18s
amd64 / test-upstream-t5x / t5x-multi-gpu (1P2G_fmha, 2, --enable-fmha 1)	1h 15m 15s
amd64 / test-upstream-t5x / t5x-multi-node (2G2N, 2, 2)	2h 0m 18s
amd64 / test-upstream-t5x / t5x-multi-node (4G2N, 4, 2)	1h 46m 17s
amd64 / test-upstream-t5x / t5x-multi-node (8G2N, 8, 2)	2h 27m 16s
amd64 / test-upstream-t5x / t5x-multi-node (2G2N_fmha, 2, 2, --enable-fmha 1)	2h 15m 13s
amd64 / test-upstream-t5x / t5x-multi-node (8G2N_fmha, 8, 2, --enable-fmha 1)	2h 30m 44s
amd64 / test-te / te-V100-unit-test	26m 33s
amd64 / test-te / te-A100-unit-test	35m 13s
amd64 / build-rosetta-pax / build-rosetta	6m 46s
amd64 / test-upstream-pax / single-process-multi-device (1, 8, 1, 1)	2h 35m 36s
amd64 / test-upstream-pax / single-process-multi-device (1, 1, 2, 4)	2h 29m 53s
amd64 / test-upstream-pax / single-process-evaluation (1, 8, 1, 1)	2h 32m 18s
amd64 / test-upstream-pax / pax-multi-node (1DP1FSDP1TP1PP, 1, 1, 1, 1, 4)	2h 42m 10s
amd64 / test-upstream-pax / pax-multi-node (8DP1FSDP1TP1PP, 1, 8, 1, 1, 4)	2h 50m 41s
amd64 / test-upstream-pax / pax-multi-node (1DP8FSDP1TP1PP, 1, 1, 8, 1, 4)	2h 38m 41s
amd64 / test-upstream-pax / pax-multi-node (2DP1FSDP1TP4PP, 4, 2, 1, 1, 4)	2h 56m 5s
amd64 / test-upstream-pax / pax-multi-node (4DP1FSDP2TP1PP, 1, 4, 1, 2, 4)	3h 18m 56s
amd64 / test-upstream-pax / pax-multi-node (16DP1FSDP1TP1PP, 1, 16, 1, 1, 4)	3h 15m 18s
amd64 / test-upstream-pax / pax-multi-node (2DP1FSDP2TP4PP, 4, 2, 1, 2, 4)	3h 12m 7s
amd64 / test-upstream-pax / pax-multi-node (LLaMA_eval, 1, 1, 8, 1, 4, true, --model-type LLaMA70BProxy --evaluate)	2h 47m 13s
amd64 / test-te / runner / launch-slurm-runner	3h 1m 36s
amd64 / test-te-multigpu / te-multi-gpu (2) / te-2GPU	2h 6m 43s
amd64 / test-te-multigpu / te-multi-gpu (4) / te-4GPU	1h 37m 40s
amd64 / test-te-multigpu / te-multi-gpu (8) / te-8GPU	1h 51m 42s
amd64 / test-triton / triton-V100-unit-test	3m 58s
amd64 / test-triton / triton-A100-unit-test	11m 46s
amd64 / test-triton / runner / launch-slurm-runner	3h 25m 36s
amd64 / test-rosetta-t5x / vit-multi-gpu-multi-node (1, 1)	3h 26m 42s
amd64 / test-rosetta-t5x / vit-single-process-multi-device (8)	3h 41m 50s
amd64 / test-rosetta-t5x / vit-multi-gpu-multi-node (1, 2)	3h 46m 7s
amd64 / test-rosetta-t5x / single-process-multi-device (1P1G_te-1, 1, --gin.train/utils.DatasetConfig.pack=False --gin.train...	4h 6m 51s
amd64 / test-rosetta-t5x / vit-multi-gpu-multi-node (8, 1)	3h 51m 23s
amd64 / test-rosetta-t5x / single-process-multi-device (1P1G_te-0, 1, --enable-te 0)	3h 28m 54s
amd64 / test-rosetta-t5x / multi-gpu-multi-node (1N1G-te-1, 1, 1, --gin.train/utils.DatasetConfig.pack=False --gin.train_eva...	3h 19m 25s
amd64 / test-rosetta-t5x / vit-multi-gpu-multi-node (8, 2)	4h 4m 16s
amd64 / test-rosetta-t5x / single-process-multi-device (1P8G_te-1, 8, --gin.train/utils.DatasetConfig.pack=False --gin.train...	3h 49m 40s
amd64 / test-rosetta-t5x / multi-gpu-multi-node (1N8G-te-1, 8, 1, --gin.train/utils.DatasetConfig.pack=False --gin.train_eva...	3h 22m 38s
amd64 / test-rosetta-t5x / multi-gpu-multi-node (2N8G-te-1, 8, 2, --gin.train/utils.DatasetConfig.pack=False --gin.train_eva...	3h 58m 50s
amd64 / test-rosetta-t5x / multi-gpu-multi-node (2N2G_te-0, 2, 2, --enable-te 0)	3h 32m 27s
amd64 / collect-docker-tags	1s
amd64 / test-rosetta-pax / rosetta-pax-multi-node-te (1DP1FSDP1TP1PP_TE, 1, 1, 1, 1, 4)	4h 14m 26s
amd64 / test-rosetta-pax / single-process-multi-device-te (1, 8, 1, 1)	4h 4m 36s
amd64 / test-rosetta-pax / rosetta-pax-multi-node-te (8DP1FSDP1TP1PP_TE, 1, 8, 1, 1, 4)	4h 11m 58s
amd64 / test-rosetta-pax / single-process-multi-device-te (1, 1, 2, 4)	4h 17m 56s
amd64 / test-rosetta-pax / rosetta-pax-multi-node-te (1DP8FSDP1TP1PP_TE, 1, 1, 8, 1, 4)	4h 6m 33s
amd64 / test-rosetta-pax / rosetta-pax-multi-node-te (4DP1FSDP2TP1PP_TE, 1, 4, 1, 2, 4)	4h 7m 19s
amd64 / test-rosetta-pax / rosetta-pax-multi-node-te (16DP1FSDP1TP1PP_TE, 1, 16, 1, 1, 4)	4h 25m 53s
amd64 / test-rosetta-pax / rosetta-pax-multi-node-te (5B_fused_attn_1, 1, 1, 8, 1, 2, --model-type 5B --enable-fused-attn)	4h 31m 34s
amd64 / test-rosetta-pax / rosetta-pax-single-node-dropout-te (1, 8, 1, 1)	4h 9m 21s
amd64 / test-rosetta-pax / rosetta-pax-multi-node (1, 8, 1, 1)	4h 36m 50s
amd64 / test-rosetta-pax / rosetta-pax-multi-node-te (5B_fused_attn_0, 1, 1, 8, 1, 2, --model-type 5B)	4h 13m 42s
amd64 / test-rosetta-pax / rosetta-pax-multi-node (1, 4, 1, 2)	4h 16m 33s
amd64 / test-rosetta-pax / rosetta-pax-multi-node-te (LLaMA_eval_TE, 1, 1, 8, 1, 4, true, --model-type LLaMA70BProxy --evalu...	4h 30m 19s
amd64 / test-rosetta-pax / rosetta-pax-multi-node (4, 2, 1, 1)	4h 34m 12s
amd64 / test-rosetta-pax / rosetta-pax-multi-node (4, 2, 1, 2)	4h 23m 8s
amd64 / test-rosetta-pax / single-process-evaluation-te (1, 8, 1, 1)	4h 33m 59s
arm64 / test-levanter	0s
arm64 / test-upstream-pax	0s
arm64 / test-te	0s
arm64 / test-te-multigpu	0s
arm64 / build-rosetta-pax / build-rosetta	16m 47s
arm64 / test-upstream-t5x	0s
arm64 / build-rosetta-t5x / build-rosetta	19m 58s
arm64 / test-rosetta-pax	0s
arm64 / collect-docker-tags	1s
arm64 / test-rosetta-t5x	0s
amd64 / test-maxtext / summary	1s
amd64 / test-maxtext / metrics	23s
amd64 / test-maxtext / sitrep / sitrep	6s
amd64 / test-maxtext / outcome	1s
amd64 / test-te-multigpu / sitrep	9s
amd64 / test-upstream-t5x / summary	1s
amd64 / test-upstream-t5x / metrics	24s
amd64 / test-upstream-t5x / sitrep / sitrep	17s
amd64 / test-upstream-t5x / outcome	1s
amd64 / test-upstream-pax / metrics	21s
amd64 / test-upstream-pax / summary	1s
amd64 / test-upstream-pax / sitrep / sitrep	13s
amd64 / test-upstream-pax / outcome	1s
amd64 / test-rosetta-t5x / metrics	19s
amd64 / test-rosetta-t5x / summary	1s
amd64 / test-rosetta-t5x / sitrep / sitrep	15s
amd64 / test-rosetta-t5x / outcome	1s
amd64 / test-rosetta-pax / metrics	27s
amd64 / test-rosetta-pax / summary	1s
amd64 / test-rosetta-pax / sitrep / sitrep	15s
amd64 / test-rosetta-pax / outcome	1s
make-publish-configs	4s
merge-new-manifest	0s
finalize / workflow-badge	3s
finalize / report	13s
finalize / upload-badge	16s
publish-containers (equinox, mock-jax-mealkit, 500, ghcr.io/nvidia/jax-toolbox-internal:839356643... / publish	14s
publish-containers (equinox, mock-jax, 900, ghcr.io/nvidia/jax-toolbox-internal:8393566433-equino... / publish	12s
publish-containers (base, mock-jax, 800, ghcr.io/nvidia/jax-toolbox-internal:8393566433-base-amd6... / publish	14s
publish-containers (equinox, mock-jax, 900, ghcr.io/nvidia/jax-toolbox-internal:8393566433-equino... / publish	12s
publish-containers (grok, mock-jax, 900, ghcr.io/nvidia/jax-toolbox-internal:8393566433-grok-amd6... / publish	23s
publish-containers (equinox, mock-jax-mealkit, 500, ghcr.io/nvidia/jax-toolbox-internal:839356643... / publish	14s
publish-containers (grok, mock-jax-mealkit, 500, ghcr.io/nvidia/jax-toolbox-internal:8393566433-g... / publish	24s
publish-containers (jax, mock-jax, 1000, ghcr.io/nvidia/jax-toolbox-internal:8393566433-jax-amd64... / publish	13s
publish-containers (maxtext, mock-jax, 900, ghcr.io/nvidia/jax-toolbox-internal:8393566433-maxtex... / publish	11s
publish-containers (jax, mock-jax-mealkit, 500, ghcr.io/nvidia/jax-toolbox-internal:8393566433-ja... / publish	12s
publish-containers (levanter, mock-jax, 900, ghcr.io/nvidia/jax-toolbox-internal:8393566433-levan... / publish	13s
publish-containers (levanter, mock-jax-mealkit, 500, ghcr.io/nvidia/jax-toolbox-internal:83935664... / publish	14s
publish-containers (maxtext, mock-jax-mealkit, 500, ghcr.io/nvidia/jax-toolbox-internal:839356643... / publish	19s
publish-containers (pax, mock-jax, 900, ghcr.io/nvidia/jax-toolbox-internal:8393566433-pax-amd64,... / publish	12s
publish-containers (pax, mock-jax-mealkit, 500, ghcr.io/nvidia/jax-toolbox-internal:8393566433-pa... / publish	26s
publish-containers (t5x, mock-jax, 900, ghcr.io/nvidia/jax-toolbox-internal:8393566433-t5x-amd64,... / publish	12s
publish-containers (t5x, mock-jax-mealkit, 500, ghcr.io/nvidia/jax-toolbox-internal:8393566433-t5... / publish	17s
publish-containers (triton, mock-jax, 900, ghcr.io/nvidia/jax-toolbox-internal:8393566433-triton-... / publish	12s
publish-containers (triton, mock-jax-mealkit, 500, ghcr.io/nvidia/jax-toolbox-internal:8393566433... / publish	13s
publish-containers (upstream-pax, mock-jax, 900, ghcr.io/nvidia/jax-toolbox-internal:8393566433-u... / publish	12s
publish-containers (upstream-pax, mock-jax-mealkit, 500, ghcr.io/nvidia/jax-toolbox-internal:8393... / publish	25s
publish-containers (upstream-t5x, mock-jax, 900, ghcr.io/nvidia/jax-toolbox-internal:8393566433-u... / publish	17s
publish-containers (upstream-t5x, mock-jax-mealkit, 500, ghcr.io/nvidia/jax-toolbox-internal:8393... / publish	15s
finalize / publish-badge	2s
	8d 13h 31m 44s

Provide feedback

Saved searches

Use saved searches to filter your results more quickly

Triton: use CUDA 12.3 tools from the base image #2293

Usage

Triton: use CUDA 12.3 tools from the base image #2293

Jobs

Run details

Run time