Skip to content
CI

nsys-jax: optimise data loading and .zip creation #3710

Sign in to view logs

nsys-jax: optimise data loading and .zip creation

nsys-jax: optimise data loading and .zip creation #3710

Run time

Learn about OS pricing on GitHub Actions
Job Run time
metadata
1s
bump-manifest
33s
arm64 / test-distribution (extra-only-distribution.sh)
8s
arm64 / test-distribution (mirror-only-distribution.sh)
6s
arm64 / test-distribution (upstream-only-distribution.sh)
5s
amd64 / test-distribution (extra-only-distribution.sh)
7s
amd64 / test-distribution (mirror-only-distribution.sh)
6s
arm64 / test-distribution (local-patch-distribution.sh)
9s
amd64 / test-distribution (upstream-only-distribution.sh)
7s
amd64 / test-distribution (local-patch-distribution.sh)
9s
amd64 / build-base / build-base
4m 7s
arm64 / build-base / build-base
4m 16s
arm64 / build-jax / build-jax
15m 19s
amd64 / build-jax / build-jax
9m 4s
amd64 / test-nsys-jax-eks
5m 8s
amd64 / build-equinox / build-equinox
3m 55s
amd64 / build-levanter / build-levanter
4m 42s
amd64 / build-triton / build-triton
20m 2s
amd64 / build-maxtext / build-maxtext
14m 13s
amd64 / build-upstream-pax / build-upstream-pax
9m 37s
amd64 / build-upstream-t5x / build-upstream-t5x
13m 28s
amd64 / build-gemma / build-gemma
12m 22s
amd64 / test-jax / jax-V100-unit-test
34m 44s
amd64 / test-nsys-jax / nsys-jax-V100-unit-test
6m 37s
amd64 / test-jax / jax-A100-unit-test
16m 12s
amd64 / test-nsys-jax / nsys-jax-A100-unit-test
7m 12s
amd64 / test-jax / runner / launch-slurm-runner
4h 37m 58s
amd64 / test-nsys-jax / runner / launch-slurm-runner
4h 30m 5s
arm64 / build-triton
0s
arm64 / build-gemma
0s
arm64 / test-nsys-jax
0s
arm64 / test-nsys-jax-eks
0s
arm64 / test-jax
0s
arm64 / build-levanter / build-levanter
6m 38s
arm64 / build-equinox / build-equinox
4m 30s
arm64 / build-maxtext / build-maxtext
15m 16s
arm64 / build-upstream-t5x / build-upstream-t5x
15m 5s
arm64 / build-upstream-pax / build-upstream-pax
11m 36s
arm64 / test-triton
0s
arm64 / test-gemma
0s
arm64 / test-nsys-jax-archive
0s
arm64 / test-levanter
0s
arm64 / test-te
0s
arm64 / build-rosetta-pax
0s
arm64 / test-te-multigpu
0s
arm64 / test-upstream-pax
0s
arm64 / test-rosetta-pax
0s
amd64 / test-triton / triton-V100-unit-test
3m 11s
amd64 / test-triton / triton-A100-unit-test
3m 54s
amd64 / test-triton / runner / launch-slurm-runner
3h 47m 33s
arm64 / test-upstream-t5x
0s
arm64 / build-rosetta-t5x / build-rosetta
15m 57s
arm64 / test-maxtext
0s
amd64 / test-levanter / levanter-V100-unit-test
12m 19s
amd64 / test-levanter / levanter-A100-unit-test
10m 39s
amd64 / test-levanter / runner / launch-slurm-runner
3h 22m 21s
arm64 / collect-docker-tags
1s
arm64 / test-rosetta-t5x
0s
amd64 / test-maxtext / maxtext-multinode (1, 1, 1, 1)
3h 31m 29s
amd64 / test-maxtext / maxtext-multinode (1, 1, 8, 1)
2h 7m 21s
amd64 / test-maxtext / maxtext-multinode (1, 1, 1, 8)
1h 49m 43s
amd64 / test-maxtext / maxtext-multinode (1, 1, 4, 2)
24m 34s
amd64 / test-maxtext / single-process-multi-device (1, 1, 2, 4)
3h 29m 26s
amd64 / test-maxtext / maxtext-multinode (1, 2, 2, 2)
10m 51s
amd64 / test-maxtext / maxtext-multinode (1, 4, 2, 2)
6h 0m 14s
amd64 / test-upstream-pax
0s
amd64 / test-te-multigpu
0s
amd64 / build-rosetta-pax
0s
amd64 / test-te
0s
amd64 / test-rosetta-pax
0s
amd64 / test-gemma / gemma-V100-unit-test
4m 30s
amd64 / test-gemma / gemma-A100-unit-test
4m 28s
amd64 / test-gemma / runner / launch-slurm-runner
3h 25m 7s
amd64 / test-upstream-t5x / t5x-multi-node (8G2N, 8, 2)
6h 0m 12s
amd64 / test-upstream-t5x / t5x-multi-gpu (1P8G, 8)
3h 21m 30s
amd64 / test-upstream-t5x / t5x-multi-node (8G2N_fmha, 8, 2, --enable-fmha 1)
6h 0m 14s
amd64 / build-rosetta-t5x / build-rosetta
12m 51s
amd64 / test-upstream-t5x / t5x-multi-gpu (1P2G_fmha, 2, --enable-fmha 1)
1h 38m 46s
amd64 / collect-docker-tags
1s
amd64 / test-rosetta-t5x / single-process-multi-device (1P1G_te-0, 1, --enable-te 0)
3h 23m 24s
amd64 / test-rosetta-t5x / multi-gpu-multi-node (2N8G-te-1, 8, 2, --gin.train/utils.DatasetConfig.pack=False --gin.train_eva...
6h 0m 12s
amd64 / test-rosetta-t5x / single-process-multi-device (1P8G_te-1, 8, --gin.train/utils.DatasetConfig.pack=False --gin.train...
1h 39m 35s
amd64 / test-rosetta-t5x / multi-gpu-multi-node (2N2G_te-0, 2, 2, --enable-te 0)
6h 0m 12s
amd64 / test-rosetta-t5x / vit-single-process-multi-device (8)
3h 29m 12s
amd64 / test-rosetta-t5x / vit-multi-gpu-multi-node (1, 1)
3h 32m 10s
amd64 / test-rosetta-t5x / vit-multi-gpu-multi-node (1, 2)
6h 0m 13s
amd64 / test-rosetta-t5x / vit-multi-gpu-multi-node (8, 1)
17m 42s
amd64 / test-rosetta-t5x / vit-multi-gpu-multi-node (8, 2)
6h 0m 12s
amd64 / test-nsys-jax-archive (ubuntu-22.04)
2m 15s
amd64 / test-nsys-jax-archive (ubuntu-24.04)
2m 37s
amd64 / test-nsys-jax-archive (macOS-latest)
2m 37s
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
6s
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
6s
amd64 / test-maxtext / test-maxtext-outcome
1s
make-publish-configs
3s
merge-new-manifest
0s
publish-containers (base, mock-jax, 800, ghcr.io/nvidia/jax-toolbox-internal:12708375325-base-amd... / publish
13s
publish-containers (equinox, mock-jax, 900, ghcr.io/nvidia/jax-toolbox-internal:12708375325-equin... / publish
17s
publish-containers (equinox, mock-jax-mealkit, 500, ghcr.io/nvidia/jax-toolbox-internal:127083753... / publish
16s
publish-containers (gemma, mock-jax, 900, ghcr.io/nvidia/jax-toolbox-internal:12708375325-gemma-a... / publish
13s
publish-containers (gemma, mock-jax-mealkit, 500, ghcr.io/nvidia/jax-toolbox-internal:12708375325... / publish
20s
publish-containers (jax, mock-jax, 1000, ghcr.io/nvidia/jax-toolbox-internal:12708375325-jax-amd6... / publish
13s
publish-containers (jax, mock-jax-mealkit, 500, ghcr.io/nvidia/jax-toolbox-internal:12708375325-j... / publish
12s
publish-containers (levanter, mock-jax, 900, ghcr.io/nvidia/jax-toolbox-internal:12708375325-leva... / publish
25s
publish-containers (maxtext, mock-jax-mealkit, 500, ghcr.io/nvidia/jax-toolbox-internal:127083753... / publish
25s
publish-containers (levanter, mock-jax-mealkit, 500, ghcr.io/nvidia/jax-toolbox-internal:12708375... / publish
13s
publish-containers (maxtext, mock-jax, 900, ghcr.io/nvidia/jax-toolbox-internal:12708375325-maxte... / publish
19s
publish-containers (t5x, mock-jax, 900, ghcr.io/nvidia/jax-toolbox-internal:12708375325-t5x-amd64... / publish
17s
publish-containers (t5x, mock-jax-mealkit, 500, ghcr.io/nvidia/jax-toolbox-internal:12708375325-t... / publish
24s
publish-containers (triton, mock-jax, 900, ghcr.io/nvidia/jax-toolbox-internal:12708375325-triton... / publish
14s
publish-containers (triton, mock-jax-mealkit, 500, ghcr.io/nvidia/jax-toolbox-internal:1270837532... / publish
20s
publish-containers (upstream-t5x, mock-jax, 900, ghcr.io/nvidia/jax-toolbox-internal:12708375325-... / publish
16s
publish-containers (upstream-t5x, mock-jax-mealkit, 500, ghcr.io/nvidia/jax-toolbox-internal:1270... / publish
17s
finalize / report
14s
finalize / workflow-badge
4s
finalize / upload-badge
5s
finalize / publish-badge
2s
3d 23h 57m 2s