Skip to content
Back to pull request #1236

Only set CUDA_DEVICE_MAX_CONNECTIONS=1 for Hopper/cc9.0 runs #3699

Sign in to view logs

Only set CUDA_DEVICE_MAX_CONNECTIONS=1 for Hopper/cc9.0 runs

Only set CUDA_DEVICE_MAX_CONNECTIONS=1 for Hopper/cc9.0 runs #3699

Run time

Learn about OS pricing on GitHub Actions
Job Run time
metadata
1s
bump-manifest
18s
arm64 / test-distribution (extra-only-distribution.sh)
7s
arm64 / test-distribution (mirror-only-distribution.sh)
7s
arm64 / test-distribution (upstream-only-distribution.sh)
5s
arm64 / test-distribution (local-patch-distribution.sh)
6s
amd64 / test-distribution (extra-only-distribution.sh)
10s
amd64 / test-distribution (mirror-only-distribution.sh)
7s
amd64 / test-distribution (upstream-only-distribution.sh)
5s
amd64 / test-distribution (local-patch-distribution.sh)
9s
arm64 / build-base / build-base
4m 36s
amd64 / build-base / build-base
4m 15s
arm64 / build-jax / build-jax
17m 25s
arm64 / build-triton
0s
arm64 / test-jax
0s
arm64 / test-nsys-jax
0s
arm64 / build-gemma
0s
arm64 / test-nsys-jax-eks
0s
arm64 / build-maxtext / build-maxtext
15m 5s
arm64 / build-levanter / build-levanter
6m 43s
arm64 / build-equinox / build-equinox
4m 47s
arm64 / build-upstream-pax / build-upstream-pax
21m 30s
arm64 / build-upstream-t5x / build-upstream-t5x
14m 53s
arm64 / test-triton
0s
arm64 / test-nsys-jax-archive
0s
arm64 / test-gemma
0s
arm64 / test-levanter
0s
amd64 / build-jax / build-jax
10m 35s
arm64 / test-maxtext
0s
arm64 / test-upstream-t5x
0s
arm64 / build-rosetta-t5x / build-rosetta
17m 11s
amd64 / test-nsys-jax-eks
5m 26s
amd64 / build-equinox / build-equinox
4m 28s
amd64 / build-maxtext / build-maxtext
13m 37s
amd64 / build-levanter / build-levanter
5m 24s
amd64 / build-triton / build-triton
19m 57s
amd64 / build-upstream-pax / build-upstream-pax
18m 37s
amd64 / build-upstream-t5x / build-upstream-t5x
13m 54s
amd64 / build-gemma / build-gemma
12m 51s
amd64 / test-jax / jax-V100-unit-test
35m 18s
amd64 / test-jax / jax-A100-unit-test
16m 38s
amd64 / test-nsys-jax / nsys-jax-V100-unit-test
7m 15s
amd64 / test-nsys-jax / nsys-jax-A100-unit-test
7m 36s
amd64 / test-jax / runner / launch-slurm-runner
6h 0m 14s
amd64 / test-nsys-jax / runner / launch-slurm-runner
5h 56m 42s
arm64 / test-upstream-pax
0s
arm64 / test-te-multigpu
0s
arm64 / test-te
0s
arm64 / build-rosetta-pax / build-rosetta
10m 3s
arm64 / test-rosetta-pax
0s
arm64 / collect-docker-tags
1s
arm64 / test-rosetta-t5x
0s
amd64 / test-triton / triton-V100-unit-test
3m 11s
amd64 / test-triton / triton-A100-unit-test
3m 41s
amd64 / test-triton / runner / launch-slurm-runner
2h 48m 56s
amd64 / test-levanter / levanter-V100-unit-test
12m 32s
amd64 / test-levanter / levanter-A100-unit-test
10m 46s
amd64 / test-levanter / runner / launch-slurm-runner
3h 10m 51s
amd64 / test-maxtext / single-process-multi-device (1, 1, 2, 4)
3h 11m 19s
amd64 / test-maxtext / maxtext-multinode (1, 1, 1, 1)
3h 9m 46s
amd64 / test-maxtext / maxtext-multinode (1, 1, 8, 1)
2h 40m 55s
amd64 / test-maxtext / maxtext-multinode (1, 1, 1, 8)
3h 43m 41s
amd64 / test-maxtext / maxtext-multinode (1, 1, 4, 2)
3h 22m 36s
amd64 / test-maxtext / maxtext-multinode (1, 2, 2, 2)
2h 26m 24s
amd64 / test-maxtext / maxtext-multinode (1, 4, 2, 2)
6h 0m 13s
amd64 / build-rosetta-pax / build-rosetta
8m 17s
amd64 / test-upstream-pax / single-process-multi-device (1, 8, 1, 1)
3h 11m 23s
amd64 / test-upstream-pax / single-process-multi-device (1, 1, 2, 4)
2h 20m 22s
amd64 / test-te / te-V100-unit-test
59m 3s
amd64 / test-te / te-A100-unit-test
1h 10m 28s
amd64 / test-upstream-pax / pax-multi-node (1DP1FSDP1TP1PP, 1, 1, 1, 1, 4)
3h 12m 10s
amd64 / test-upstream-pax / pax-multi-node (8DP1FSDP1TP1PP, 1, 8, 1, 1, 4)
2h 17m 42s
amd64 / test-upstream-pax / pax-multi-node (1DP8FSDP1TP1PP, 1, 1, 8, 1, 4)
3h 44m 49s
amd64 / test-upstream-pax / single-process-evaluation (1, 8, 1, 1)
3h 11m 56s
amd64 / test-upstream-pax / pax-multi-node (2DP1FSDP1TP4PP, 4, 2, 1, 1, 4)
3h 23m 56s
amd64 / test-upstream-pax / pax-multi-node (4DP1FSDP2TP1PP, 1, 4, 1, 2, 4)
2h 24m 0s
amd64 / test-upstream-pax / pax-multi-node (16DP1FSDP1TP1PP, 1, 16, 1, 1, 4)
6h 0m 11s
amd64 / test-upstream-pax / pax-multi-node (2DP1FSDP2TP4PP, 4, 2, 1, 2, 4)
6h 0m 10s
amd64 / test-upstream-pax / pax-multi-node (LLaMA_eval, 1, 1, 8, 1, 4, true, --model-type LLaMA70BProxy --evaluate)
3m 57s
amd64 / test-te / runner / launch-slurm-runner
3h 18m 14s
amd64 / test-te-multigpu / te-multi-gpu (2) / te-2GPU
3h 21m 43s
amd64 / test-te-multigpu / te-multi-gpu (4) / te-4GPU
3h 23m 8s
amd64 / test-te-multigpu / te-multi-gpu (8) / te-8GPU
3h 25m 51s
amd64 / build-rosetta-t5x / build-rosetta
14m 27s
amd64 / test-upstream-t5x / t5x-multi-node (8G2N, 8, 2)
6h 0m 10s
amd64 / test-upstream-t5x / t5x-multi-node (8G2N_fmha, 8, 2, --enable-fmha 1)
6h 0m 14s
amd64 / test-upstream-t5x / t5x-multi-gpu (1P8G, 8)
3h 10m 36s
amd64 / test-upstream-t5x / t5x-multi-gpu (1P2G_fmha, 2, --enable-fmha 1)
2h 11m 22s
amd64 / test-gemma / gemma-V100-unit-test
4m 19s
amd64 / test-gemma / gemma-A100-unit-test
1s
amd64 / test-gemma / runner / launch-slurm-runner
4h 5m 22s
amd64 / test-rosetta-pax / single-process-multi-device-te (1, 8, 1, 1)
5h 52m 26s
amd64 / test-rosetta-pax / rosetta-pax-multi-node-te (1DP1FSDP1TP1PP_TE, 1, 1, 1, 1, 4)
5h 54m 22s
amd64 / test-rosetta-pax / single-process-multi-device-te (1, 1, 2, 4)
1h 23m 10s
amd64 / test-rosetta-pax / rosetta-pax-multi-node-te (8DP1FSDP1TP1PP_TE, 1, 8, 1, 1, 4)
1h 25m 9s
amd64 / test-rosetta-pax / rosetta-pax-multi-node (1, 8, 1, 1)
5h 53m 33s
amd64 / test-rosetta-pax / rosetta-pax-multi-node (1, 4, 1, 2)
1h 23m 57s
amd64 / test-rosetta-pax / rosetta-pax-multi-node-te (1DP8FSDP1TP1PP_TE, 1, 1, 8, 1, 4)
3h 48m 18s
amd64 / test-rosetta-pax / rosetta-pax-multi-node (4, 2, 1, 1)
3h 51m 23s
amd64 / test-rosetta-pax / rosetta-pax-multi-node (4, 2, 1, 2)
6h 0m 12s
amd64 / test-rosetta-pax / rosetta-pax-multi-node-te (4DP1FSDP2TP1PP_TE, 1, 4, 1, 2, 4)
2h 22m 1s
amd64 / test-rosetta-pax / rosetta-pax-multi-node-te (16DP1FSDP1TP1PP_TE, 1, 16, 1, 1, 4)
6h 0m 15s
amd64 / test-rosetta-pax / rosetta-pax-multi-node-te (5B_fused_attn_1, 1, 1, 8, 1, 2, --model-type 5B)
43m 40s
amd64 / test-rosetta-pax / rosetta-pax-multi-node-te (5B_fused_attn_0, 1, 1, 8, 1, 2, --model-type 5B --disable-fused-attn)
44m 48s
amd64 / test-rosetta-pax / rosetta-pax-multi-node-te (LLaMA_eval_TE, 1, 1, 8, 1, 4, true, --model-type LLaMA70BProxy --evalu...
24m 48s
amd64 / test-rosetta-pax / rosetta-pax-single-node-dropout-te (1, 8, 1, 1)
5h 52m 15s
amd64 / test-rosetta-pax / single-process-evaluation-te (1, 8, 1, 1)
5h 31m 3s
amd64 / collect-docker-tags
1s
amd64 / test-rosetta-t5x / single-process-multi-device (1P1G_te-0, 1, --enable-te 0)
5h 36m 35s
amd64 / test-rosetta-t5x / single-process-multi-device (1P8G_te-1, 8, --gin.train/utils.DatasetConfig.pack=False --gin.train...
1h 27m 19s
amd64 / test-rosetta-t5x / multi-gpu-multi-node (2N8G-te-1, 8, 2, --gin.train/utils.DatasetConfig.pack=False --gin.train_eva...
6h 0m 14s
amd64 / test-rosetta-t5x / multi-gpu-multi-node (2N2G_te-0, 2, 2, --enable-te 0)
6h 0m 11s
amd64 / test-rosetta-t5x / vit-single-process-multi-device (8)
5h 43m 42s
amd64 / test-rosetta-t5x / vit-multi-gpu-multi-node (1, 1)
5h 37m 22s
amd64 / test-rosetta-t5x / vit-multi-gpu-multi-node (1, 2)
6h 0m 12s
amd64 / test-rosetta-t5x / vit-multi-gpu-multi-node (8, 1)
2h 11m 41s
amd64 / test-rosetta-t5x / vit-multi-gpu-multi-node (8, 2)
6h 0m 12s
amd64 / test-nsys-jax-archive (ubuntu-22.04)
2m 36s
amd64 / test-nsys-jax-archive (ubuntu-24.04)
2m 41s
amd64 / test-nsys-jax-archive (macOS-latest)
2m 45s
amd64 / test-te-multigpu / sitrep
8s
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
12s
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
13s
amd64 / test-rosetta-t5x / test-t5x-rosetta-outcome
1s
amd64 / test-maxtext / test-maxtext-summary
1s
amd64 / test-maxtext / test-maxtext-metrics
0s
amd64 / test-maxtext / test-maxtext-sitrep / sitrep
22s
amd64 / test-maxtext / test-maxtext-outcome
1s
amd64 / test-rosetta-pax / test-pax-rosetta-summary
1s
amd64 / test-rosetta-pax / test-pax-rosetta-metrics
0s
amd64 / test-rosetta-pax / test-pax-rosetta-sitrep / sitrep
7s
amd64 / test-rosetta-pax / test-pax-rosetta-outcome
1s
amd64 / test-upstream-pax / test-upstream-pax-summary
1s
amd64 / test-upstream-pax / test-upstream-pax-metrics
0s
amd64 / test-upstream-pax / test-upstream-pax-sitrep / sitrep
13s
amd64 / test-upstream-pax / test-upstream-pax-outcome
1s
make-publish-configs
3s
merge-new-manifest
0s
publish-containers (equinox, mock-jax-mealkit, 500, ghcr.io/nvidia/jax-toolbox-internal:126933732... / publish
17s
publish-containers (gemma, mock-jax, 900, ghcr.io/nvidia/jax-toolbox-internal:12693373254-gemma-a... / publish
10s
publish-containers (base, mock-jax, 800, ghcr.io/nvidia/jax-toolbox-internal:12693373254-base-amd... / publish
21s
publish-containers (jax, mock-jax, 1000, ghcr.io/nvidia/jax-toolbox-internal:12693373254-jax-amd6... / publish
25s
publish-containers (equinox, mock-jax, 900, ghcr.io/nvidia/jax-toolbox-internal:12693373254-equin... / publish
12s
publish-containers (jax, mock-jax-mealkit, 500, ghcr.io/nvidia/jax-toolbox-internal:12693373254-j... / publish
16s
publish-containers (gemma, mock-jax-mealkit, 500, ghcr.io/nvidia/jax-toolbox-internal:12693373254... / publish
20s
publish-containers (levanter, mock-jax-mealkit, 500, ghcr.io/nvidia/jax-toolbox-internal:12693373... / publish
26s
publish-containers (maxtext, mock-jax, 900, ghcr.io/nvidia/jax-toolbox-internal:12693373254-maxte... / publish
27s
publish-containers (levanter, mock-jax, 900, ghcr.io/nvidia/jax-toolbox-internal:12693373254-leva... / publish
26s
publish-containers (maxtext, mock-jax-mealkit, 500, ghcr.io/nvidia/jax-toolbox-internal:126933732... / publish
19s
publish-containers (pax, mock-jax-mealkit, 500, ghcr.io/nvidia/jax-toolbox-internal:12693373254-p... / publish
16s
publish-containers (t5x, mock-jax, 900, ghcr.io/nvidia/jax-toolbox-internal:12693373254-t5x-amd64... / publish
16s
publish-containers (pax, mock-jax, 900, ghcr.io/nvidia/jax-toolbox-internal:12693373254-pax-amd64... / publish
15s
publish-containers (triton, mock-jax, 900, ghcr.io/nvidia/jax-toolbox-internal:12693373254-triton... / publish
20s
publish-containers (triton, mock-jax-mealkit, 500, ghcr.io/nvidia/jax-toolbox-internal:1269337325... / publish
9s
publish-containers (t5x, mock-jax-mealkit, 500, ghcr.io/nvidia/jax-toolbox-internal:12693373254-t... / publish
24s
publish-containers (upstream-pax, mock-jax, 900, ghcr.io/nvidia/jax-toolbox-internal:12693373254-... / publish
27s
publish-containers (upstream-pax, mock-jax-mealkit, 500, ghcr.io/nvidia/jax-toolbox-internal:1269... / publish
25s
publish-containers (upstream-t5x, mock-jax-mealkit, 500, ghcr.io/nvidia/jax-toolbox-internal:1269... / publish
12s
publish-containers (upstream-t5x, mock-jax, 900, ghcr.io/nvidia/jax-toolbox-internal:12693373254-... / publish
19s
finalize / report
13s
finalize / workflow-badge
6s
finalize / upload-badge
7s
finalize / publish-badge
4s
9d 7h 20m 29s