Skip to content

Update the dockerfile base image to cuda-dl-base #3739

Update the dockerfile base image to cuda-dl-base

Update the dockerfile base image to cuda-dl-base #3739

Triggered via pull request January 15, 2025 11:14
Status Cancelled
Total duration 3h 49m 18s
Artifacts 34

ci.yaml

on: pull_request
metadata
0s
metadata
bump-manifest
17s
bump-manifest
Matrix: amd64 / test-distribution
Matrix: arm64 / test-distribution
amd64  /  ...  /  build-base
2m 58s
amd64 / build-base / build-base
arm64  /  ...  /  build-base
3m 16s
arm64 / build-base / build-base
amd64  /  ...  /  build-jax
17m 31s
amd64 / build-jax / build-jax
arm64  /  ...  /  build-jax
27m 59s
arm64 / build-jax / build-jax
amd64  /  ...  /  build-upstream-pax
3m 42s
amd64 / build-upstream-pax / build-upstream-pax
Matrix: amd64 / test-jax / run-unit-test
amd64  /  ...  /  build-upstream-t5x
7m 5s
amd64 / build-upstream-t5x / build-upstream-t5x
amd64  /  ...  /  launch-slurm-runner
58m 0s
amd64 / test-jax / runner / launch-slurm-runner
amd64  /  test-nsys-jax-eks
1m 33s
amd64 / test-nsys-jax-eks
amd64  /  ...  /  build-gemma
5m 44s
amd64 / build-gemma / build-gemma
amd64  /  ...  /  build-levanter
6m 30s
amd64 / build-levanter / build-levanter
amd64  /  ...  /  build-maxtext
9m 20s
amd64 / build-maxtext / build-maxtext
amd64  /  ...  /  build-triton
18m 23s
amd64 / build-triton / build-triton
Matrix: amd64 / test-nsys-jax / run-unit-test
amd64  /  ...  /  build-equinox
3m 44s
amd64 / build-equinox / build-equinox
amd64  /  ...  /  launch-slurm-runner
1h 4m
amd64 / test-nsys-jax / runner / launch-slurm-runner
arm64  /  ...  /  build-upstream-pax
4m 34s
arm64 / build-upstream-pax / build-upstream-pax
Matrix: arm64 / test-jax / run-unit-test
Waiting for pending jobs
arm64  /  ...  /  build-upstream-t5x
8m 38s
arm64 / build-upstream-t5x / build-upstream-t5x
arm64  /  test-nsys-jax-eks
0s
arm64 / test-nsys-jax-eks
arm64  /  ...  /  launch-slurm-runner
arm64 / test-jax / runner / launch-slurm-runner
arm64  /  ...  /  build
arm64 / build-gemma / build
arm64  /  ...  /  build-levanter
6m 27s
arm64 / build-levanter / build-levanter
arm64  /  ...  /  build-maxtext
9m 36s
arm64 / build-maxtext / build-maxtext
arm64  /  ...  /  build
arm64 / build-triton / build
Matrix: arm64 / test-nsys-jax / run-unit-test
Waiting for pending jobs
arm64  /  ...  /  build-equinox
4m 27s
arm64 / build-equinox / build-equinox
arm64  /  ...  /  launch-slurm-runner
arm64 / test-nsys-jax / runner / launch-slurm-runner
amd64  /  ...  /  build-rosetta
amd64 / build-rosetta-pax / build-rosetta
Matrix: amd64 / test-te / run-unit-test
Waiting for pending jobs
amd64  /  ...  /  launch-slurm-runner
amd64 / test-te / runner / launch-slurm-runner
Matrix: amd64 / test-upstream-pax / pax-multi-node
Waiting for pending jobs
Matrix: amd64 / test-upstream-pax / single-process-evaluation
Waiting for pending jobs
Matrix: amd64 / test-upstream-pax / single-process-multi-device
Waiting for pending jobs
Matrix: amd64 / test-te-multigpu / te-multi-gpu
Waiting for pending jobs
amd64  /  ...  /  build-rosetta
14m 29s
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-gemma / run-unit-test
amd64  /  ...  /  launch-slurm-runner
1h 18m
amd64 / test-gemma / runner / launch-slurm-runner
Matrix: amd64 / test-levanter / run-unit-test
amd64  /  ...  /  launch-slurm-runner
1h 31m
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
1h 54m
amd64 / test-triton / runner / launch-slurm-runner
Matrix: amd64 / test-nsys-jax-archive
arm64  /  ...  /  build-rosetta
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
15m 31s
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-gemma / run-unit-test
Waiting for pending jobs
arm64  /  ...  /  launch-slurm-runner
arm64 / test-gemma / runner / launch-slurm-runner
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: arm64 / test-nsys-jax-archive
Matrix: amd64 / test-rosetta-pax / rosetta-pax-multi-node-te
Waiting for pending jobs
Matrix: amd64 / test-rosetta-pax / rosetta-pax-multi-node
Waiting for pending jobs
Matrix: amd64 / test-rosetta-pax / rosetta-pax-single-node-dropout-te
Waiting for pending jobs
Matrix: amd64 / test-rosetta-pax / single-process-evaluation-te
Waiting for pending jobs
Matrix: amd64 / test-rosetta-pax / single-process-multi-device-te
Waiting for pending jobs
amd64  /  ...  /  test-upstream-pax-summary
amd64 / test-upstream-pax / test-upstream-pax-summary
amd64  /  ...  /  test-upstream-pax-metrics
amd64 / test-upstream-pax / test-upstream-pax-metrics
amd64  /  ...  /  sitrep
amd64 / test-te-multigpu / sitrep
amd64  /  collect-docker-tags
0s
amd64 / collect-docker-tags
Matrix: amd64 / test-rosetta-t5x / multi-gpu-multi-node
Matrix: amd64 / test-rosetta-t5x / single-process-multi-device
Matrix: amd64 / test-rosetta-t5x / vit-multi-gpu-multi-node
Matrix: amd64 / test-rosetta-t5x / vit-single-process-multi-device
amd64  /  ...  /  test-upstream-t5x-summary
0s
amd64 / test-upstream-t5x / test-upstream-t5x-summary
amd64  /  ...  /  test-upstream-t5x-metrics
0s
amd64 / test-upstream-t5x / test-upstream-t5x-metrics
amd64  /  ...  /  test-maxtext-summary
0s
amd64 / test-maxtext / test-maxtext-summary
amd64  /  ...  /  test-maxtext-metrics
0s
amd64 / test-maxtext / 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  /  ...  /  test-upstream-pax-summary
arm64 / test-upstream-pax / test-upstream-pax-summary
arm64  /  ...  /  test-upstream-pax-metrics
arm64 / test-upstream-pax / 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  /  ...  /  test-upstream-t5x-summary
arm64 / test-upstream-t5x / test-upstream-t5x-summary
arm64  /  ...  /  test-upstream-t5x-metrics
arm64 / test-upstream-t5x / test-upstream-t5x-metrics
arm64  /  ...  /  test-maxtext-summary
arm64 / test-maxtext / test-maxtext-summary
arm64  /  ...  /  test-maxtext-metrics
arm64 / test-maxtext / test-maxtext-metrics
amd64  /  ...  /  test-pax-rosetta-summary
amd64 / test-rosetta-pax / test-pax-rosetta-summary
amd64  /  ...  /  test-pax-rosetta-metrics
amd64 / test-rosetta-pax / test-pax-rosetta-metrics
amd64  /  ...  /  sitrep
amd64 / test-upstream-pax / test-upstream-pax-sitrep / sitrep
amd64  /  ...  /  test-t5x-rosetta-summary
0s
amd64 / test-rosetta-t5x / test-t5x-rosetta-summary
amd64  /  ...  /  test-t5x-rosetta-metrics
0s
amd64 / test-rosetta-t5x / test-t5x-rosetta-metrics
amd64  /  ...  /  sitrep
amd64 / test-upstream-t5x / test-upstream-t5x-sitrep / sitrep
amd64  /  ...  /  sitrep
amd64 / test-maxtext / test-maxtext-sitrep / sitrep
arm64  /  ...  /  test-pax-rosetta-summary
arm64 / test-rosetta-pax / test-pax-rosetta-summary
arm64  /  ...  /  test-pax-rosetta-metrics
arm64 / test-rosetta-pax / test-pax-rosetta-metrics
arm64  /  ...  /  sitrep
arm64 / test-upstream-pax / test-upstream-pax-sitrep / sitrep
arm64  /  ...  /  test-t5x-rosetta-summary
arm64 / test-rosetta-t5x / test-t5x-rosetta-summary
arm64  /  ...  /  test-t5x-rosetta-metrics
arm64 / test-rosetta-t5x / test-t5x-rosetta-metrics
arm64  /  ...  /  sitrep
arm64 / test-upstream-t5x / test-upstream-t5x-sitrep / sitrep
arm64  /  ...  /  sitrep
arm64 / test-maxtext / test-maxtext-sitrep / sitrep
amd64  /  ...  /  sitrep
amd64 / test-rosetta-pax / test-pax-rosetta-sitrep / sitrep
amd64  /  ...  /  test-upstream-pax-outcome
amd64 / test-upstream-pax / test-upstream-pax-outcome
amd64  /  ...  /  sitrep
amd64 / test-rosetta-t5x / test-t5x-rosetta-sitrep / sitrep
amd64  /  ...  /  test-upstream-t5x-outcome
0s
amd64 / test-upstream-t5x / test-upstream-t5x-outcome
amd64  /  ...  /  test-maxtext-outcome
0s
amd64 / test-maxtext / test-maxtext-outcome
arm64  /  ...  /  sitrep
arm64 / test-rosetta-pax / test-pax-rosetta-sitrep / sitrep
arm64  /  ...  /  test-upstream-pax-outcome
arm64 / test-upstream-pax / test-upstream-pax-outcome
arm64  /  ...  /  sitrep
arm64 / test-rosetta-t5x / test-t5x-rosetta-sitrep / sitrep
arm64  /  ...  /  test-upstream-t5x-outcome
arm64 / test-upstream-t5x / test-upstream-t5x-outcome
arm64  /  ...  /  test-maxtext-outcome
arm64 / test-maxtext / test-maxtext-outcome
amd64  /  ...  /  test-pax-rosetta-outcome
amd64 / test-rosetta-pax / test-pax-rosetta-outcome
amd64  /  ...  /  test-t5x-rosetta-outcome
0s
amd64 / test-rosetta-t5x / test-t5x-rosetta-outcome
arm64  /  ...  /  test-pax-rosetta-outcome
arm64 / test-rosetta-pax / test-pax-rosetta-outcome
arm64  /  ...  /  test-t5x-rosetta-outcome
arm64 / test-rosetta-t5x / test-t5x-rosetta-outcome
make-publish-configs
0s
make-publish-configs
merge-new-manifest
0s
merge-new-manifest
Matrix: publish-containers
Waiting for pending jobs
finalize  /  workflow-badge
finalize / workflow-badge
finalize  /  report
finalize / report
finalize  /  upload-badge
finalize / upload-badge
finalize  /  publish-badge
finalize / publish-badge
Fit to window
Zoom out
Zoom in

Annotations

12 errors
amd64 / build-upstream-pax / build-upstream-pax
buildx failed with: ERROR: failed to solve: process "/bin/sh -c <<\"EOF\" bash -exu -o pipefail\ngit-clone.sh ${URLREF_LINGVO} ${SRC_PATH_LINGVO}\npushd ${SRC_PATH_LINGVO}\n\nCPU_ARCH=\"$(dpkg --print-architecture)\"\nif [[ \"${CPU_ARCH}\" == \"arm64\" ]]; then\n\n# Use aarch distribution of protobufs\npatch -p1 <<\"EOFINNER\"\ndiff --git a/lingvo/repo.bzl b/lingvo/repo.bzl\nindex ce65822d2..d9c0277aa 100644\n--- a/lingvo/repo.bzl\n+++ b/lingvo/repo.bzl\n@@ -232,9 +232,9 @@ filegroup(\n )\n \"\"\",\n urls = [\n- \"https://github.com/protocolbuffers/protobuf/releases/download/v21.9/protoc-21.9-linux-x86_64.zip\",\n+ \"https://github.com/protocolbuffers/protobuf/releases/download/v21.9/protoc-21.9-linux-aarch_64.zip\",\n ],\n- sha256 = \"3cd951aff8ce713b94cde55e12378f505f2b89d47bf080508cf77e3934f680b6\",\n+ sha256 = \"a584286dfa8ebb17032ece206ed74d5e9931e2edb9016e427be2a0dab3b21071\",\n )\n\n def icu():\nEOFINNER\n\nfi\n\npip install tensorflow_datasets==4.9.2 auditwheel tensorflow==2.18.0\nfor pattern in \\\n \"s|tensorflow=|#tensorflow=|g\" \\\n \"s|dataclasses=|#dataclasses=|g\" \\\n \"s|==.*||g\" \\\n; do\n sed -i \"${pattern}\" ${SRC_PATH_LINGVO}/docker/dev.requirements.txt\ndone\n# Lingvo support only python < 3.12, so we hack it and update dependencies\n# to be able to build for py-3.12\nfor pattern in \\\n \"s|tensorflow-text~=2.13.0|tensorflow-text~=2.18.1|g\" \\\n \"s|tensorflow~=2.13.0|tensorflow~=2.18.0|g\" \\\n \"s|python_requires='>=3.8,<3.11'|python_requires='>=3.8,<3.13'|\" \\\n; do\n sed -i \"${pattern}\" ${SRC_PATH_LINGVO}/pip_package/setup.py;\ndone\npip install -r docker/dev.requirements.txt\n\n# Some tests are flaky right now, so we skip running the tests.\nBUILD_ARCH=\"x86_64\"\nif [[ \"$CPU_ARCH\" == \"arm64\" ]]; then\n BUILD_ARCH=\"aarch64\";\nfi\nsed -i 's/manylinux2014_x86_64/manylinux_2_38_'\"${BUILD_ARCH}\"'/' pip_package/build.sh\nSKIP_TESTS=1 PYTHON_MINOR_VERSION=$(python --version | cut -d ' ' -f 2 | cut -d '.' -f 2) pip_package/build.sh\nEOF" did not complete successfully: exit code: 1
arm64 / build-upstream-pax / build-upstream-pax
buildx failed with: ERROR: failed to solve: process "/bin/sh -c <<\"EOF\" bash -exu -o pipefail\ngit-clone.sh ${URLREF_LINGVO} ${SRC_PATH_LINGVO}\npushd ${SRC_PATH_LINGVO}\n\nCPU_ARCH=\"$(dpkg --print-architecture)\"\nif [[ \"${CPU_ARCH}\" == \"arm64\" ]]; then\n\n# Use aarch distribution of protobufs\npatch -p1 <<\"EOFINNER\"\ndiff --git a/lingvo/repo.bzl b/lingvo/repo.bzl\nindex ce65822d2..d9c0277aa 100644\n--- a/lingvo/repo.bzl\n+++ b/lingvo/repo.bzl\n@@ -232,9 +232,9 @@ filegroup(\n )\n \"\"\",\n urls = [\n- \"https://github.com/protocolbuffers/protobuf/releases/download/v21.9/protoc-21.9-linux-x86_64.zip\",\n+ \"https://github.com/protocolbuffers/protobuf/releases/download/v21.9/protoc-21.9-linux-aarch_64.zip\",\n ],\n- sha256 = \"3cd951aff8ce713b94cde55e12378f505f2b89d47bf080508cf77e3934f680b6\",\n+ sha256 = \"a584286dfa8ebb17032ece206ed74d5e9931e2edb9016e427be2a0dab3b21071\",\n )\n\n def icu():\nEOFINNER\n\nfi\n\npip install tensorflow_datasets==4.9.2 auditwheel tensorflow==2.18.0\nfor pattern in \\\n \"s|tensorflow=|#tensorflow=|g\" \\\n \"s|dataclasses=|#dataclasses=|g\" \\\n \"s|==.*||g\" \\\n; do\n sed -i \"${pattern}\" ${SRC_PATH_LINGVO}/docker/dev.requirements.txt\ndone\n# Lingvo support only python < 3.12, so we hack it and update dependencies\n# to be able to build for py-3.12\nfor pattern in \\\n \"s|tensorflow-text~=2.13.0|tensorflow-text~=2.18.1|g\" \\\n \"s|tensorflow~=2.13.0|tensorflow~=2.18.0|g\" \\\n \"s|python_requires='>=3.8,<3.11'|python_requires='>=3.8,<3.13'|\" \\\n; do\n sed -i \"${pattern}\" ${SRC_PATH_LINGVO}/pip_package/setup.py;\ndone\npip install -r docker/dev.requirements.txt\n\n# Some tests are flaky right now, so we skip running the tests.\nBUILD_ARCH=\"x86_64\"\nif [[ \"$CPU_ARCH\" == \"arm64\" ]]; then\n BUILD_ARCH=\"aarch64\";\nfi\nsed -i 's/manylinux2014_x86_64/manylinux_2_38_'\"${BUILD_ARCH}\"'/' pip_package/build.sh\nSKIP_TESTS=1 PYTHON_MINOR_VERSION=$(python --version | cut -d ' ' -f 2 | cut -d '.' -f 2) pip_package/build.sh\nEOF" did not complete successfully: exit code: 1
amd64 / test-nsys-jax / nsys-jax-A100-unit-test
Process completed with exit code 8.
amd64 / test-triton / triton-A100-unit-test
Process completed with exit code 1.
amd64 / test-maxtext / maxtext-multinode (1, 1, 4, 2)
Canceling since a higher priority waiting request for 'CI-sbosisio/cuda-dl-base' exists
amd64 / test-maxtext / maxtext-multinode (1, 1, 4, 2)
The operation was canceled.
amd64 / test-rosetta-t5x / multi-gpu-multi-node (2N8G-te-1, 8, 2, --gin.train/utils.DatasetConfig.pack=False --gin.train_eva...
Canceling since a higher priority waiting request for 'CI-sbosisio/cuda-dl-base' exists
amd64 / test-rosetta-t5x / vit-multi-gpu-multi-node (1, 2)
Canceling since a higher priority waiting request for 'CI-sbosisio/cuda-dl-base' exists
amd64 / test-rosetta-t5x / vit-multi-gpu-multi-node (1, 2)
The operation was canceled.
amd64 / test-upstream-t5x / t5x-multi-node (8G2N, 8, 2)
Canceling since a higher priority waiting request for 'CI-sbosisio/cuda-dl-base' exists
amd64 / test-upstream-t5x / t5x-multi-node (8G2N, 8, 2)
The operation was canceled.

Artifacts

Produced during runtime
Name Size
artifact-base-build-amd64
567 Bytes
artifact-base-build-arm64
566 Bytes
artifact-equinox-build-amd64
569 Bytes
artifact-equinox-build-arm64
567 Bytes
artifact-gemma-build-amd64
559 Bytes
artifact-jax-build-amd64
554 Bytes
artifact-jax-build-arm64
553 Bytes
artifact-levanter-build-amd64
572 Bytes
artifact-levanter-build-arm64
571 Bytes
artifact-maxtext-build-amd64
569 Bytes
artifact-maxtext-build-arm64
568 Bytes
artifact-pax-build-amd64
472 Bytes
artifact-pax-build-arm64
471 Bytes
artifact-rosetta-build-t5x-amd64
584 Bytes
artifact-rosetta-build-t5x-arm64
584 Bytes
artifact-t5x-build-amd64
568 Bytes
artifact-t5x-build-arm64
567 Bytes
artifact-triton-build-amd64
566 Bytes
bumped-manifest
82.1 KB
gemma-unit-test-A100
6.54 KB
jax-unit-test-A100
23.8 KB
levanter-unit-test-A100
7.35 KB
nsys-jax-unit-test-A100
14.5 KB
rosetta-t5x-12787137104-1P1G_te-0
5.67 MB
rosetta-t5x-12787137104-1P8G_te-1
5.29 MB
rosetta-t5x-vit-12787137104-VIT1G1N
21.5 KB
rosetta-t5x-vit-12787137104-VIT1P8G
22 KB
triton-unit-test-A100
3.03 KB
upstream-maxtext-12787137104-1DP1FSDP1TP1PP
15.9 KB
upstream-maxtext-12787137104-1DP1FSDP8TP1PP
22.4 KB
upstream-maxtext-12787137104-1DP2FSDP4TP1PP_single_process
16.2 KB
upstream-maxtext-12787137104-1DP8FSDP1TP1PP
22.1 KB
upstream-t5x-12787137104-1P2G_fmha
6.19 MB
upstream-t5x-12787137104-1P8G
6.46 MB