Skip to content
Back to pull request #1211

Refactor build-jax.sh and pin orbax-checkpoint #3663

Sign in to view logs

Refactor build-jax.sh and pin orbax-checkpoint

Refactor build-jax.sh and pin orbax-checkpoint #3663

Triggered via pull request December 21, 2024 06:47
@DwarKapexDwarKapex
opened #1211
25.01-devel-orbax-build-jax
Status Failure
Total duration 7h 43m 6s
Artifacts 58

ci.yaml

on: pull_request
metadata
0s
metadata
bump-manifest
18s
bump-manifest
Matrix: amd64 / test-distribution
Matrix: arm64 / test-distribution
amd64  /  ...  /  build-base
4m 15s
amd64 / build-base / build-base
arm64  /  ...  /  build-base
4m 11s
arm64 / build-base / build-base
amd64  /  ...  /  build-jax
19m 31s
amd64 / build-jax / build-jax
arm64  /  ...  /  build-jax
23m 6s
arm64 / build-jax / build-jax
amd64  /  ...  /  build-upstream-pax
13m 33s
amd64 / build-upstream-pax / build-upstream-pax
Matrix: amd64 / test-jax / run-unit-test
amd64  /  ...  /  build-upstream-t5x
9m 6s
amd64 / build-upstream-t5x / build-upstream-t5x
amd64  /  ...  /  launch-slurm-runner
19m 37s
amd64 / test-jax / runner / launch-slurm-runner
amd64  /  test-nsys-jax-eks
5m 14s
amd64 / test-nsys-jax-eks
amd64  /  ...  /  build-gemma
11m 44s
amd64 / build-gemma / build-gemma
amd64  /  ...  /  build-levanter
4m 48s
amd64 / build-levanter / build-levanter
amd64  /  ...  /  build-maxtext
14m 33s
amd64 / build-maxtext / build-maxtext
amd64  /  ...  /  build-triton
11m 56s
amd64 / build-triton / build-triton
Matrix: amd64 / test-nsys-jax / run-unit-test
amd64  /  ...  /  build-equinox
3m 38s
amd64 / build-equinox / build-equinox
amd64  /  ...  /  launch-slurm-runner
28m 40s
amd64 / test-nsys-jax / runner / launch-slurm-runner
arm64  /  ...  /  build-upstream-pax
20m 56s
arm64 / build-upstream-pax / build-upstream-pax
Matrix: arm64 / test-jax / run-unit-test
Waiting for pending jobs
arm64  /  ...  /  build-upstream-t5x
14m 22s
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 44s
arm64 / build-levanter / build-levanter
arm64  /  ...  /  build-maxtext
15m 19s
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 49s
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
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-gemma / run-unit-test
amd64  /  ...  /  launch-slurm-runner
35m 30s
amd64 / test-gemma / runner / launch-slurm-runner
Matrix: amd64 / test-levanter / run-unit-test
amd64  /  ...  /  launch-slurm-runner
32m 46s
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
34m 25s
amd64 / test-triton / runner / launch-slurm-runner
Matrix: amd64 / test-nsys-jax-archive
arm64  /  ...  /  build-rosetta
4m 3s
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
11m 7s
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
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  /  ...  /  test-upstream-t5x-summary
amd64 / test-upstream-t5x / test-upstream-t5x-summary
amd64  /  ...  /  test-upstream-t5x-metrics
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
amd64 / test-rosetta-t5x / test-t5x-rosetta-summary
amd64  /  ...  /  test-t5x-rosetta-metrics
amd64 / test-rosetta-t5x / test-t5x-rosetta-metrics
amd64  /  ...  /  sitrep
amd64 / test-upstream-t5x / test-upstream-t5x-sitrep / sitrep
amd64  /  ...  /  sitrep
6s
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
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
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
3s
make-publish-configs
merge-new-manifest
0s
merge-new-manifest
Matrix: publish-containers
finalize  /  workflow-badge
3s
finalize / workflow-badge
finalize  /  report
8s
finalize / report
finalize  /  upload-badge
8s
finalize / upload-badge
finalize  /  publish-badge
4s
finalize / publish-badge
Fit to window
Zoom out
Zoom in

Annotations

13 errors
arm64 / build-rosetta-pax / build-rosetta
buildx failed with: ERROR: failed to solve: process "/bin/sh -c pip-finalize.sh" did not complete successfully: exit code: 1
amd64 / build-upstream-pax / build-upstream-pax
buildx failed with: ERROR: failed to solve: process "/bin/sh -c pip-finalize.sh" did not complete successfully: exit code: 1
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
amd64 / test-nsys-jax-archive (ubuntu-22.04)
Process completed with exit code 1.
amd64 / test-nsys-jax-archive (macOS-latest)
The job was canceled because "ubuntu-22_04" failed.
amd64 / test-nsys-jax-archive (macOS-latest)
The operation was canceled.
amd64 / test-nsys-jax-archive (ubuntu-24.04)
The job was canceled because "ubuntu-22_04" failed.
amd64 / test-nsys-jax-archive (ubuntu-24.04)
The operation was canceled.
amd64 / test-triton / triton-A100-unit-test
Process completed with exit code 1.
amd64 / test-triton / triton-V100-unit-test
Process completed with exit code 1.
amd64 / test-maxtext / maxtext-multinode (1, 4, 2, 2)
The job running on runner jumpbox-vc69x-vnblg has exceeded the maximum execution time of 360 minutes.
amd64 / test-maxtext / maxtext-multinode (1, 4, 2, 2)
The operation was canceled.
amd64 / test-maxtext / test-maxtext-outcome
Process completed with exit code 1.

Artifacts

Produced during runtime
Name Size
artifact-base-build-amd64
569 Bytes
artifact-base-build-arm64
569 Bytes
artifact-equinox-build-amd64
569 Bytes
artifact-equinox-build-arm64
569 Bytes
artifact-final-report
3.1 KB
artifact-gemma-build-amd64
559 Bytes
artifact-jax-build-amd64
554 Bytes
artifact-jax-build-arm64
553 Bytes
artifact-levanter-build-amd64
574 Bytes
artifact-levanter-build-arm64
572 Bytes
artifact-maxtext-build-amd64
567 Bytes
artifact-maxtext-build-arm64
567 Bytes
artifact-maxtext-test
654 Bytes
artifact-pax-build-amd64
515 Bytes
artifact-pax-build-arm64
567 Bytes
artifact-rosetta-build-pax-arm64
530 Bytes
artifact-rosetta-build-t5x-arm64
585 Bytes
artifact-t5x-build-amd64
515 Bytes
artifact-t5x-build-arm64
566 Bytes
artifact-triton-build-amd64
565 Bytes
artifact-workflow-metadata
276 Bytes
bumped-manifest
81.2 KB
final-base
254 Bytes
final-equinox
263 Bytes
final-gemma
257 Bytes
final-jax
251 Bytes
final-levanter
265 Bytes
final-maxtext
263 Bytes
final-t5x
251 Bytes
final-triton
259 Bytes
final-upstream-pax
276 Bytes
final-upstream-t5x
277 Bytes
gemma-unit-test-A100
5.13 KB
gemma-unit-test-V100
3.81 KB
jax-unit-test-A100
20.3 KB
jax-unit-test-V100
27.2 KB
levanter-unit-test-A100
7.74 KB
levanter-unit-test-V100
9.5 KB
mealkit-equinox
272 Bytes
mealkit-gemma
266 Bytes
mealkit-jax
261 Bytes
mealkit-levanter
274 Bytes
mealkit-maxtext
271 Bytes
mealkit-pax
260 Bytes
mealkit-t5x
262 Bytes
mealkit-triton
269 Bytes
mealkit-upstream-pax
286 Bytes
mealkit-upstream-t5x
286 Bytes
nsys-jax-unit-test-A100
29.3 MB
nsys-jax-unit-test-V100
16 MB
triton-unit-test-A100
3.06 KB
triton-unit-test-V100
3.09 KB
upstream-maxtext-12443066143-1DP1FSDP1TP1PP
15.3 KB
upstream-maxtext-12443066143-1DP1FSDP8TP1PP
21.7 KB
upstream-maxtext-12443066143-1DP2FSDP4TP1PP_single_process
15.7 KB
upstream-maxtext-12443066143-1DP4FSDP2TP1PP
21.7 KB
upstream-maxtext-12443066143-1DP8FSDP1TP1PP
22 KB
upstream-maxtext-12443066143-2DP2FSDP2TP1PP
21.9 KB