Skip to content
Back to pull request #757

Enable TE fused attention by default in rosetta pax container #2560

Sign in to view logs

Enable TE fused attention by default in rosetta pax container

Enable TE fused attention by default in rosetta pax container #2560

Run time

Learn about OS pricing on GitHub Actions
Job Run time
metadata
1s
bump-manifest
21s
arm64 / test-distribution (extra-only-distribution.sh)
10s
amd64 / test-distribution (extra-only-distribution.sh)
8s
arm64 / test-distribution (mirror-only-distribution.sh)
7s
amd64 / test-distribution (mirror-only-distribution.sh)
9s
arm64 / test-distribution (upstream-only-distribution.sh)
7s
amd64 / test-distribution (upstream-only-distribution.sh)
10s
arm64 / test-distribution (local-patch-distribution.sh)
8s
amd64 / test-distribution (local-patch-distribution.sh)
9s
arm64 / build-base / build-base
6m 42s
amd64 / build-base / build-base
6m 33s
amd64 / build-jax / build-jax
7m 53s
arm64 / build-jax / build-jax
14m 10s
amd64 / build-equinox / build-equinox
11m 17s
amd64 / build-maxtext / build-maxtext
5m 7s
amd64 / build-levanter / build-levanter
4m 14s
amd64 / build-triton / build-triton
14m 30s
amd64 / build-grok / build-grok
3m 8s
amd64 / build-upstream-t5x / build-upstream-t5x
2m 14s
amd64 / build-upstream-pax / build-upstream-pax
6m 6s
amd64 / test-jax / jax-V100-unit-test
24m 46s
amd64 / test-jax / jax-A100-unit-test
14m 23s
amd64 / test-pallas / pallas-V100-unit-test
5m 1s
amd64 / test-pallas / pallas-A100-unit-test
5m 36s
amd64 / test-jax / runner / launch-slurm-runner
4h 59m 5s
amd64 / test-pallas / runner / launch-slurm-runner
4h 52m 46s
amd64 / test-levanter / levanter-V100-unit-test
14m 32s
amd64 / test-levanter / levanter-A100-unit-test
10m 57s
amd64 / test-levanter / runner / launch-slurm-runner
4h 58m 13s
arm64 / build-maxtext
0s
arm64 / build-triton
0s
arm64 / test-jax
0s
arm64 / test-pallas
0s
arm64 / build-equinox / build-equinox
6m 8s
arm64 / build-levanter / build-levanter
8m 24s
arm64 / build-grok / build-grok
6m 13s
arm64 / build-upstream-t5x / build-upstream-t5x
2m 44s
arm64 / build-upstream-pax / build-upstream-pax
28m 2s
arm64 / test-maxtext
0s
arm64 / test-triton
0s
amd64 / test-maxtext / single-process-multi-device (1, 1, 2, 4)
4h 11m 17s
amd64 / test-maxtext / maxtext-multinode (1, 1, 1, 1)
4h 31m 53s
amd64 / test-maxtext / maxtext-multinode (1, 1, 8, 1)
3h 22m 15s
amd64 / test-maxtext / maxtext-multinode (1, 1, 1, 8)
4h 27m 19s
amd64 / test-maxtext / maxtext-multinode (1, 1, 4, 2)
3h 41m 18s
amd64 / test-maxtext / maxtext-multinode (1, 2, 2, 2)
4h 12m 57s
amd64 / test-maxtext / maxtext-multinode (1, 4, 2, 2)
5h 1m 31s
amd64 / test-upstream-t5x
0s
amd64 / build-rosetta-t5x
0s
amd64 / test-rosetta-t5x
0s
arm64 / build-rosetta-t5x
0s
arm64 / test-upstream-t5x
0s
arm64 / test-rosetta-t5x
0s
arm64 / test-levanter
0s
amd64 / test-upstream-pax / single-process-multi-device (1, 8, 1, 1)
4h 47m 20s
amd64 / test-upstream-pax / single-process-multi-device (1, 1, 2, 4)
5h 18m 5s
amd64 / test-upstream-pax / pax-multi-node (1DP1FSDP1TP1PP, 1, 1, 1, 1, 4)
5h 14m 12s
amd64 / test-te / te-V100-unit-test
46m 58s
amd64 / test-upstream-pax / pax-multi-node (8DP1FSDP1TP1PP, 1, 8, 1, 1, 4)
5h 3m 14s
amd64 / test-te / te-A100-unit-test
36m 39s
amd64 / test-upstream-pax / pax-multi-node (1DP8FSDP1TP1PP, 1, 1, 8, 1, 4)
5h 6m 26s
amd64 / test-upstream-pax / pax-multi-node (2DP1FSDP1TP4PP, 4, 2, 1, 1, 4)
4h 57m 20s
amd64 / test-upstream-pax / pax-multi-node (4DP1FSDP2TP1PP, 1, 4, 1, 2, 4)
5h 33m 38s
amd64 / test-upstream-pax / pax-multi-node (16DP1FSDP1TP1PP, 1, 16, 1, 1, 4)
5h 33m 41s
amd64 / test-upstream-pax / pax-multi-node (2DP1FSDP2TP4PP, 4, 2, 1, 2, 4)
5h 42m 28s
amd64 / test-upstream-pax / pax-multi-node (LLaMA_eval, 1, 1, 8, 1, 4, true, --model-type LLaMA70BProxy --evaluate)
5h 11m 17s
amd64 / build-rosetta-pax / build-rosetta
6m 22s
amd64 / test-upstream-pax / single-process-evaluation (1, 8, 1, 1)
5h 36m 2s
amd64 / test-te / runner / launch-slurm-runner
5h 29m 23s
amd64 / test-te-multigpu / te-multi-gpu (4) / te-4GPU
5h 0m 28s
amd64 / test-te-multigpu / te-multi-gpu (8) / te-8GPU
4h 50m 46s
amd64 / test-te-multigpu / te-multi-gpu (2) / te-2GPU
5h 21m 31s
amd64 / test-triton / triton-V100-unit-test
3m 49s
amd64 / test-triton / triton-A100-unit-test
3m 58s
amd64 / test-triton / runner / launch-slurm-runner
5h 46m 35s
amd64 / collect-docker-tags
1s
amd64 / test-rosetta-pax / rosetta-pax-multi-node (1, 8, 1, 1)
5h 43m 17s
amd64 / test-rosetta-pax / rosetta-pax-multi-node (1, 4, 1, 2)
5h 54m 23s
amd64 / test-rosetta-pax / rosetta-pax-multi-node-te (1DP1FSDP1TP1PP_TE, 1, 1, 1, 1, 4)
5h 44m 7s
amd64 / test-rosetta-pax / rosetta-pax-multi-node (4, 2, 1, 1)
5h 40m 44s
amd64 / test-rosetta-pax / rosetta-pax-multi-node-te (8DP1FSDP1TP1PP_TE, 1, 8, 1, 1, 4)
6h 0m 16s
amd64 / test-rosetta-pax / rosetta-pax-multi-node (4, 2, 1, 2)
6h 0m 6s
amd64 / test-rosetta-pax / single-process-multi-device-te (1, 8, 1, 1)
6h 0m 15s
amd64 / test-rosetta-pax / rosetta-pax-multi-node-te (1DP8FSDP1TP1PP_TE, 1, 1, 8, 1, 4)
6h 0m 16s
amd64 / test-rosetta-pax / single-process-multi-device-te (1, 1, 2, 4)
5h 46m 43s
amd64 / test-rosetta-pax / rosetta-pax-multi-node-te (4DP1FSDP2TP1PP_TE, 1, 4, 1, 2, 4)
6h 0m 3s
amd64 / test-rosetta-pax / rosetta-pax-multi-node-te (16DP1FSDP1TP1PP_TE, 1, 16, 1, 1, 4)
6h 0m 12s
amd64 / test-rosetta-pax / rosetta-pax-multi-node-te (5B_fused_attn_1, 1, 1, 8, 1, 2, --model-type 5B --enable-fused-attn)
6h 0m 16s
amd64 / test-rosetta-pax / rosetta-pax-multi-node-te (5B_fused_attn_0, 1, 1, 8, 1, 2, --model-type 5B)
5h 50m 43s
amd64 / test-rosetta-pax / rosetta-pax-multi-node-te (LLaMA_eval_TE, 1, 1, 8, 1, 4, true, --model-type LLaMA70BProxy --evalu...
5h 51m 32s
amd64 / test-rosetta-pax / rosetta-pax-single-node-dropout-te (1, 8, 1, 1)
6h 0m 15s
amd64 / test-rosetta-pax / single-process-evaluation-te (1, 8, 1, 1)
6h 0m 14s
arm64 / test-upstream-pax
0s
arm64 / test-te
0s
arm64 / test-te-multigpu
0s
arm64 / build-rosetta-pax / build-rosetta
13m 39s
arm64 / collect-docker-tags
1s
arm64 / test-rosetta-pax
0s
amd64 / test-maxtext / summary
1s
amd64 / test-maxtext / metrics
17s
amd64 / test-maxtext / sitrep / sitrep
6s
amd64 / test-maxtext / outcome
1s
amd64 / test-te-multigpu / sitrep
8s
amd64 / test-upstream-pax / summary
1s
amd64 / test-upstream-pax / metrics
18s
amd64 / test-upstream-pax / sitrep / sitrep
15s
amd64 / test-upstream-pax / outcome
1s
amd64 / test-rosetta-pax / summary
1s
amd64 / test-rosetta-pax / metrics
0s
amd64 / test-rosetta-pax / sitrep / sitrep
8s
amd64 / test-rosetta-pax / outcome
1s
make-publish-configs
3s
merge-new-manifest
0s
publish-containers (equinox, mock-jax, 900, ghcr.io/nvidia/jax-toolbox-internal:8807083247-equino... / publish
13s
publish-containers (grok, mock-jax-mealkit, 500, ghcr.io/nvidia/jax-toolbox-internal:8807083247-g... / publish
13s
publish-containers (grok, mock-jax, 900, ghcr.io/nvidia/jax-toolbox-internal:8807083247-grok-amd6... / publish
12s
publish-containers (equinox, mock-jax-mealkit, 500, ghcr.io/nvidia/jax-toolbox-internal:880708324... / publish
25s
publish-containers (jax, mock-jax, 1000, ghcr.io/nvidia/jax-toolbox-internal:8807083247-jax-amd64... / publish
13s
publish-containers (base, mock-jax, 800, ghcr.io/nvidia/jax-toolbox-internal:8807083247-base-amd6... / publish
14s
publish-containers (jax, mock-jax-mealkit, 500, ghcr.io/nvidia/jax-toolbox-internal:8807083247-ja... / publish
12s
publish-containers (levanter, mock-jax, 900, ghcr.io/nvidia/jax-toolbox-internal:8807083247-levan... / publish
17s
publish-containers (maxtext, mock-jax, 900, ghcr.io/nvidia/jax-toolbox-internal:8807083247-maxtex... / publish
13s
publish-containers (maxtext, mock-jax-mealkit, 500, ghcr.io/nvidia/jax-toolbox-internal:880708324... / publish
21s
publish-containers (triton, mock-jax, 900, ghcr.io/nvidia/jax-toolbox-internal:8807083247-triton-... / publish
11s
publish-containers (levanter, mock-jax-mealkit, 500, ghcr.io/nvidia/jax-toolbox-internal:88070832... / publish
13s
publish-containers (pax, mock-jax-mealkit, 500, ghcr.io/nvidia/jax-toolbox-internal:8807083247-pa... / publish
16s
publish-containers (pax, mock-jax, 900, ghcr.io/nvidia/jax-toolbox-internal:8807083247-pax-amd64,... / publish
12s
publish-containers (triton, mock-jax-mealkit, 500, ghcr.io/nvidia/jax-toolbox-internal:8807083247... / publish
10s
publish-containers (upstream-pax, mock-jax, 900, ghcr.io/nvidia/jax-toolbox-internal:8807083247-u... / publish
26s
publish-containers (upstream-t5x, mock-jax-mealkit, 500, ghcr.io/nvidia/jax-toolbox-internal:8807... / publish
13s
publish-containers (upstream-pax, mock-jax-mealkit, 500, ghcr.io/nvidia/jax-toolbox-internal:8807... / publish
21s
finalize / workflow-badge
5s
finalize / report
17s
finalize / upload-badge
8s
finalize / publish-badge
2s
9d 12h 52m 27s