Skip to content

nsys-jax: roll back to 2024.4.1 and handle single-gpu profiles of single module executions better #3197

nsys-jax: roll back to 2024.4.1 and handle single-gpu profiles of single module executions better

nsys-jax: roll back to 2024.4.1 and handle single-gpu profiles of single module executions better #3197

Triggered via pull request August 9, 2024 10:31
Status Failure
Total duration 5h 35m 13s
Artifacts 121

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 22s
amd64 / build-base / build-base
arm64  /  ...  /  build-base
7m 8s
arm64 / build-base / build-base
amd64  /  ...  /  build-jax
12m 47s
amd64 / build-jax / build-jax
arm64  /  ...  /  build-jax
23m 37s
arm64 / build-jax / build-jax
amd64  /  ...  /  build-upstream-pax
6m 13s
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
6m 22s
amd64 / build-upstream-t5x / build-upstream-t5x
amd64  /  ...  /  launch-slurm-runner
2h 44m
amd64 / test-jax / runner / launch-slurm-runner
amd64  /  ...  /  launch-slurm-runner
2h 43m
amd64 / test-pallas / runner / launch-slurm-runner
amd64  /  ...  /  build-gemma
5m 13s
amd64 / build-gemma / build-gemma
amd64  /  ...  /  build-levanter
4m 30s
amd64 / build-levanter / build-levanter
amd64  /  ...  /  build-maxtext
5m 44s
amd64 / build-maxtext / build-maxtext
amd64  /  ...  /  build-triton
10m 52s
amd64 / build-triton / build-triton
amd64  /  ...  /  build-equinox
3m 10s
amd64 / build-equinox / build-equinox
amd64  /  ...  /  build-grok
3m 8s
amd64 / build-grok / build-grok
arm64  /  ...  /  build-upstream-pax
25m 28s
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
17m 43s
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 58s
arm64 / build-levanter / build-levanter
arm64  /  ...  /  build-maxtext
17m 3s
arm64 / build-maxtext / build-maxtext
arm64  /  ...  /  build
arm64 / build-triton / build
arm64  /  ...  /  build-equinox
3m 56s
arm64 / build-equinox / build-equinox
arm64  /  ...  /  build-grok
4m 2s
arm64 / build-grok / build-grok
amd64  /  ...  /  build-rosetta
6m 54s
amd64 / build-rosetta-pax / build-rosetta
Matrix: amd64 / test-te / run-unit-test
amd64  /  ...  /  launch-slurm-runner
3h 36m
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 38s
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
2h 43m
amd64 / test-gemma / runner / launch-slurm-runner
Matrix: amd64 / test-levanter / run-unit-test
amd64  /  ...  /  launch-slurm-runner
2h 54m
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
3h 52m
amd64 / test-triton / runner / launch-slurm-runner
arm64  /  ...  /  build-rosetta
11m 33s
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
13m 41s
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
20s
amd64 / test-upstream-pax / test-upstream-pax-metrics
amd64  /  ...  /  sitrep
5s
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
13s
amd64 / test-upstream-t5x / test-upstream-t5x-metrics
amd64  /  ...  /  test-maxtext-summary
0s
amd64 / test-maxtext / test-maxtext-summary
amd64  /  ...  /  test-maxtext-metrics
14s
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
12s
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
18s
amd64 / test-rosetta-t5x / test-t5x-rosetta-metrics
amd64  /  ...  /  sitrep
9s
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
9s
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
15s
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
14s
finalize / workflow-badge
finalize  /  report
18s
finalize / report
finalize  /  upload-badge
11s
finalize / upload-badge
finalize  /  publish-badge
2s
finalize / publish-badge
Fit to window
Zoom out
Zoom in

Annotations

5 errors
amd64 / test-jax / jax-A100-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-pax / test-pax-rosetta-outcome
Process completed with exit code 1.
amd64 / test-rosetta-t5x / test-t5x-rosetta-outcome
Process completed with exit code 1.

Artifacts

Produced during runtime
Name Size
artifact-base-build-amd64 Expired
566 Bytes
artifact-base-build-arm64 Expired
565 Bytes
artifact-equinox-build-amd64 Expired
569 Bytes
artifact-equinox-build-arm64 Expired
569 Bytes
artifact-final-report Expired
12.7 KB
artifact-gemma-build-amd64 Expired
560 Bytes
artifact-grok-build-amd64 Expired
558 Bytes
artifact-grok-build-arm64 Expired
559 Bytes
artifact-jax-build-amd64 Expired
554 Bytes
artifact-jax-build-arm64 Expired
554 Bytes
artifact-levanter-build-amd64 Expired
572 Bytes
artifact-levanter-build-arm64 Expired
574 Bytes
artifact-maxtext-build-amd64 Expired
568 Bytes
artifact-maxtext-build-arm64 Expired
568 Bytes
artifact-maxtext-test Expired
2.75 KB
artifact-pax-build-amd64 Expired
567 Bytes
artifact-pax-build-arm64 Expired
567 Bytes
artifact-rosetta-build-pax-amd64 Expired
583 Bytes
artifact-rosetta-build-pax-arm64 Expired
583 Bytes
artifact-rosetta-build-t5x-amd64 Expired
585 Bytes
artifact-rosetta-build-t5x-arm64 Expired
584 Bytes
artifact-rosetta-pax-mgmn-test Expired
2.83 KB
artifact-rosetta-t5x-mgmn-test Expired
3.25 KB
artifact-t5x-build-amd64 Expired
568 Bytes
artifact-t5x-build-arm64 Expired
568 Bytes
artifact-triton-build-amd64 Expired
564 Bytes
artifact-upstream-pax-mgmn-test Expired
2.44 KB
artifact-upstream-t5x-mgmn-test Expired
2.13 KB
artifact-workflow-metadata Expired
278 Bytes
bumped-manifest Expired
71.5 KB
final-base Expired
254 Bytes
final-equinox Expired
263 Bytes
final-gemma Expired
257 Bytes
final-grok Expired
254 Bytes
final-jax Expired
251 Bytes
final-levanter Expired
265 Bytes
final-maxtext Expired
262 Bytes
final-pax Expired
251 Bytes
final-t5x Expired
251 Bytes
final-triton Expired
259 Bytes
final-upstream-pax Expired
276 Bytes
final-upstream-t5x Expired
276 Bytes
gemma-unit-test-A100 Expired
4.24 KB
gemma-unit-test-V100 Expired
2.89 KB
jax-unit-test-A100 Expired
18.1 KB
jax-unit-test-V100 Expired
17.5 KB
levanter-unit-test-A100 Expired
12.8 KB
levanter-unit-test-V100 Expired
13.1 KB
mealkit-equinox Expired
271 Bytes
mealkit-gemma Expired
265 Bytes
mealkit-grok Expired
262 Bytes
mealkit-jax Expired
261 Bytes
mealkit-levanter Expired
274 Bytes
mealkit-maxtext Expired
271 Bytes
mealkit-pax Expired
261 Bytes
mealkit-t5x Expired
261 Bytes
mealkit-triton Expired
268 Bytes
mealkit-upstream-pax Expired
285 Bytes
mealkit-upstream-t5x Expired
285 Bytes
pallas-unit-test-A100 Expired
7.26 KB
pallas-unit-test-V100 Expired
7.14 KB
rosetta-pax-10317785303-16DP1FSDP1TP1PP_TE Expired
601 KB
rosetta-pax-10317785303-1DP1FSDP1TP1PP_TE Expired
86.2 KB
rosetta-pax-10317785303-1DP2FSDP4TP1PP_single_process_TE Expired
105 KB
rosetta-pax-10317785303-1DP8FSDP1TP1PP_TE Expired
331 KB
rosetta-pax-10317785303-2DP1FSDP1TP4PP Expired
197 KB
rosetta-pax-10317785303-2DP1FSDP2TP4PP Expired
351 KB
rosetta-pax-10317785303-4DP1FSDP2TP1PP Expired
358 KB
rosetta-pax-10317785303-4DP1FSDP2TP1PP_TE Expired
328 KB
rosetta-pax-10317785303-5B_fused_attn_0 Expired
201 KB
rosetta-pax-10317785303-5B_fused_attn_1 Expired
205 KB
rosetta-pax-10317785303-8DP1FSDP1TP1PP Expired
363 KB
rosetta-pax-10317785303-8DP1FSDP1TP1PP_TE Expired
331 KB
rosetta-pax-10317785303-8DP1FSDP1TP1PP_eval_TE Expired
76.5 KB
rosetta-pax-10317785303-8DP1FSDP1TP1PP_single_process_TE Expired
105 KB
rosetta-pax-10317785303-8DP_TE_dropout Expired
324 KB
rosetta-pax-10317785303-LLaMA_eval_TE Expired
224 KB
rosetta-pax-metrics-test-log Expired
8.59 KB
rosetta-t5x-10317785303-1P1G_te-0 Expired
4.58 MB
rosetta-t5x-10317785303-1P8G_te-1 Expired
5.09 MB
rosetta-t5x-10317785303-2N2G_te-0 Expired
4.74 MB
rosetta-t5x-10317785303-2N8G-te-1 Expired
5.14 MB
rosetta-t5x-metrics-test-log Expired
7.62 KB
rosetta-t5x-vit-10317785303-VIT1G1N Expired
21.7 KB
rosetta-t5x-vit-10317785303-VIT1G2N Expired
23.3 KB
rosetta-t5x-vit-10317785303-VIT1P8G Expired
22 KB
rosetta-t5x-vit-10317785303-VIT8G1N Expired
32.5 KB
rosetta-t5x-vit-10317785303-VIT8G2N Expired
43.9 KB
te-2GPU Expired
6 KB
te-4GPU Expired
6.03 KB
te-8GPU Expired
6.07 KB
te-multigpu-test Expired
528 Bytes
te-unit-test-A100 Expired
380 KB
te-unit-test-V100 Expired
376 KB
triton-unit-test-A100 Expired
4.76 KB
triton-unit-test-V100 Expired
4.75 KB
upstream-maxtext-10317785303-1DP1FSDP1TP1PP Expired
12.9 KB
upstream-maxtext-10317785303-1DP1FSDP8TP1PP Expired
18.2 KB
upstream-maxtext-10317785303-1DP2FSDP4TP1PP_single_process Expired
13 KB
upstream-maxtext-10317785303-1DP4FSDP2TP1PP Expired
18.2 KB
upstream-maxtext-10317785303-1DP8FSDP1TP1PP Expired
18 KB
upstream-maxtext-10317785303-2DP2FSDP2TP1PP Expired
17.9 KB
upstream-maxtext-10317785303-4DP2FSDP2TP1PP Expired
24.4 KB
upstream-maxtext-metrics-test-log Expired
4.54 KB
upstream-pax-10317785303-16DP1FSDP1TP1PP Expired
732 KB
upstream-pax-10317785303-1DP1FSDP1TP1PP Expired
141 KB
upstream-pax-10317785303-1DP2FSDP4TP1PP_single_process Expired
141 KB
upstream-pax-10317785303-1DP8FSDP1TP1PP Expired
414 KB
upstream-pax-10317785303-2DP1FSDP1TP4PP Expired
198 KB
upstream-pax-10317785303-2DP1FSDP2TP4PP Expired
346 KB
upstream-pax-10317785303-4DP1FSDP2TP1PP Expired
409 KB
upstream-pax-10317785303-8DP1FSDP1TP1PP Expired
417 KB
upstream-pax-10317785303-8DP1FSDP1TP1PP_eval Expired
107 KB
upstream-pax-10317785303-8DP1FSDP1TP1PP_single_process Expired
141 KB
upstream-pax-10317785303-LLaMA_eval Expired
229 KB
upstream-pax-metrics-test-log Expired
6.77 KB
upstream-t5x-10317785303-1P2G_fmha Expired
5.17 MB
upstream-t5x-10317785303-1P8G Expired
5.27 MB
upstream-t5x-10317785303-8G2N Expired
5.38 MB
upstream-t5x-10317785303-8G2N_fmha Expired
5.29 MB
upstream-t5x-metrics-test-log Expired
8.81 KB