Skip to content
CI

Dockerfile.jax: install JAX for TE build #3219

Sign in to view logs

Dockerfile.jax: install JAX for TE build

Dockerfile.jax: install JAX for TE build #3219

Triggered via pull request August 14, 2024 16:10
@oluptonolupton
opened #997
olupton/fix-te
Status Cancelled
Total duration 2h 34m 28s
Artifacts 73

ci.yaml

on: pull_request
metadata
0s
metadata
bump-manifest
19s
bump-manifest
Matrix: amd64 / test-distribution
Matrix: arm64 / test-distribution
amd64  /  ...  /  build-base
4m 19s
amd64 / build-base / build-base
arm64  /  ...  /  build-base
8m 25s
arm64 / build-base / build-base
amd64  /  ...  /  build-jax
21m 15s
amd64 / build-jax / build-jax
arm64  /  ...  /  build-jax
30m 46s
arm64 / build-jax / build-jax
amd64  /  ...  /  build-upstream-pax
6m 15s
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
7m 5s
amd64 / build-upstream-t5x / build-upstream-t5x
amd64  /  ...  /  launch-slurm-runner
18m 28s
amd64 / test-jax / runner / launch-slurm-runner
amd64  /  ...  /  launch-slurm-runner
5m 53s
amd64 / test-pallas / runner / launch-slurm-runner
amd64  /  ...  /  build-gemma
5m 19s
amd64 / build-gemma / build-gemma
amd64  /  ...  /  build-levanter
4m 7s
amd64 / build-levanter / build-levanter
amd64  /  ...  /  build-maxtext
6m 7s
amd64 / build-maxtext / build-maxtext
amd64  /  ...  /  build-triton
1m 44s
amd64 / build-triton / build-triton
amd64  /  ...  /  build-equinox
3m 9s
amd64 / build-equinox / build-equinox
arm64  /  ...  /  build-upstream-pax
26m 22s
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
26m 38s
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
arm64 / build-gemma / build
arm64  /  ...  /  build-levanter
5m 45s
arm64 / build-levanter / build-levanter
arm64  /  ...  /  build-maxtext
17m 50s
arm64 / build-maxtext / build-maxtext
arm64  /  ...  /  build
arm64 / build-triton / build
arm64  /  ...  /  build-equinox
4m 10s
arm64 / build-equinox / build-equinox
amd64  /  ...  /  build-rosetta
6m 35s
amd64 / build-rosetta-pax / build-rosetta
Matrix: amd64 / test-te / run-unit-test
amd64  /  ...  /  launch-slurm-runner
1h 4m
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
8m 5s
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
17m 40s
amd64 / test-gemma / runner / launch-slurm-runner
Matrix: amd64 / test-levanter / run-unit-test
amd64  /  ...  /  launch-slurm-runner
18m 44s
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
Waiting for pending jobs
amd64  /  ...  /  launch-slurm-runner
amd64 / test-triton / runner / launch-slurm-runner
arm64  /  ...  /  build-rosetta
12m 1s
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
14m 28s
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: 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
4s
amd64 / test-upstream-pax / test-upstream-pax-summary
amd64  /  ...  /  test-upstream-pax-metrics
16s
amd64 / test-upstream-pax / test-upstream-pax-metrics
amd64  /  ...  /  sitrep
6s
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
15s
amd64 / test-upstream-t5x / test-upstream-t5x-metrics
amd64  /  ...  /  test-maxtext-summary
0s
amd64 / test-maxtext / test-maxtext-summary
amd64  /  ...  /  test-maxtext-metrics
15s
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
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
26s
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
15s
amd64 / test-upstream-t5x / test-upstream-t5x-sitrep / sitrep
amd64  /  ...  /  sitrep
16s
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
0s
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
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

36 errors
amd64 / build-triton / build-triton
buildx failed with: ERROR: failed to solve: process "/bin/sh -c <<\"EOF\" bash -ex\nmkdir /opt/llvm-build\npushd /opt/llvm-build\ncmake -G Ninja \\\n -DCMAKE_BUILD_TYPE=Release \\\n -DCMAKE_C_COMPILER=clang \\\n -DCMAKE_CXX_COMPILER=clang++ \\\n -DLLVM_ENABLE_ASSERTIONS=ON \\\n -DLLVM_ENABLE_PROJECTS=\"mlir;llvm\" \\\n -DLLVM_TARGETS_TO_BUILD=\"host;NVPTX\" \\\n /opt/checkout/external/llvm-raw/llvm\nninja\nEOF" did not complete successfully: exit code: 1
amd64 / test-te-multigpu / te-multi-gpu (2) / te-2GPU
Process completed with exit code 1.
amd64 / test-te-multigpu / te-multi-gpu (8) / te-8GPU
Process completed with exit code 1.
amd64 / test-te-multigpu / te-multi-gpu (4) / te-4GPU
Process completed with exit code 1.
amd64 / test-upstream-pax / test-upstream-pax-outcome
Process completed with exit code 1.
amd64 / test-upstream-t5x / test-upstream-t5x-outcome
Process completed with exit code 1.
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-olupton/fix-te' exists
amd64 / test-rosetta-t5x / multi-gpu-multi-node (2N8G-te-1, 8, 2, --gin.train/utils.DatasetConfig.pack=False --gin.train_eva...
The operation was canceled.
amd64 / test-rosetta-t5x / vit-multi-gpu-multi-node (1, 1)
Canceling since a higher priority waiting request for 'CI-olupton/fix-te' exists
amd64 / test-rosetta-t5x / vit-multi-gpu-multi-node (1, 1)
The operation was canceled.
amd64 / test-rosetta-t5x / vit-single-process-multi-device (8)
Canceling since a higher priority waiting request for 'CI-olupton/fix-te' exists
amd64 / test-rosetta-t5x / vit-single-process-multi-device (8)
The operation was canceled.
amd64 / test-rosetta-t5x / single-process-multi-device (1P1G_te-0, 1, --enable-te 0)
Canceling since a higher priority waiting request for 'CI-olupton/fix-te' exists
amd64 / test-rosetta-t5x / single-process-multi-device (1P1G_te-0, 1, --enable-te 0)
The operation was canceled.
amd64 / test-rosetta-pax / rosetta-pax-multi-node (4, 2, 1, 2)
Canceling since a higher priority waiting request for 'CI-olupton/fix-te' exists
amd64 / test-rosetta-pax / rosetta-pax-multi-node (4, 2, 1, 2)
The operation was canceled.
amd64 / test-rosetta-t5x / vit-multi-gpu-multi-node (1, 2)
Canceling since a higher priority waiting request for 'CI-olupton/fix-te' exists
amd64 / test-rosetta-t5x / vit-multi-gpu-multi-node (8, 2)
Canceling since a higher priority waiting request for 'CI-olupton/fix-te' exists
amd64 / test-rosetta-t5x / vit-multi-gpu-multi-node (1, 2)
The operation was canceled.
amd64 / test-rosetta-t5x / vit-multi-gpu-multi-node (8, 2)
The operation was canceled.
amd64 / test-rosetta-pax / single-process-multi-device-te (1, 8, 1, 1)
Canceling since a higher priority waiting request for 'CI-olupton/fix-te' exists
amd64 / test-rosetta-pax / single-process-multi-device-te (1, 8, 1, 1)
The operation was canceled.
amd64 / test-rosetta-t5x / multi-gpu-multi-node (2N2G_te-0, 2, 2, --enable-te 0)
Canceling since a higher priority waiting request for 'CI-olupton/fix-te' exists
amd64 / test-rosetta-t5x / multi-gpu-multi-node (2N2G_te-0, 2, 2, --enable-te 0)
The operation was canceled.
amd64 / test-rosetta-pax / single-process-evaluation-te (1, 8, 1, 1)
Canceling since a higher priority waiting request for 'CI-olupton/fix-te' exists
amd64 / test-rosetta-pax / single-process-evaluation-te (1, 8, 1, 1)
The operation was canceled.
amd64 / test-rosetta-t5x / single-process-multi-device (1P8G_te-1, 8, --gin.train/utils.DatasetConfig.pack=False --gin.train...
Canceling since a higher priority waiting request for 'CI-olupton/fix-te' exists
amd64 / test-rosetta-t5x / single-process-multi-device (1P8G_te-1, 8, --gin.train/utils.DatasetConfig.pack=False --gin.train...
The operation was canceled.
amd64 / test-rosetta-t5x / vit-multi-gpu-multi-node (8, 1)
Canceling since a higher priority waiting request for 'CI-olupton/fix-te' exists
amd64 / test-rosetta-t5x / vit-multi-gpu-multi-node (8, 1)
The operation was canceled.
amd64 / test-rosetta-pax / single-process-multi-device-te (1, 1, 2, 4)
Canceling since a higher priority waiting request for 'CI-olupton/fix-te' exists
amd64 / test-rosetta-pax / single-process-multi-device-te (1, 1, 2, 4)
The operation was canceled.
amd64 / test-rosetta-pax / rosetta-pax-multi-node-te (1DP1FSDP1TP1PP_TE, 1, 1, 1, 1, 4)
Canceling since a higher priority waiting request for 'CI-olupton/fix-te' exists
amd64 / test-rosetta-pax / rosetta-pax-multi-node-te (1DP1FSDP1TP1PP_TE, 1, 1, 1, 1, 4)
The operation was canceled.
amd64 / test-rosetta-pax / rosetta-pax-multi-node-te (5B_fused_attn_0, 1, 1, 8, 1, 2, --model-type 5B --disable-fused-attn)
Canceling since a higher priority waiting request for 'CI-olupton/fix-te' exists
amd64 / test-rosetta-pax / rosetta-pax-multi-node-te (5B_fused_attn_0, 1, 1, 8, 1, 2, --model-type 5B --disable-fused-attn)
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
570 Bytes
artifact-equinox-build-arm64
568 Bytes
artifact-gemma-build-amd64
558 Bytes
artifact-jax-build-amd64
553 Bytes
artifact-jax-build-arm64
553 Bytes
artifact-levanter-build-amd64
572 Bytes
artifact-levanter-build-arm64
572 Bytes
artifact-maxtext-build-amd64
567 Bytes
artifact-maxtext-build-arm64
568 Bytes
artifact-maxtext-test
2.71 KB
artifact-pax-build-amd64
567 Bytes
artifact-pax-build-arm64
568 Bytes
artifact-rosetta-build-pax-amd64
583 Bytes
artifact-rosetta-build-pax-arm64
584 Bytes
artifact-rosetta-build-t5x-amd64
585 Bytes
artifact-rosetta-build-t5x-arm64
583 Bytes
artifact-t5x-build-amd64
570 Bytes
artifact-t5x-build-arm64
566 Bytes
artifact-triton-build-amd64
469 Bytes
artifact-upstream-pax-mgmn-test
2.43 KB
artifact-upstream-t5x-mgmn-test
2.12 KB
bumped-manifest
76.5 KB
gemma-unit-test-A100
4.24 KB
gemma-unit-test-V100
2.89 KB
jax-unit-test-A100
36.7 KB
jax-unit-test-V100
17.5 KB
levanter-unit-test-A100
10.2 KB
levanter-unit-test-V100
9.57 KB
pallas-unit-test-A100
7.32 KB
pallas-unit-test-V100
7.14 KB
rosetta-pax-10391134763-16DP1FSDP1TP1PP_TE
604 KB
rosetta-pax-10391134763-1DP8FSDP1TP1PP_TE
323 KB
rosetta-pax-10391134763-2DP1FSDP1TP4PP
261 KB
rosetta-pax-10391134763-4DP1FSDP2TP1PP
360 KB
rosetta-pax-10391134763-4DP1FSDP2TP1PP_TE
328 KB
rosetta-pax-10391134763-5B_fused_attn_1
202 KB
rosetta-pax-10391134763-8DP1FSDP1TP1PP
356 KB
rosetta-pax-10391134763-8DP1FSDP1TP1PP_TE
330 KB
rosetta-pax-10391134763-8DP_TE_dropout
328 KB
rosetta-pax-10391134763-LLaMA_eval_TE
213 KB
te-2GPU
5.17 KB
te-4GPU
5.19 KB
te-8GPU
5.25 KB
te-multigpu-test
528 Bytes
te-unit-test-A100
378 KB
te-unit-test-V100
372 KB
upstream-maxtext-10391134763-1DP1FSDP1TP1PP
13.1 KB
upstream-maxtext-10391134763-1DP1FSDP8TP1PP
18.7 KB
upstream-maxtext-10391134763-1DP2FSDP4TP1PP_single_process
13.3 KB
upstream-maxtext-10391134763-1DP4FSDP2TP1PP
18.8 KB
upstream-maxtext-10391134763-1DP8FSDP1TP1PP
18.3 KB
upstream-maxtext-10391134763-2DP2FSDP2TP1PP
18.3 KB
upstream-maxtext-10391134763-4DP2FSDP2TP1PP
24 KB
upstream-maxtext-metrics-test-log
4.57 KB
upstream-pax-10391134763-16DP1FSDP1TP1PP
720 KB
upstream-pax-10391134763-1DP1FSDP1TP1PP
141 KB
upstream-pax-10391134763-1DP2FSDP4TP1PP_single_process
142 KB
upstream-pax-10391134763-1DP8FSDP1TP1PP
414 KB
upstream-pax-10391134763-2DP1FSDP1TP4PP
195 KB
upstream-pax-10391134763-2DP1FSDP2TP4PP
351 KB
upstream-pax-10391134763-4DP1FSDP2TP1PP
417 KB
upstream-pax-10391134763-8DP1FSDP1TP1PP
412 KB
upstream-pax-10391134763-8DP1FSDP1TP1PP_eval
107 KB
upstream-pax-10391134763-8DP1FSDP1TP1PP_single_process
141 KB
upstream-pax-10391134763-LLaMA_eval
237 KB
upstream-pax-metrics-test-log
6.74 KB
upstream-t5x-10391134763-1P2G_fmha
5.14 MB
upstream-t5x-10391134763-1P8G
5.23 MB
upstream-t5x-10391134763-8G2N
5.34 MB
upstream-t5x-10391134763-8G2N_fmha
5.25 MB
upstream-t5x-metrics-test-log
8.75 KB