Skip to content
CI

Collect recursively and filter GPU tests using jax_test_gpu tag #3404

Sign in to view logs

Collect recursively and filter GPU tests using jax_test_gpu tag

Collect recursively and filter GPU tests using jax_test_gpu tag #3404

Triggered via pull request October 10, 2024 06:15
@oluptonolupton
reopened #1091
aportnoy/use-jax_test_gpu
Status Failure
Total duration 3h 6m 7s
Artifacts 119

ci.yaml

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

Annotations

11 errors and 4 warnings
amd64 / test-levanter / levanter-A100-unit-test
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 (2) / te-2GPU
Process completed with exit code 1.
amd64 / test-jax / jax-A100-unit-test
Process completed with exit code 1.
amd64 / test-te-multigpu / te-multi-gpu (4) / te-4GPU
Process completed with exit code 1.
amd64 / test-jax / jax-V100-unit-test
Process completed with exit code 1.
amd64 / test-levanter / levanter-V100-unit-test
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 / test-t5x-rosetta-outcome
Process completed with exit code 1.
amd64 / test-rosetta-pax / test-pax-rosetta-outcome
Process completed with exit code 1.
amd64 / test-rosetta-pax / rosetta-pax-multi-node-te (5B_fused_attn_0, 1, 1, 8, 1, 2, --model-type 5B --disable-fused-attn)
Failed to download action 'https://api.github.com/repos/webfactory/ssh-agent/tarball/dc588b651fe13675774614f8e6a936a468676387'. Error: The request was canceled due to the configured HttpClient.Timeout of 100 seconds elapsing.
amd64 / test-rosetta-pax / rosetta-pax-multi-node-te (5B_fused_attn_0, 1, 1, 8, 1, 2, --model-type 5B --disable-fused-attn)
Back off 26.631 seconds before retry.
amd64 / test-rosetta-pax / rosetta-pax-multi-node-te (5B_fused_attn_0, 1, 1, 8, 1, 2, --model-type 5B --disable-fused-attn)
Failed to download action 'https://api.github.com/repos/webfactory/ssh-agent/tarball/dc588b651fe13675774614f8e6a936a468676387'. Error: The request was canceled due to the configured HttpClient.Timeout of 100 seconds elapsing.
amd64 / test-rosetta-pax / rosetta-pax-multi-node-te (5B_fused_attn_0, 1, 1, 8, 1, 2, --model-type 5B --disable-fused-attn)
Back off 12.597 seconds before retry.

Artifacts

Produced during runtime
Name Size
artifact-base-build-amd64
567 Bytes
artifact-base-build-arm64
567 Bytes
artifact-equinox-build-amd64
570 Bytes
artifact-equinox-build-arm64
570 Bytes
artifact-final-report
10.4 KB
artifact-gemma-build-amd64
560 Bytes
artifact-jax-build-amd64
551 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
569 Bytes
artifact-maxtext-test
670 Bytes
artifact-pax-build-amd64
566 Bytes
artifact-pax-build-arm64
567 Bytes
artifact-rosetta-build-pax-amd64
584 Bytes
artifact-rosetta-build-pax-arm64
583 Bytes
artifact-rosetta-build-t5x-amd64
584 Bytes
artifact-rosetta-build-t5x-arm64
584 Bytes
artifact-rosetta-pax-mgmn-test
1.97 KB
artifact-rosetta-t5x-mgmn-test
2.74 KB
artifact-t5x-build-amd64
568 Bytes
artifact-t5x-build-arm64
566 Bytes
artifact-triton-build-amd64
564 Bytes
artifact-upstream-pax-mgmn-test
2.79 KB
artifact-upstream-t5x-mgmn-test
2.11 KB
artifact-workflow-metadata
277 Bytes
bumped-manifest
71.3 KB
final-base
254 Bytes
final-equinox
262 Bytes
final-gemma
256 Bytes
final-jax
251 Bytes
final-levanter
266 Bytes
final-maxtext
262 Bytes
final-pax
251 Bytes
final-t5x
251 Bytes
final-triton
259 Bytes
final-upstream-pax
276 Bytes
final-upstream-t5x
276 Bytes
gemma-unit-test-A100
4.08 KB
gemma-unit-test-V100
2.74 KB
jax-unit-test-A100
3.98 MB
jax-unit-test-V100
3.92 MB
levanter-unit-test-A100
6.84 KB
levanter-unit-test-V100
6.82 KB
mealkit-equinox
271 Bytes
mealkit-gemma
265 Bytes
mealkit-jax
261 Bytes
mealkit-levanter
274 Bytes
mealkit-maxtext
271 Bytes
mealkit-pax
261 Bytes
mealkit-t5x
261 Bytes
mealkit-triton
268 Bytes
mealkit-upstream-pax
285 Bytes
mealkit-upstream-t5x
285 Bytes
nsys-jax-unit-test-A100
29.2 MB
nsys-jax-unit-test-V100
16 MB
pallas-unit-test-A100
7.52 KB
pallas-unit-test-V100
7.36 KB
rosetta-pax-11268489580-16DP1FSDP1TP1PP_TE
477 KB
rosetta-pax-11268489580-1DP1FSDP1TP1PP_TE
86.7 KB
rosetta-pax-11268489580-1DP2FSDP4TP1PP_single_process_TE
61.6 KB
rosetta-pax-11268489580-1DP8FSDP1TP1PP_TE
249 KB
rosetta-pax-11268489580-2DP1FSDP1TP4PP
268 KB
rosetta-pax-11268489580-2DP1FSDP2TP4PP
497 KB
rosetta-pax-11268489580-4DP1FSDP2TP1PP
364 KB
rosetta-pax-11268489580-4DP1FSDP2TP1PP_TE
252 KB
rosetta-pax-11268489580-5B_fused_attn_0
168 KB
rosetta-pax-11268489580-5B_fused_attn_1
168 KB
rosetta-pax-11268489580-8DP1FSDP1TP1PP
366 KB
rosetta-pax-11268489580-8DP1FSDP1TP1PP_TE
246 KB
rosetta-pax-11268489580-8DP1FSDP1TP1PP_eval_TE
74.9 KB
rosetta-pax-11268489580-8DP1FSDP1TP1PP_single_process_TE
61.7 KB
rosetta-pax-11268489580-8DP_TE_dropout
256 KB
rosetta-pax-11268489580-LLaMA_eval_TE
158 KB
rosetta-pax-metrics-test-log
7.63 KB
rosetta-t5x-11268489580-1P1G_te-0
4.45 MB
rosetta-t5x-11268489580-1P8G_te-1
4.02 MB
rosetta-t5x-11268489580-2N2G_te-0
4.58 MB
rosetta-t5x-11268489580-2N8G-te-1
4.05 MB
rosetta-t5x-metrics-test-log
7.9 KB
rosetta-t5x-vit-11268489580-VIT1G1N
21.7 KB
rosetta-t5x-vit-11268489580-VIT1G2N
23.5 KB
rosetta-t5x-vit-11268489580-VIT1P8G
22 KB
rosetta-t5x-vit-11268489580-VIT8G1N
33.4 KB
rosetta-t5x-vit-11268489580-VIT8G2N
45.8 KB
te-2GPU
8.4 KB
te-4GPU
8.44 KB
te-8GPU
8.49 KB
te-multigpu-test
527 Bytes
te-unit-test-A100
19.7 KB
te-unit-test-V100
240 KB
triton-unit-test-A100
4.89 KB
triton-unit-test-V100
4.9 KB
upstream-maxtext-11268489580-1DP1FSDP1TP1PP
13.8 KB
upstream-maxtext-11268489580-1DP1FSDP8TP1PP
19.3 KB
upstream-maxtext-11268489580-1DP2FSDP4TP1PP_single_process
14 KB
upstream-maxtext-11268489580-1DP4FSDP2TP1PP
19.2 KB
upstream-maxtext-11268489580-1DP8FSDP1TP1PP
19.3 KB
upstream-maxtext-11268489580-2DP2FSDP2TP1PP
19.4 KB
upstream-maxtext-11268489580-4DP2FSDP2TP1PP
25 KB
upstream-maxtext-metrics-test-log
1.87 KB
upstream-pax-11268489580-16DP1FSDP1TP1PP
765 KB
upstream-pax-11268489580-1DP1FSDP1TP1PP
141 KB
upstream-pax-11268489580-1DP2FSDP4TP1PP_single_process
142 KB
upstream-pax-11268489580-1DP8FSDP1TP1PP
417 KB
upstream-pax-11268489580-2DP1FSDP1TP4PP
291 KB
upstream-pax-11268489580-2DP1FSDP2TP4PP
540 KB
upstream-pax-11268489580-4DP1FSDP2TP1PP
428 KB
upstream-pax-11268489580-8DP1FSDP1TP1PP
419 KB
upstream-pax-11268489580-8DP1FSDP1TP1PP_eval
108 KB
upstream-pax-11268489580-8DP1FSDP1TP1PP_single_process
142 KB
upstream-pax-11268489580-LLaMA_eval
241 KB
upstream-pax-metrics-test-log
7.08 KB
upstream-t5x-11268489580-1P2G_fmha
5.05 MB
upstream-t5x-11268489580-1P8G
5.14 MB
upstream-t5x-11268489580-8G2N
5.25 MB
upstream-t5x-11268489580-8G2N_fmha
5.16 MB
upstream-t5x-metrics-test-log
8.68 KB