nsys-jax: optimise data loading and .zip creation #3633
Triggered via pull request
December 10, 2024 11:28
olupton
Status
Cancelled
Total duration
2h 42m 13s
Artifacts
47
ci.yaml
on: pull_request
metadata
0s
Matrix: amd64 / test-distribution
Matrix: arm64 / test-distribution
Matrix: amd64 / test-jax / run-unit-test
amd64
/
...
/
launch-slurm-runner
1h 16m
amd64
/
test-nsys-jax-eks
3m 58s
Matrix: amd64 / test-nsys-jax / run-unit-test
Matrix: arm64 / test-jax / run-unit-test
Waiting for pending jobs
arm64
/
test-nsys-jax-eks
0s
arm64
/
...
/
launch-slurm-runner
Matrix: arm64 / test-nsys-jax / run-unit-test
Waiting for pending jobs
Matrix: amd64 / test-te / run-unit-test
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-upstream-t5x / t5x-multi-gpu
Matrix: amd64 / test-upstream-t5x / t5x-multi-node
Matrix: amd64 / test-gemma / run-unit-test
Matrix: amd64 / test-levanter / run-unit-test
Matrix: amd64 / test-maxtext / maxtext-multinode
Matrix: amd64 / test-maxtext / single-process-multi-device
Matrix: amd64 / test-triton / run-unit-test
Matrix: amd64 / test-nsys-jax-archive
Matrix: arm64 / test-te / run-unit-test
Waiting for pending jobs
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-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
Matrix: arm64 / test-levanter / run-unit-test
Waiting for pending jobs
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
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
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
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
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
Matrix: publish-containers
Waiting for pending jobs
finalize
/
publish-badge
Annotations
35 errors and 1 warning
Artifacts
Produced during runtime
| Name | Size | |
|---|---|---|
|
artifact-base-build-amd64
|
568 Bytes |
|
|
artifact-base-build-arm64
|
566 Bytes |
|
|
artifact-equinox-build-amd64
|
569 Bytes |
|
|
artifact-equinox-build-arm64
|
568 Bytes |
|
|
artifact-gemma-build-amd64
|
558 Bytes |
|
|
artifact-jax-build-amd64
|
552 Bytes |
|
|
artifact-jax-build-arm64
|
552 Bytes |
|
|
artifact-levanter-build-amd64
|
572 Bytes |
|
|
artifact-levanter-build-arm64
|
572 Bytes |
|
|
artifact-maxtext-build-amd64
|
568 Bytes |
|
|
artifact-maxtext-build-arm64
|
568 Bytes |
|
|
artifact-pax-build-amd64
|
569 Bytes |
|
|
artifact-pax-build-arm64
|
566 Bytes |
|
|
artifact-rosetta-build-pax-amd64
|
583 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
|
567 Bytes |
|
|
artifact-t5x-build-arm64
|
567 Bytes |
|
|
artifact-triton-build-amd64
|
563 Bytes |
|
|
bumped-manifest
|
81.2 KB |
|
|
gemma-unit-test-A100
|
5.1 KB |
|
|
gemma-unit-test-V100
|
3.83 KB |
|
|
jax-unit-test-A100
|
20.2 KB |
|
|
jax-unit-test-V100
|
27 KB |
|
|
levanter-unit-test-A100
|
6.92 KB |
|
|
levanter-unit-test-V100
|
6.9 KB |
|
|
nsys-jax-unit-test-A100
|
15 KB |
|
|
nsys-jax-unit-test-V100
|
14.8 KB |
|
|
rosetta-pax-12255166222-1DP1FSDP1TP1PP_TE
|
3.55 KB |
|
|
rosetta-pax-12255166222-8DP1FSDP1TP1PP
|
6.21 KB |
|
|
rosetta-pax-12255166222-8DP1FSDP1TP1PP_eval_TE
|
3.67 KB |
|
|
rosetta-pax-12255166222-8DP1FSDP1TP1PP_single_process_TE
|
3.73 KB |
|
|
rosetta-pax-12255166222-8DP_TE_dropout
|
6.23 KB |
|
|
te-2GPU
|
6.93 KB |
|
|
te-4GPU
|
6.96 KB |
|
|
te-8GPU
|
7.01 KB |
|
|
te-multigpu-test
|
524 Bytes |
|
|
te-unit-test-A100
|
5.88 KB |
|
|
te-unit-test-V100
|
5.94 KB |
|
|
triton-unit-test-A100
|
3.07 KB |
|
|
triton-unit-test-V100
|
3.1 KB |
|
|
upstream-maxtext-12255166222-1DP1FSDP1TP1PP
|
14.9 KB |
|
|
upstream-maxtext-12255166222-1DP2FSDP4TP1PP_single_process
|
15.3 KB |
|
|
upstream-pax-12255166222-1DP1FSDP1TP1PP
|
2.65 KB |
|
|
upstream-pax-12255166222-8DP1FSDP1TP1PP_eval
|
2.74 KB |
|
|
upstream-pax-12255166222-8DP1FSDP1TP1PP_single_process
|
2.75 KB |
|