Add FMHA PAXML test #3021

Run time

Learn about OS pricing on GitHub Actions

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

Provide feedback

Saved searches

Use saved searches to filter your results more quickly

Add FMHA PAXML test #3021

Usage

Add FMHA PAXML test #3021

Jobs

Run details

Run time