Skip to content

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

Enable TE fused attention by default in rosetta pax container

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

Triggered via pull request April 23, 2024 20:53
Status Failure
Total duration 6h 40m 44s
Artifacts 80

ci.yaml

on: pull_request
metadata
0s
metadata
bump-manifest
21s
bump-manifest
Matrix: amd64 / test-distribution
Matrix: arm64 / test-distribution
amd64  /  ...  /  build-base
6m 33s
amd64 / build-base / build-base
arm64  /  ...  /  build-base
6m 42s
arm64 / build-base / build-base
amd64  /  ...  /  build-jax
7m 53s
amd64 / build-jax / build-jax
arm64  /  ...  /  build-jax
14m 10s
arm64 / build-jax / build-jax
amd64  /  ...  /  build-upstream-pax
6m 6s
amd64 / build-upstream-pax / build-upstream-pax
Matrix: amd64 / test-jax / run-unit-test
Matrix: amd64 / test-pallas / run-unit-test
amd64  /  ...  /  build-upstream-t5x
2m 14s
amd64 / build-upstream-t5x / build-upstream-t5x
amd64  /  ...  /  launch-slurm-runner
4h 59m
amd64 / test-jax / runner / launch-slurm-runner
amd64  /  ...  /  launch-slurm-runner
4h 52m
amd64 / test-pallas / runner / launch-slurm-runner
amd64  /  ...  /  build-levanter
4m 14s
amd64 / build-levanter / build-levanter
amd64  /  ...  /  build-maxtext
5m 7s
amd64 / build-maxtext / build-maxtext
amd64  /  ...  /  build-triton
14m 30s
amd64 / build-triton / build-triton
amd64  /  ...  /  build-equinox
11m 17s
amd64 / build-equinox / build-equinox
amd64  /  ...  /  build-grok
3m 8s
amd64 / build-grok / build-grok
arm64  /  ...  /  build-upstream-pax
28m 2s
arm64 / build-upstream-pax / build-upstream-pax
Matrix: arm64 / test-jax / run-unit-test
Waiting for pending jobs
Matrix: arm64 / test-pallas / run-unit-test
Waiting for pending jobs
arm64  /  ...  /  build-upstream-t5x
2m 44s
arm64 / build-upstream-t5x / build-upstream-t5x
arm64  /  ...  /  launch-slurm-runner
arm64 / test-jax / runner / launch-slurm-runner
arm64  /  ...  /  launch-slurm-runner
arm64 / test-pallas / runner / launch-slurm-runner
arm64  /  ...  /  build-levanter
8m 24s
arm64 / build-levanter / build-levanter
arm64  /  ...  /  build
arm64 / build-maxtext / build
arm64  /  ...  /  build
arm64 / build-triton / build
arm64  /  ...  /  build-equinox
6m 8s
arm64 / build-equinox / build-equinox
arm64  /  ...  /  build-grok
6m 13s
arm64 / build-grok / build-grok
amd64  /  ...  /  build-rosetta
6m 22s
amd64 / build-rosetta-pax / build-rosetta
Matrix: amd64 / test-te / run-unit-test
amd64  /  ...  /  launch-slurm-runner
5h 29m
amd64 / test-te / runner / launch-slurm-runner
Matrix: amd64 / test-upstream-pax / pax-multi-node
Matrix: amd64 / test-upstream-pax / single-process-evaluation
Matrix: amd64 / test-upstream-pax / single-process-multi-device
Matrix: amd64 / test-te-multigpu / te-multi-gpu
amd64  /  ...  /  build-rosetta
amd64 / build-rosetta-t5x / build-rosetta
Matrix: amd64 / test-upstream-t5x / t5x-multi-gpu
Waiting for pending jobs
Matrix: amd64 / test-upstream-t5x / t5x-multi-node
Waiting for pending jobs
Matrix: amd64 / test-levanter / run-unit-test
amd64  /  ...  /  launch-slurm-runner
4h 58m
amd64 / test-levanter / runner / launch-slurm-runner
Matrix: amd64 / test-maxtext / maxtext-multinode
Matrix: amd64 / test-maxtext / single-process-multi-device
Matrix: amd64 / test-triton / run-unit-test
amd64  /  ...  /  launch-slurm-runner
5h 46m
amd64 / test-triton / runner / launch-slurm-runner
arm64  /  ...  /  build-rosetta
13m 39s
arm64 / build-rosetta-pax / build-rosetta
Matrix: arm64 / test-te / run-unit-test
Waiting for pending jobs
arm64  /  ...  /  launch-slurm-runner
arm64 / test-te / runner / launch-slurm-runner
Matrix: arm64 / test-upstream-pax / pax-multi-node
Waiting for pending jobs
Matrix: arm64 / test-upstream-pax / single-process-evaluation
Waiting for pending jobs
Matrix: arm64 / test-upstream-pax / single-process-multi-device
Waiting for pending jobs
Matrix: arm64 / test-te-multigpu / te-multi-gpu
Waiting for pending jobs
arm64  /  ...  /  build-rosetta
arm64 / build-rosetta-t5x / build-rosetta
Matrix: arm64 / test-upstream-t5x / t5x-multi-gpu
Waiting for pending jobs
Matrix: arm64 / test-upstream-t5x / t5x-multi-node
Waiting for pending jobs
Matrix: arm64 / test-levanter / run-unit-test
Waiting for pending jobs
arm64  /  ...  /  launch-slurm-runner
arm64 / test-levanter / runner / launch-slurm-runner
Matrix: arm64 / test-maxtext / maxtext-multinode
Waiting for pending jobs
Matrix: arm64 / test-maxtext / single-process-multi-device
Waiting for pending jobs
Matrix: arm64 / test-triton / run-unit-test
Waiting for pending jobs
arm64  /  ...  /  launch-slurm-runner
arm64 / test-triton / runner / launch-slurm-runner
Matrix: amd64 / test-rosetta-pax / rosetta-pax-multi-node-te
Matrix: amd64 / test-rosetta-pax / rosetta-pax-multi-node
Matrix: amd64 / test-rosetta-pax / rosetta-pax-single-node-dropout-te
Matrix: amd64 / test-rosetta-pax / single-process-evaluation-te
Matrix: amd64 / test-rosetta-pax / single-process-multi-device-te
amd64  /  ...  /  summary
0s
amd64 / test-upstream-pax / summary
amd64  /  ...  /  metrics
18s
amd64 / test-upstream-pax / metrics
amd64  /  ...  /  sitrep
8s
amd64 / test-te-multigpu / sitrep
amd64  /  collect-docker-tags
0s
amd64 / collect-docker-tags
Matrix: amd64 / test-rosetta-t5x / multi-gpu-multi-node
Waiting for pending jobs
Matrix: amd64 / test-rosetta-t5x / single-process-multi-device
Waiting for pending jobs
Matrix: amd64 / test-rosetta-t5x / vit-multi-gpu-multi-node
Waiting for pending jobs
Matrix: amd64 / test-rosetta-t5x / vit-single-process-multi-device
Waiting for pending jobs
amd64  /  ...  /  summary
amd64 / test-upstream-t5x / summary
amd64  /  ...  /  metrics
amd64 / test-upstream-t5x / metrics
amd64  /  ...  /  summary
0s
amd64 / test-maxtext / summary
amd64  /  ...  /  metrics
17s
amd64 / test-maxtext / metrics
Matrix: arm64 / test-rosetta-pax / rosetta-pax-multi-node-te
Waiting for pending jobs
Matrix: arm64 / test-rosetta-pax / rosetta-pax-multi-node
Waiting for pending jobs
Matrix: arm64 / test-rosetta-pax / rosetta-pax-single-node-dropout-te
Waiting for pending jobs
Matrix: arm64 / test-rosetta-pax / single-process-evaluation-te
Waiting for pending jobs
Matrix: arm64 / test-rosetta-pax / single-process-multi-device-te
Waiting for pending jobs
arm64  /  ...  /  summary
arm64 / test-upstream-pax / summary
arm64  /  ...  /  metrics
arm64 / test-upstream-pax / metrics
arm64  /  ...  /  sitrep
arm64 / test-te-multigpu / sitrep
arm64  /  collect-docker-tags
0s
arm64 / collect-docker-tags
Matrix: arm64 / test-rosetta-t5x / multi-gpu-multi-node
Waiting for pending jobs
Matrix: arm64 / test-rosetta-t5x / single-process-multi-device
Waiting for pending jobs
Matrix: arm64 / test-rosetta-t5x / vit-multi-gpu-multi-node
Waiting for pending jobs
Matrix: arm64 / test-rosetta-t5x / vit-single-process-multi-device
Waiting for pending jobs
arm64  /  ...  /  summary
arm64 / test-upstream-t5x / summary
arm64  /  ...  /  metrics
arm64 / test-upstream-t5x / metrics
arm64  /  ...  /  summary
arm64 / test-maxtext / summary
arm64  /  ...  /  metrics
arm64 / test-maxtext / metrics
amd64  /  ...  /  summary
0s
amd64 / test-rosetta-pax / summary
amd64  /  ...  /  metrics
0s
amd64 / test-rosetta-pax / metrics
amd64  /  ...  /  sitrep
15s
amd64 / test-upstream-pax / sitrep / sitrep
amd64  /  ...  /  summary
amd64 / test-rosetta-t5x / summary
amd64  /  ...  /  metrics
amd64 / test-rosetta-t5x / metrics
amd64  /  ...  /  sitrep
amd64 / test-upstream-t5x / sitrep / sitrep
amd64  /  ...  /  sitrep
6s
amd64 / test-maxtext / sitrep / sitrep
arm64  /  ...  /  summary
arm64 / test-rosetta-pax / summary
arm64  /  ...  /  metrics
arm64 / test-rosetta-pax / metrics
arm64  /  ...  /  sitrep
arm64 / test-upstream-pax / sitrep / sitrep
arm64  /  ...  /  summary
arm64 / test-rosetta-t5x / summary
arm64  /  ...  /  metrics
arm64 / test-rosetta-t5x / metrics
arm64  /  ...  /  sitrep
arm64 / test-upstream-t5x / sitrep / sitrep
arm64  /  ...  /  sitrep
arm64 / test-maxtext / sitrep / sitrep
amd64  /  ...  /  sitrep
8s
amd64 / test-rosetta-pax / sitrep / sitrep
amd64  /  ...  /  outcome
0s
amd64 / test-upstream-pax / outcome
amd64  /  ...  /  sitrep
amd64 / test-rosetta-t5x / sitrep / sitrep
amd64  /  ...  /  outcome
amd64 / test-upstream-t5x / outcome
amd64  /  ...  /  outcome
0s
amd64 / test-maxtext / outcome
arm64  /  ...  /  sitrep
arm64 / test-rosetta-pax / sitrep / sitrep
arm64  /  ...  /  outcome
arm64 / test-upstream-pax / outcome
arm64  /  ...  /  sitrep
arm64 / test-rosetta-t5x / sitrep / sitrep
arm64  /  ...  /  outcome
arm64 / test-upstream-t5x / outcome
arm64  /  ...  /  outcome
arm64 / test-maxtext / outcome
amd64  /  ...  /  outcome
0s
amd64 / test-rosetta-pax / outcome
amd64  /  ...  /  outcome
amd64 / test-rosetta-t5x / outcome
arm64  /  ...  /  outcome
arm64 / test-rosetta-pax / outcome
arm64  /  ...  /  outcome
arm64 / test-rosetta-t5x / outcome
make-publish-configs
3s
make-publish-configs
merge-new-manifest
0s
merge-new-manifest
Matrix: publish-containers
finalize  /  workflow-badge
5s
finalize / workflow-badge
finalize  /  report
17s
finalize / report
finalize  /  upload-badge
8s
finalize / upload-badge
finalize  /  publish-badge
2s
finalize / publish-badge
Fit to window
Zoom out
Zoom in

Annotations

25 errors
amd64 / build-upstream-t5x / build-upstream-t5x
buildx failed with: ERROR: failed to solve: process "/bin/sh -c pip-finalize.sh" did not complete successfully: exit code: 1
arm64 / build-upstream-t5x / build-upstream-t5x
buildx failed with: ERROR: failed to solve: process "/bin/sh -c <<\"EOT\" bash -exu\nset -o pipefail\n\ngit clone https://github.com/google/array_record.git /tmp/array_record\ncd /tmp/array_record\ngit checkout v0.5.0\n\nexport CROSSTOOL_TOP=\"@ml2014_aarch64_config_aarch64//crosstool:toolchain\"\nexport AUDITWHEEL_PLATFORM=\"manylinux2014_aarch64\"\n./oss/build_whl.sh\nEOT" did not complete successfully: exit code: 1
amd64 / test-maxtext / metrics
Process completed with exit code 1.
amd64 / test-maxtext / outcome
Process completed with exit code 1.
amd64 / test-upstream-pax / metrics
Process completed with exit code 1.
amd64 / test-upstream-pax / outcome
Process completed with exit code 1.
amd64 / test-rosetta-pax / rosetta-pax-multi-node-te (4DP1FSDP2TP1PP_TE, 1, 4, 1, 2, 4)
The job running on runner GitHub Actions 154 has exceeded the maximum execution time of 360 minutes.
amd64 / test-rosetta-pax / rosetta-pax-multi-node (4, 2, 1, 2)
The job running on runner GitHub Actions 493 has exceeded the maximum execution time of 360 minutes.
amd64 / test-rosetta-pax / rosetta-pax-multi-node (4, 2, 1, 2)
The operation was canceled.
amd64 / test-rosetta-pax / rosetta-pax-multi-node-te (16DP1FSDP1TP1PP_TE, 1, 16, 1, 1, 4)
The job running on runner GitHub Actions 453 has exceeded the maximum execution time of 360 minutes.
amd64 / test-rosetta-pax / rosetta-pax-single-node-dropout-te (1, 8, 1, 1)
The job running on runner GitHub Actions 188 has exceeded the maximum execution time of 360 minutes.
amd64 / test-rosetta-pax / single-process-evaluation-te (1, 8, 1, 1)
The job running on runner GitHub Actions 345 has exceeded the maximum execution time of 360 minutes.
amd64 / test-rosetta-pax / single-process-evaluation-te (1, 8, 1, 1)
The operation was canceled.
amd64 / test-rosetta-pax / single-process-multi-device-te (1, 8, 1, 1)
The job running on runner GitHub Actions 350 has exceeded the maximum execution time of 360 minutes.
amd64 / test-rosetta-pax / single-process-multi-device-te (1, 8, 1, 1)
The operation was canceled.
amd64 / test-rosetta-pax / rosetta-pax-multi-node-te (8DP1FSDP1TP1PP_TE, 1, 8, 1, 1, 4)
The job running on runner GitHub Actions 130 has exceeded the maximum execution time of 360 minutes.
amd64 / test-rosetta-pax / rosetta-pax-multi-node-te (5B_fused_attn_1, 1, 1, 8, 1, 2, --model-type 5B --enable-fused-attn)
The job running on runner GitHub Actions 184 has exceeded the maximum execution time of 360 minutes.
amd64 / test-rosetta-pax / rosetta-pax-multi-node-te (1DP8FSDP1TP1PP_TE, 1, 1, 8, 1, 4)
The job running on runner GitHub Actions 470 has exceeded the maximum execution time of 360 minutes.
amd64 / test-rosetta-pax / outcome
Process completed with exit code 1.

Artifacts

Produced during runtime
Name Size
artifact-base-build-amd64 Expired
570 Bytes
artifact-base-build-arm64 Expired
569 Bytes
artifact-equinox-build-amd64 Expired
572 Bytes
artifact-equinox-build-arm64 Expired
571 Bytes
artifact-final-report Expired
912 Bytes
artifact-grok-build-amd64 Expired
559 Bytes
artifact-grok-build-arm64 Expired
560 Bytes
artifact-jax-build-amd64 Expired
554 Bytes
artifact-jax-build-arm64 Expired
555 Bytes
artifact-levanter-build-amd64 Expired
574 Bytes
artifact-levanter-build-arm64 Expired
575 Bytes
artifact-maxtext-build-amd64 Expired
571 Bytes
artifact-maxtext-test Expired
669 Bytes
artifact-pax-build-amd64 Expired
570 Bytes
artifact-pax-build-arm64 Expired
571 Bytes
artifact-rosetta-build-pax-amd64 Expired
587 Bytes
artifact-rosetta-build-pax-arm64 Expired
586 Bytes
artifact-rosetta-pax-mgmn-test Expired
694 Bytes
artifact-t5x-build-amd64 Expired
516 Bytes
artifact-t5x-build-arm64 Expired
473 Bytes
artifact-triton-build-amd64 Expired
567 Bytes
artifact-upstream-pax-mgmn-test Expired
705 Bytes
artifact-workflow-metadata Expired
278 Bytes
final-base Expired
254 Bytes
final-equinox Expired
263 Bytes
final-grok Expired
254 Bytes
final-jax Expired
250 Bytes
final-levanter Expired
266 Bytes
final-maxtext Expired
262 Bytes
final-pax Expired
251 Bytes
final-triton Expired
259 Bytes
final-upstream-pax Expired
276 Bytes
jax-unit-test-A100 Expired
19.1 KB
jax-unit-test-V100 Expired
20.3 KB
levanter-unit-test-A100 Expired
12.6 KB
levanter-unit-test-V100 Expired
13.6 KB
mealkit-equinox Expired
272 Bytes
mealkit-grok Expired
263 Bytes
mealkit-jax Expired
261 Bytes
mealkit-levanter Expired
274 Bytes
mealkit-maxtext Expired
271 Bytes
mealkit-pax Expired
261 Bytes
mealkit-triton Expired
267 Bytes
mealkit-upstream-pax Expired
286 Bytes
mealkit-upstream-t5x Expired
285 Bytes
pallas-unit-test-A100 Expired
11.1 KB
pallas-unit-test-V100 Expired
11.1 KB
rosetta-pax-8807083247-1DP1FSDP1TP1PP_TE Expired
88.3 KB
rosetta-pax-8807083247-1DP2FSDP4TP1PP_single_process_TE Expired
106 KB
rosetta-pax-8807083247-2DP1FSDP1TP4PP Expired
288 KB
rosetta-pax-8807083247-4DP1FSDP2TP1PP Expired
373 KB
rosetta-pax-8807083247-5B_fused_attn_0 Expired
389 KB
rosetta-pax-8807083247-8DP1FSDP1TP1PP Expired
368 KB
rosetta-pax-8807083247-LLaMA_eval_TE Expired
222 KB
te-2GPU Expired
7.15 KB
te-4GPU Expired
7.17 KB
te-8GPU Expired
7.2 KB
te-multigpu-test Expired
529 Bytes
te-unit-test-A100 Expired
1.08 MB
te-unit-test-V100 Expired
1.05 MB
triton-unit-test-A100 Expired
4.45 KB
triton-unit-test-V100 Expired
4.5 KB
upstream-maxtext-8807083247-1DP1FSDP1TP1PP Expired
9.69 KB
upstream-maxtext-8807083247-1DP1FSDP8TP1PP Expired
12.4 KB
upstream-maxtext-8807083247-1DP2FSDP4TP1PP_single_process Expired
9.83 KB
upstream-maxtext-8807083247-1DP4FSDP2TP1PP Expired
12.5 KB
upstream-maxtext-8807083247-1DP8FSDP1TP1PP Expired
576 Bytes
upstream-maxtext-8807083247-2DP2FSDP2TP1PP Expired
12.4 KB
upstream-maxtext-8807083247-4DP2FSDP2TP1PP Expired
15 KB
upstream-pax-8807083247-16DP1FSDP1TP1PP Expired
746 KB
upstream-pax-8807083247-1DP1FSDP1TP1PP Expired
142 KB
upstream-pax-8807083247-1DP2FSDP4TP1PP_single_process Expired
143 KB
upstream-pax-8807083247-1DP8FSDP1TP1PP Expired
420 KB
upstream-pax-8807083247-2DP1FSDP1TP4PP Expired
307 KB
upstream-pax-8807083247-2DP1FSDP2TP4PP Expired
568 KB
upstream-pax-8807083247-4DP1FSDP2TP1PP Expired
974 Bytes
upstream-pax-8807083247-8DP1FSDP1TP1PP Expired
421 KB
upstream-pax-8807083247-8DP1FSDP1TP1PP_eval Expired
107 KB
upstream-pax-8807083247-8DP1FSDP1TP1PP_single_process Expired
142 KB
upstream-pax-8807083247-LLaMA_eval Expired
242 KB