nsys-jax post-processing: treat host-device copies as 1-device collectives #3374
oluptonci.yaml
on: pull_request
metadata
0s
Matrix: amd64 / test-distribution
Matrix: arm64 / test-distribution
Matrix: amd64 / test-jax / run-unit-test
Matrix: amd64 / test-nsys-jax / run-unit-test
Matrix: amd64 / test-pallas / run-unit-test
amd64
/
...
/
launch-slurm-runner
1h 43m
amd64
/
...
/
launch-slurm-runner
1h 51m
amd64
/
...
/
launch-slurm-runner
1h 49m
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
/
...
/
launch-slurm-runner
arm64
/
...
/
launch-slurm-runner
arm64
/
...
/
launch-slurm-runner
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: 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: 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
84 errors and 1 warning
Artifacts
Produced during runtime
| Name | Size | |
|---|---|---|
|
artifact-base-build-amd64
|
567 Bytes |
|
|
artifact-base-build-arm64
|
567 Bytes |
|
|
artifact-equinox-build-amd64
|
569 Bytes |
|
|
artifact-equinox-build-arm64
|
569 Bytes |
|
|
artifact-gemma-build-amd64
|
560 Bytes |
|
|
artifact-jax-build-amd64
|
553 Bytes |
|
|
artifact-jax-build-arm64
|
554 Bytes |
|
|
artifact-levanter-build-amd64
|
572 Bytes |
|
|
artifact-levanter-build-arm64
|
572 Bytes |
|
|
artifact-maxtext-build-amd64
|
568 Bytes |
|
|
artifact-maxtext-build-arm64
|
567 Bytes |
|
|
artifact-pax-build-amd64
|
565 Bytes |
|
|
artifact-pax-build-arm64
|
567 Bytes |
|
|
artifact-rosetta-build-pax-amd64
|
584 Bytes |
|
|
artifact-rosetta-build-pax-arm64
|
584 Bytes |
|
|
artifact-rosetta-build-t5x-amd64
|
585 Bytes |
|
|
artifact-rosetta-build-t5x-arm64
|
530 Bytes |
|
|
artifact-t5x-build-amd64
|
568 Bytes |
|
|
artifact-t5x-build-arm64
|
567 Bytes |
|
|
artifact-triton-build-amd64
|
565 Bytes |
|
|
bumped-manifest
|
71.3 KB |
|
|
gemma-unit-test-A100
|
4.09 KB |
|
|
gemma-unit-test-V100
|
2.74 KB |
|
|
jax-unit-test-A100
|
19.4 KB |
|
|
jax-unit-test-V100
|
23 KB |
|
|
levanter-unit-test-A100
|
6.67 KB |
|
|
levanter-unit-test-V100
|
6.68 KB |
|
|
nsys-jax-unit-test-A100
|
30.1 MB |
|
|
nsys-jax-unit-test-V100
|
15.9 MB |
|
|
pallas-unit-test-A100
|
7.42 KB |
|
|
pallas-unit-test-V100
|
7.27 KB |
|
|
te-2GPU
|
5.37 KB |
|
|
te-4GPU
|
5.42 KB |
|
|
te-8GPU
|
5.43 KB |
|
|
te-multigpu-test
|
524 Bytes |
|
|
te-unit-test-V100
|
626 KB |
|
|
triton-unit-test-V100
|
4.83 KB |
|
|
upstream-maxtext-11124481819-1DP1FSDP1TP1PP
|
13.9 KB |
|
|
upstream-maxtext-11124481819-1DP1FSDP8TP1PP
|
16.9 KB |
|
|
upstream-maxtext-11124481819-1DP2FSDP4TP1PP_single_process
|
14 KB |
|
|
upstream-maxtext-11124481819-1DP4FSDP2TP1PP
|
19.3 KB |
|
|
upstream-maxtext-11124481819-1DP8FSDP1TP1PP
|
19.1 KB |
|
|
upstream-maxtext-11124481819-2DP2FSDP2TP1PP
|
19.1 KB |
|
|
upstream-pax-11124481819-1DP2FSDP4TP1PP_single_process
|
142 KB |
|
|
upstream-pax-11124481819-2DP1FSDP1TP4PP
|
296 KB |
|
|
upstream-pax-11124481819-4DP1FSDP2TP1PP
|
419 KB |
|
|
upstream-pax-11124481819-8DP1FSDP1TP1PP_eval
|
108 KB |
|
|
upstream-pax-11124481819-8DP1FSDP1TP1PP_single_process
|
142 KB |
|
|
upstream-pax-11124481819-LLaMA_eval
|
242 KB |
|