Skip to content

Re-enable environment variables to guard against hangs with Triton gemms #2196

Re-enable environment variables to guard against hangs with Triton gemms

Re-enable environment variables to guard against hangs with Triton gemms #2196

Triggered via pull request March 7, 2024 18:54
Status Failure
Total duration 6h 6m 56s
Artifacts 79

ci.yaml

on: pull_request
metadata
0s
metadata
bump-manifest
44s
bump-manifest
Matrix: amd64 / test-distribution
Matrix: arm64 / test-distribution
amd64  /  ...  /  build-base
5m 38s
amd64 / build-base / build-base
arm64  /  ...  /  build-base
5m 38s
arm64 / build-base / build-base
amd64  /  ...  /  build-jax
12m 32s
amd64 / build-jax / build-jax
arm64  /  ...  /  build-jax
18m 26s
arm64 / build-jax / build-jax
amd64  /  ...  /  build-upstream-pax
6m 36s
amd64 / build-upstream-pax / build-upstream-pax
amd64  /  ...  /  build-upstream-t5x
6m 53s
amd64 / build-upstream-t5x / build-upstream-t5x
Matrix: amd64 / test-jax / run-unit-test
Matrix: amd64 / test-pallas / run-unit-test
amd64  /  ...  /  build-levanter
2m 32s
amd64 / build-levanter / build-levanter
amd64  /  ...  /  build-maxtext
5m 11s
amd64 / build-maxtext / build-maxtext
amd64  /  ...  /  build-triton
22m 37s
amd64 / build-triton / build-triton
amd64  /  ...  /  launch-slurm-runner
56m 22s
amd64 / test-jax / runner / launch-slurm-runner
amd64  /  ...  /  launch-slurm-runner
1h 0m
amd64 / test-pallas / runner / launch-slurm-runner
arm64  /  ...  /  build-upstream-pax
34m 31s
arm64 / build-upstream-pax / build-upstream-pax
arm64  /  ...  /  build-upstream-t5x
34m 25s
arm64 / build-upstream-t5x / build-upstream-t5x
Matrix: arm64 / test-jax / run-unit-test
Waiting for pending jobs
Matrix: arm64 / test-pallas / run-unit-test
Waiting for pending jobs
arm64  /  ...  /  build-levanter
2m 14s
arm64 / build-levanter / build-levanter
arm64  /  ...  /  build
arm64 / build-maxtext / build
arm64  /  ...  /  build
arm64 / build-triton / build
arm64  /  ...  /  launch-slurm-runner
arm64 / test-jax / runner / launch-slurm-runner
arm64  /  ...  /  launch-slurm-runner
arm64 / test-pallas / runner / launch-slurm-runner
amd64  /  ...  /  build-rosetta
6m 31s
amd64 / build-rosetta-pax / build-rosetta
Matrix: amd64 / test-te / te-multi-gpu
amd64  /  ...  /  te-unit-tests
35m 51s
amd64 / test-te / te-unit-tests
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
amd64  /  ...  /  build-rosetta
1m 48s
amd64 / build-rosetta-t5x / build-rosetta
Matrix: amd64 / test-upstream-t5x / t5x-multi-gpu
Matrix: amd64 / test-upstream-t5x / t5x-multi-node
Matrix: amd64 / test-levanter / run-unit-test
Waiting for pending jobs
amd64  /  ...  /  launch-slurm-runner
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 15m
amd64 / test-triton / runner / launch-slurm-runner
arm64  /  ...  /  build-rosetta
17m 24s
arm64 / build-rosetta-pax / build-rosetta
Matrix: arm64 / test-te / te-multi-gpu
Waiting for pending jobs
arm64  /  ...  /  te-unit-tests
arm64 / test-te / te-unit-tests
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
arm64  /  ...  /  build-rosetta
1m 24s
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
26s
amd64 / test-upstream-pax / metrics
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
0s
amd64 / test-upstream-t5x / summary
amd64  /  ...  /  metrics
14s
amd64 / test-upstream-t5x / metrics
amd64  /  ...  /  summary
0s
amd64 / test-maxtext / summary
amd64  /  ...  /  metrics
16s
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  /  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
15s
amd64 / test-rosetta-pax / metrics
amd64  /  ...  /  sitrep
10s
amd64 / test-upstream-pax / sitrep / sitrep
amd64  /  ...  /  summary
amd64 / test-rosetta-t5x / summary
amd64  /  ...  /  metrics
amd64 / test-rosetta-t5x / metrics
amd64  /  ...  /  sitrep
10s
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
10s
amd64 / test-rosetta-pax / sitrep / sitrep
amd64  /  ...  /  outcome
2s
amd64 / test-upstream-pax / outcome
amd64  /  ...  /  sitrep
amd64 / test-rosetta-t5x / sitrep / sitrep
amd64  /  ...  /  outcome
0s
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
finalize  /  workflow-badge
3s
finalize / workflow-badge
make-publish-configs
4s
make-publish-configs
merge-new-manifest
0s
merge-new-manifest
finalize  /  report
10s
finalize / report
finalize  /  upload-badge
5s
finalize / upload-badge
Matrix: publish-containers
finalize  /  publish-badge
4s
finalize / publish-badge
Fit to window
Zoom out
Zoom in

Annotations

8 errors and 2 warnings
amd64 / build-levanter / build-levanter
buildx failed with: ERROR: failed to solve: process "/bin/sh -c pip-finalize.sh" did not complete successfully: exit code: 1
arm64 / build-levanter / build-levanter
buildx failed with: ERROR: failed to solve: process "/bin/sh -c pip-finalize.sh" did not complete successfully: exit code: 1
amd64 / build-rosetta-t5x / build-rosetta
buildx failed with: ERROR: failed to solve: process "/bin/sh -c <<\"EOF\" bash -exu\nMANIFEST_DIR=$(dirname ${MANIFEST_FILE})\nif [[ \"${UPDATE_PATCHES}\" != \"true\" && \"${UPDATE_PATCHES}\" != \"false\" ]]; then\n echo \"UPDATE_PATCHES can only be true or false\"\n exit 1\nfi\nif [[ \"${UPDATE_PATCHES}\" == \"true\" ]]; then\n cp -r /mnt/jax-toolbox/.github/container/patches ${MANIFEST_DIR}/\n cp /mnt/jax-toolbox/.github/container/manifest.yaml ${MANIFEST_DIR}/manifest.yaml\n cp /mnt/jax-toolbox/.github/container/create-distribution.sh ${MANIFEST_DIR}/create-distribution.sh\n # TODO: remove\n cp /mnt/jax-toolbox/.github/container/pip-finalize.sh /usr/local/bin/\nfi\ncp -r /mnt/jax-toolbox/rosetta /opt/rosetta\n\nif [[ -n \"${UPDATED_TE_REF}\" ]]; then\n TE_INSTALL_DIR=/opt/transformer-engine\n yq e \".transformer-engine.latest_verified_commit = \\\"${UPDATED_TE_REF}\\\"\" -i $MANIFEST_FILE\n # Install from source instead of pre-built wheel\n sed -i -E 's@( file:///opt/transformer-engine)/dist/[^ ]*@\\1@' /opt/pip-tools.d/requirements-te.in\n git -C $TE_INSTALL_DIR fetch -a\n if [[ \"${UPDATED_TE_REF}\" =~ ^pull/ ]]; then\n PR_ID=$(cut -d/ -f2 <<<\"${UPDATED_TE_REF}\")\n git -C $TE_INSTALL_DIR fetch origin ${UPDATED_TE_REF}:PR-${PR_ID}\n git -C $TE_INSTALL_DIR checkout PR-${PR_ID}\n else\n git -C $TE_INSTALL_DIR checkout ${UPDATED_TE_REF}\n fi\nfi\n\n# Setting the username/email is required to author commits from patches\ngit config --global user.email \"${GIT_USER_EMAIL}\"\ngit config --global user.name \"${GIT_USER_NAME}\"\n\nbash ${MANIFEST_DIR}/create-distribution.sh \\\n --manifest ${MANIFEST_FILE} \\\n --package t5x\nbash ${MANIFEST_DIR}/create-distribution.sh \\\n --manifest ${MANIFEST_FILE} \\\n --package flax\n# Remove .gitconfig to avoid end-user authoring commits as the \"build user\"\nrm -f ~/.gitconfig\n\necho \"--extra-index-url https://developer.download.nvidia.com/compute/redist\" >> /opt/pip-tools.d/requirements-t5x.in\necho \"-e file:///opt/rosetta\" >> /opt/pip-tools.d/requirements-t5x.in\nEOF" did not complete successfully: exit code: 1
arm64 / build-rosetta-t5x / build-rosetta
buildx failed with: ERROR: failed to solve: process "/bin/sh -c <<\"EOF\" bash -exu\nMANIFEST_DIR=$(dirname ${MANIFEST_FILE})\nif [[ \"${UPDATE_PATCHES}\" != \"true\" && \"${UPDATE_PATCHES}\" != \"false\" ]]; then\n echo \"UPDATE_PATCHES can only be true or false\"\n exit 1\nfi\nif [[ \"${UPDATE_PATCHES}\" == \"true\" ]]; then\n cp -r /mnt/jax-toolbox/.github/container/patches ${MANIFEST_DIR}/\n cp /mnt/jax-toolbox/.github/container/manifest.yaml ${MANIFEST_DIR}/manifest.yaml\n cp /mnt/jax-toolbox/.github/container/create-distribution.sh ${MANIFEST_DIR}/create-distribution.sh\n # TODO: remove\n cp /mnt/jax-toolbox/.github/container/pip-finalize.sh /usr/local/bin/\nfi\ncp -r /mnt/jax-toolbox/rosetta /opt/rosetta\n\nif [[ -n \"${UPDATED_TE_REF}\" ]]; then\n TE_INSTALL_DIR=/opt/transformer-engine\n yq e \".transformer-engine.latest_verified_commit = \\\"${UPDATED_TE_REF}\\\"\" -i $MANIFEST_FILE\n # Install from source instead of pre-built wheel\n sed -i -E 's@( file:///opt/transformer-engine)/dist/[^ ]*@\\1@' /opt/pip-tools.d/requirements-te.in\n git -C $TE_INSTALL_DIR fetch -a\n if [[ \"${UPDATED_TE_REF}\" =~ ^pull/ ]]; then\n PR_ID=$(cut -d/ -f2 <<<\"${UPDATED_TE_REF}\")\n git -C $TE_INSTALL_DIR fetch origin ${UPDATED_TE_REF}:PR-${PR_ID}\n git -C $TE_INSTALL_DIR checkout PR-${PR_ID}\n else\n git -C $TE_INSTALL_DIR checkout ${UPDATED_TE_REF}\n fi\nfi\n\n# Setting the username/email is required to author commits from patches\ngit config --global user.email \"${GIT_USER_EMAIL}\"\ngit config --global user.name \"${GIT_USER_NAME}\"\n\nbash ${MANIFEST_DIR}/create-distribution.sh \\\n --manifest ${MANIFEST_FILE} \\\n --package t5x\nbash ${MANIFEST_DIR}/create-distribution.sh \\\n --manifest ${MANIFEST_FILE} \\\n --package flax\n# Remove .gitconfig to avoid end-user authoring commits as the \"build user\"\nrm -f ~/.gitconfig\n\necho \"--extra-index-url https://developer.download.nvidia.com/compute/redist\" >> /opt/pip-tools.d/requirements-t5x.in\necho \"-e file:///opt/rosetta\" >> /opt/pip-tools.d/requirements-t5x.in\nEOF" did not complete successfully: exit code: 1
amd64 / test-upstream-t5x / outcome
Process completed with exit code 1.
amd64 / test-maxtext / outcome
Process completed with exit code 1.
amd64 / test-upstream-pax / outcome
Process completed with exit code 1.
amd64 / test-rosetta-pax / outcome
Process completed with exit code 1.
finalize / upload-badge
Node.js 16 actions are deprecated. Please update the following actions to use Node.js 20: actions/github-script@v6. For more information see: https://github.blog/changelog/2023-09-22-github-actions-transitioning-from-node-16-to-node-20/.
finalize / publish-badge
Node.js 16 actions are deprecated. Please update the following actions to use Node.js 20: actions/github-script@v6. For more information see: https://github.blog/changelog/2023-09-22-github-actions-transitioning-from-node-16-to-node-20/.

Artifacts

Produced during runtime
Name Size
MaxText-8193340831-1DP1FSDP1TP1PP Expired
9.46 KB
MaxText-8193340831-1DP1FSDP8TP1PP Expired
10.2 KB
MaxText-8193340831-1DP2FSDP4TP1PP_single_process Expired
9.6 KB
MaxText-8193340831-1DP4FSDP2TP1PP Expired
10.2 KB
MaxText-8193340831-1DP8FSDP1TP1PP Expired
10.2 KB
MaxText-8193340831-2DP2FSDP2TP1PP Expired
10.3 KB
MaxText-8193340831-4DP2FSDP2TP1PP Expired
12.7 KB
artifact-base-build-amd64 Expired
569 Bytes
artifact-base-build-arm64 Expired
570 Bytes
artifact-final-report Expired
976 Bytes
artifact-jax-build-amd64 Expired
563 Bytes
artifact-jax-build-arm64 Expired
562 Bytes
artifact-levanter-build-amd64 Expired
517 Bytes
artifact-levanter-build-arm64 Expired
516 Bytes
artifact-maxtext-build-amd64 Expired
571 Bytes
artifact-maxtext-test Expired
653 Bytes
artifact-pax-build-amd64 Expired
570 Bytes
artifact-pax-build-arm64 Expired
572 Bytes
artifact-rosetta-build-pax-amd64 Expired
587 Bytes
artifact-rosetta-build-pax-arm64 Expired
586 Bytes
artifact-rosetta-build-t5x-amd64 Expired
489 Bytes
artifact-rosetta-build-t5x-arm64 Expired
489 Bytes
artifact-rosetta-pax-mgmn-test Expired
1.24 KB
artifact-t5x-build-amd64 Expired
571 Bytes
artifact-t5x-build-arm64 Expired
570 Bytes
artifact-te-1g-test Expired
7.22 KB
artifact-te-2g-test Expired
7.21 KB
artifact-te-unit-test Expired
1.03 MB
artifact-triton-build-amd64 Expired
568 Bytes
artifact-upstream-pax-mgmn-test Expired
703 Bytes
artifact-upstream-t5x-mgmn-test Expired
678 Bytes
artifact-workflow-metadata Expired
278 Bytes
jax-unit-test-A100 Expired
12.2 KB
jax-unit-test-V100 Expired
13.2 KB
pallas-unit-test-A100 Expired
9.46 KB
pallas-unit-test-V100 Expired
9.46 KB
rosetta-pax-8193340831-16DP1FSDP1TP1PP_TE Expired
223 KB
rosetta-pax-8193340831-1DP1FSDP1TP1PP_TE Expired
88.8 KB
rosetta-pax-8193340831-1DP2FSDP4TP1PP_single_process_TE Expired
106 KB
rosetta-pax-8193340831-1DP8FSDP1TP1PP_TE Expired
122 KB
rosetta-pax-8193340831-2DP1FSDP1TP4PP Expired
119 KB
rosetta-pax-8193340831-2DP1FSDP2TP4PP Expired
234 KB
rosetta-pax-8193340831-4DP1FSDP2TP1PP Expired
121 KB
rosetta-pax-8193340831-4DP1FSDP2TP1PP_TE Expired
121 KB
rosetta-pax-8193340831-5B_fused_attn_0 Expired
124 KB
rosetta-pax-8193340831-5B_fused_attn_1 Expired
121 KB
rosetta-pax-8193340831-8DP1FSDP1TP1PP Expired
119 KB
rosetta-pax-8193340831-8DP1FSDP1TP1PP_TE Expired
116 KB
rosetta-pax-8193340831-8DP1FSDP1TP1PP_eval_TE Expired
76.8 KB
rosetta-pax-8193340831-8DP1FSDP1TP1PP_single_process_TE Expired
106 KB
rosetta-pax-8193340831-8DP_TE_dropout Expired
121 KB
rosetta-pax-metrics-test-log Expired
5.75 KB
triton-unit-test-A100 Expired
3.15 KB
triton-unit-test-V100 Expired
3.19 KB
upstream-MaxText-metrics-test-log Expired
4.29 KB
upstream-pax-8193340831-16DP1FSDP1TP1PP Expired
230 KB
upstream-pax-8193340831-1DP1FSDP1TP1PP Expired
142 KB
upstream-pax-8193340831-1DP2FSDP4TP1PP_single_process Expired
143 KB
upstream-pax-8193340831-1DP8FSDP1TP1PP Expired
120 KB
upstream-pax-8193340831-2DP1FSDP1TP4PP Expired
123 KB
upstream-pax-8193340831-2DP1FSDP2TP4PP Expired
228 KB
upstream-pax-8193340831-4DP1FSDP2TP1PP Expired
121 KB
upstream-pax-8193340831-8DP1FSDP1TP1PP Expired
118 KB
upstream-pax-8193340831-8DP1FSDP1TP1PP_eval Expired
106 KB
upstream-pax-8193340831-8DP1FSDP1TP1PP_single_process Expired
142 KB
upstream-pax-metrics-test-log Expired
4.23 KB
upstream-t5x-8193340831-1G1N Expired
36.6 KB
upstream-t5x-8193340831-1G2N Expired
7.95 KB
upstream-t5x-8193340831-1P1G Expired
36.3 KB
upstream-t5x-8193340831-1P2G Expired
36.5 KB
upstream-t5x-8193340831-1P4G Expired
36.4 KB
upstream-t5x-8193340831-1P8G Expired
36.3 KB
upstream-t5x-8193340831-2G1N Expired
7.88 KB
upstream-t5x-8193340831-2G2N Expired
10.8 KB
upstream-t5x-8193340831-4G1N Expired
10.7 KB
upstream-t5x-8193340831-4G2N Expired
16.9 KB
upstream-t5x-8193340831-8G1N Expired
15.8 KB
upstream-t5x-8193340831-8G2N Expired
26.9 KB
upstream-t5x-metrics-test-log Expired
5.44 KB