Skip to content
CI

Add JetStream to MaxText container #3701

Sign in to view logs

Add JetStream to MaxText container

Add JetStream to MaxText container #3701

Triggered via pull request January 9, 2025 19:27
@DwarKapexDwarKapex
synchronize #1058
vkozlov/jetstream-4-maxtext
Status Cancelled
Total duration 2h 13m 55s
Artifacts 21

ci.yaml

on: pull_request
metadata
0s
metadata
bump-manifest
22s
bump-manifest
Matrix: amd64 / test-distribution
Matrix: arm64 / test-distribution
amd64  /  ...  /  build-base
4m 22s
amd64 / build-base / build-base
arm64  /  ...  /  build-base
4m 50s
arm64 / build-base / build-base
amd64  /  ...  /  build-jax
13m 56s
amd64 / build-jax / build-jax
arm64  /  ...  /  build-jax
20m 22s
arm64 / build-jax / build-jax
amd64  /  ...  /  build-upstream-pax
18m 21s
amd64 / build-upstream-pax / build-upstream-pax
amd64  /  ...  /  build-maxtext
9m 15s
amd64 / build-maxtext / build-maxtext
Matrix: amd64 / test-jax / run-unit-test
amd64  /  ...  /  build-upstream-t5x
14m 0s
amd64 / build-upstream-t5x / build-upstream-t5x
amd64  /  ...  /  launch-slurm-runner
0s
amd64 / test-jax / runner / launch-slurm-runner
amd64  /  test-nsys-jax-eks
5m 14s
amd64 / test-nsys-jax-eks
amd64  /  ...  /  build-gemma
20m 15s
amd64 / build-gemma / build-gemma
amd64  /  ...  /  build-levanter
6m 47s
amd64 / build-levanter / build-levanter
amd64  /  ...  /  build-triton
12m 32s
amd64 / build-triton / build-triton
Matrix: amd64 / test-nsys-jax / run-unit-test
amd64  /  ...  /  build-equinox
3m 55s
amd64 / build-equinox / build-equinox
amd64  /  ...  /  launch-slurm-runner
0s
amd64 / test-nsys-jax / runner / launch-slurm-runner
arm64  /  ...  /  build-upstream-pax
20m 58s
arm64 / build-upstream-pax / build-upstream-pax
arm64  /  ...  /  build-maxtext
8m 31s
arm64 / build-maxtext / build-maxtext
Matrix: arm64 / test-jax / run-unit-test
Waiting for pending jobs
arm64  /  ...  /  build-upstream-t5x
14m 50s
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 45s
arm64 / build-levanter / build-levanter
arm64  /  ...  /  build
arm64 / build-triton / build
Matrix: arm64 / test-nsys-jax / run-unit-test
Waiting for pending jobs
arm64  /  ...  /  build-equinox
4m 44s
arm64 / build-equinox / build-equinox
arm64  /  ...  /  launch-slurm-runner
arm64 / test-nsys-jax / runner / launch-slurm-runner
amd64  /  ...  /  build-rosetta
8m 23s
amd64 / build-rosetta-pax / build-rosetta
Matrix: amd64 / test-te / run-unit-test
amd64  /  ...  /  launch-slurm-runner
0s
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
Matrix: amd64 / test-jetstream / run-unit-test
Waiting for pending jobs
amd64  /  ...  /  launch-slurm-runner
amd64 / test-jetstream / runner / launch-slurm-runner
Matrix: amd64 / test-maxtext / maxtext-multinode
Waiting for pending jobs
Matrix: amd64 / test-maxtext / single-process-multi-device
Waiting for pending jobs
amd64  /  ...  /  build-rosetta
14m 7s
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
0s
amd64 / test-gemma / runner / launch-slurm-runner
Matrix: amd64 / test-levanter / run-unit-test
amd64  /  ...  /  launch-slurm-runner
0s
amd64 / test-levanter / runner / launch-slurm-runner
Matrix: amd64 / test-triton / run-unit-test
amd64  /  ...  /  launch-slurm-runner
0s
amd64 / test-triton / runner / launch-slurm-runner
Matrix: amd64 / test-nsys-jax-archive
arm64  /  ...  /  build-rosetta
10m 5s
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
Matrix: arm64 / test-jetstream / run-unit-test
Waiting for pending jobs
arm64  /  ...  /  launch-slurm-runner
arm64 / test-jetstream / 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
arm64  /  ...  /  build-rosetta
17m 40s
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-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
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  /  ...  /  test-upstream-pax-summary
0s
amd64 / test-upstream-pax / test-upstream-pax-summary
amd64  /  ...  /  test-upstream-pax-metrics
0s
amd64 / test-upstream-pax / test-upstream-pax-metrics
amd64  /  ...  /  sitrep
0s
amd64 / test-te-multigpu / sitrep
amd64  /  ...  /  test-maxtext-summary
amd64 / test-maxtext / test-maxtext-summary
amd64  /  ...  /  test-maxtext-metrics
amd64 / test-maxtext / test-maxtext-metrics
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
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  /  ...  /  test-maxtext-summary
arm64 / test-maxtext / test-maxtext-summary
arm64  /  ...  /  test-maxtext-metrics
arm64 / test-maxtext / test-maxtext-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  /  ...  /  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
amd64  /  ...  /  test-pax-rosetta-summary
0s
amd64 / test-rosetta-pax / test-pax-rosetta-summary
amd64  /  ...  /  test-pax-rosetta-metrics
0s
amd64 / test-rosetta-pax / test-pax-rosetta-metrics
amd64  /  ...  /  sitrep
amd64 / test-upstream-pax / test-upstream-pax-sitrep / sitrep
amd64  /  ...  /  sitrep
amd64 / test-maxtext / test-maxtext-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
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  /  ...  /  sitrep
arm64 / test-maxtext / test-maxtext-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
amd64  /  ...  /  sitrep
amd64 / test-rosetta-pax / test-pax-rosetta-sitrep / sitrep
amd64  /  ...  /  test-upstream-pax-outcome
0s
amd64 / test-upstream-pax / test-upstream-pax-outcome
amd64  /  ...  /  test-maxtext-outcome
amd64 / test-maxtext / test-maxtext-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
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  /  ...  /  test-maxtext-outcome
arm64 / test-maxtext / test-maxtext-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
amd64  /  ...  /  test-pax-rosetta-outcome
0s
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

37 errors
amd64 / build-maxtext / build-maxtext
buildx failed with: ERROR: failed to solve: process "/bin/sh -c <<\"EOF\" bash -ex\ngit-clone.sh ${URLREF_JETSTREAM} ${SRC_PATH_JETSTREAM}\necho \"-r ${SRC_PATH_JETSTREAM}/requirements.txt\" >> /opt/pip-tools.d/requirements-jetstream.in\nEOF" did not complete successfully: exit code: 128
arm64 / build-maxtext / build-maxtext
buildx failed with: ERROR: failed to solve: process "/bin/sh -c <<\"EOF\" bash -ex\ngit-clone.sh ${URLREF_JETSTREAM} ${SRC_PATH_JETSTREAM}\necho \"-r ${SRC_PATH_JETSTREAM}/requirements.txt\" >> /opt/pip-tools.d/requirements-jetstream.in\nEOF" did not complete successfully: exit code: 128
amd64 / test-jax / jax-A100-unit-test
The run was canceled by @DwarKapex.
amd64 / test-jax / jax-V100-unit-test
The run was canceled by @DwarKapex.
amd64 / test-nsys-jax / nsys-jax-A100-unit-test
The run was canceled by @DwarKapex.
amd64 / test-nsys-jax / nsys-jax-V100-unit-test
The run was canceled by @DwarKapex.
amd64 / test-triton / triton-A100-unit-test
The run was canceled by @DwarKapex.
amd64 / test-levanter / levanter-A100-unit-test
The run was canceled by @DwarKapex.
amd64 / test-levanter / levanter-V100-unit-test
The run was canceled by @DwarKapex.
amd64 / test-gemma / gemma-A100-unit-test
The run was canceled by @DwarKapex.
amd64 / test-triton / triton-V100-unit-test
The run was canceled by @DwarKapex.
amd64 / test-gemma / gemma-V100-unit-test
The run was canceled by @DwarKapex.
amd64 / test-te / te-A100-unit-test
The run was canceled by @DwarKapex.
amd64 / test-upstream-pax / single-process-evaluation (1, 8, 1, 1)
The run was canceled by @DwarKapex.
amd64 / test-te / te-V100-unit-test
The run was canceled by @DwarKapex.
amd64 / test-upstream-pax / pax-multi-node (1DP1FSDP1TP1PP, 1, 1, 1, 1, 4)
The run was canceled by @DwarKapex.
amd64 / test-upstream-pax / single-process-multi-device (1, 8, 1, 1)
The run was canceled by @DwarKapex.
amd64 / test-upstream-t5x / t5x-multi-node (8G2N, 8, 2)
The run was canceled by @DwarKapex.
amd64 / test-rosetta-pax / rosetta-pax-multi-node-te (1DP1FSDP1TP1PP_TE, 1, 1, 1, 1, 4)
The run was canceled by @DwarKapex.
amd64 / test-upstream-t5x / t5x-multi-gpu (1P8G, 8)
The run was canceled by @DwarKapex.
amd64 / test-rosetta-pax / rosetta-pax-single-node-dropout-te (1, 8, 1, 1)
The run was canceled by @DwarKapex.
amd64 / test-rosetta-pax / rosetta-pax-multi-node (1, 8, 1, 1)
The run was canceled by @DwarKapex.
amd64 / test-rosetta-pax / single-process-evaluation-te (1, 8, 1, 1)
The run was canceled by @DwarKapex.
amd64 / test-jax / runner / launch-slurm-runner
The run was canceled by @DwarKapex.
amd64 / test-rosetta-pax / single-process-multi-device-te (1, 8, 1, 1)
The run was canceled by @DwarKapex.
amd64 / test-triton / runner / launch-slurm-runner
The run was canceled by @DwarKapex.
amd64 / test-nsys-jax / runner / launch-slurm-runner
The run was canceled by @DwarKapex.
amd64 / test-levanter / runner / launch-slurm-runner
The run was canceled by @DwarKapex.
amd64 / test-rosetta-t5x / vit-single-process-multi-device (8)
The run was canceled by @DwarKapex.
amd64 / test-rosetta-t5x / single-process-multi-device (1P1G_te-0, 1, --enable-te 0)
The run was canceled by @DwarKapex.
amd64 / test-gemma / runner / launch-slurm-runner
The run was canceled by @DwarKapex.
amd64 / test-rosetta-t5x / multi-gpu-multi-node (2N8G-te-1, 8, 2, --gin.train/utils.DatasetConfig.pack=False --gin.train_eva...
The run was canceled by @DwarKapex.
amd64 / test-rosetta-t5x / vit-multi-gpu-multi-node (1, 1)
The run was canceled by @DwarKapex.
amd64 / test-te / runner / launch-slurm-runner
The run was canceled by @DwarKapex.
amd64 / test-te-multigpu / te-multi-gpu (2) / te-2GPU
The run was canceled by @DwarKapex.
amd64 / test-te-multigpu / te-multi-gpu (4) / te-4GPU
The run was canceled by @DwarKapex.
amd64 / test-te-multigpu / te-multi-gpu (8) / te-8GPU
The run was canceled by @DwarKapex.

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
568 Bytes
artifact-gemma-build-amd64
561 Bytes
artifact-jax-build-amd64
553 Bytes
artifact-jax-build-arm64
553 Bytes
artifact-levanter-build-amd64
574 Bytes
artifact-levanter-build-arm64
572 Bytes
artifact-maxtext-build-amd64
473 Bytes
artifact-maxtext-build-arm64
473 Bytes
artifact-pax-build-amd64
567 Bytes
artifact-pax-build-arm64
568 Bytes
artifact-rosetta-build-pax-amd64
582 Bytes
artifact-rosetta-build-pax-arm64
583 Bytes
artifact-rosetta-build-t5x-amd64
583 Bytes
artifact-rosetta-build-t5x-arm64
585 Bytes
artifact-t5x-build-amd64
569 Bytes
artifact-t5x-build-arm64
567 Bytes
artifact-triton-build-amd64
565 Bytes
bumped-manifest
81.2 KB