nsys-jax: re-work to be more pip-install-able #3569
Triggered via pull request
November 25, 2024 10:46
Status
Cancelled
Total duration
1d 0h 21m 2s
Artifacts
61
ci.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: arm64 / test-jax / run-unit-test
Waiting for pending jobs
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
Waiting for pending jobs
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
58 errors
amd64 / build-triton / build-triton
buildx failed with: ERROR: failed to solve: process "/bin/sh -c <<\"EOF\" bash -ex\npushd /opt/checkout/external/triton\nmkdir dist\n# Make sure the wheel is labelled as 3.1 (it is ~HEAD), not 3.0. The upstream branch\n# structure seems to have a 3.1 release branch that has not been merged back to main.\nsed -i -e 's|version=\"3\\.0\\.0\"|version=\"3.1.0\"|g' python/setup.py\n# Do not compile with -Werror\nsed -i -e 's|-Werror||g' CMakeLists.txt\n# The LLVM build above does not enable these libraries\nsed -i -e 's|\\(LLVMAMDGPU.*\\)|# \\1|g' CMakeLists.txt\n# Do not build tests\nsed -i -e 's|^\\s*add_subdirectory(unittest)|# unit tests disabled|' CMakeLists.txt\n# Avoid error due to forward declaration of Module\nsed -i -e '/#include \"llvm\\/IR\\/IRBuilder.h\"/a #include \"llvm/IR/Module.h\"' test/lib/Instrumentation/GPUHello.cpp\n# Google has patches that mess with include paths in source files\nsed -i -e '/include_directories(${PROJECT_SOURCE_DIR}\\/third_party)/a include_directories(${PROJECT_SOURCE_DIR}/third_party/amd/include)' CMakeLists.txt\nsed -i -e '/include_directories(${PROJECT_BINARY_DIR}\\/third_party)/a include_directories(${PROJECT_BINARY_DIR}/third_party/amd/include)' CMakeLists.txt\nsed -i -e '/include_directories(${PROJECT_SOURCE_DIR}\\/third_party)/a include_directories(${PROJECT_SOURCE_DIR}/third_party/nvidia/include)' CMakeLists.txt\nsed -i -e '/include_directories(${PROJECT_BINARY_DIR}\\/third_party)/a include_directories(${PROJECT_BINARY_DIR}/third_party/nvidia/include)' CMakeLists.txt\n# Extra patches to Triton maintained in XLA. These are already applied in the working directory.\nXLA_TRITON_PATCHES=\"${SRC_PATH_XLA}/third_party/triton\"\n# Use clang to match Google etc.\nLLVM_INCLUDE_DIRS=/opt/llvm-build/include \\\n LLVM_LIBRARY_DIR=/opt/llvm-build/lib \\\n LLVM_SYSPATH=/opt/llvm-build \\\n TRITON_BUILD_WITH_CLANG_LLD=1 \\\n pip wheel --verbose --wheel-dir=dist/ python/\n# Clean up the wheel build directory so it doesn't end up bloating the container\nrm -rf python/build\nEOF" did not complete successfully: exit code: 1
|
arm64 / build-maxtext / build-maxtext
buildx failed with: ERROR: failed to solve: process "/bin/sh -c pip-finalize.sh" did not complete successfully: exit code: 1
|
arm64 / build-upstream-pax / build-upstream-pax
buildx failed with: ERROR: failed to solve: process "/bin/sh -c pip-finalize.sh" did not complete successfully: exit code: 1
|
arm64 / build-upstream-t5x / build-upstream-t5x
buildx failed with: ERROR: failed to solve: process "/bin/sh -c pip-finalize.sh" did not complete successfully: exit code: 1
|
amd64 / test-levanter / levanter-V100-unit-test
Process completed with exit code 1.
|
amd64 / test-nsys-jax-archive (ubuntu-22.04)
Process completed with exit code 127.
|
amd64 / test-nsys-jax-archive (macOS-latest)
Process completed with exit code 127.
|
amd64 / test-nsys-jax-archive (ubuntu-24.04)
The job was canceled because "ubuntu-22_04" failed.
|
amd64 / test-nsys-jax-archive (ubuntu-24.04)
Process completed with exit code 127.
|
amd64 / test-levanter / levanter-A100-unit-test
Process completed with exit code 1.
|
amd64 / test-upstream-t5x / t5x-multi-node (8G2N, 8, 2)
The job running on runner jumpbox-vc69x-zvvns has exceeded the maximum execution time of 360 minutes.
|
amd64 / test-upstream-t5x / t5x-multi-node (8G2N, 8, 2)
The operation was canceled.
|
amd64 / test-upstream-pax / single-process-multi-device (1, 8, 1, 1)
The job running on runner jumpbox-vc69x-t7fgn has exceeded the maximum execution time of 360 minutes.
|
amd64 / test-upstream-pax / single-process-multi-device (1, 8, 1, 1)
The operation was canceled.
|
amd64 / test-upstream-pax / pax-multi-node (1DP1FSDP1TP1PP, 1, 1, 1, 1, 4)
The job running on runner jumpbox-vc69x-dl5ck has exceeded the maximum execution time of 360 minutes.
|
amd64 / test-upstream-pax / pax-multi-node (1DP1FSDP1TP1PP, 1, 1, 1, 1, 4)
The operation was canceled.
|
amd64 / test-upstream-pax / single-process-evaluation (1, 8, 1, 1)
The job running on runner jumpbox-vc69x-wsbsj has exceeded the maximum execution time of 360 minutes.
|
amd64 / test-upstream-pax / single-process-evaluation (1, 8, 1, 1)
The operation was canceled.
|
amd64 / test-rosetta-t5x / multi-gpu-multi-node (2N8G-te-1, 8, 2, --gin.train/utils.DatasetConfig.pack=False --gin.train_eva...
The job running on runner jumpbox-vc69x-6ddpb has exceeded the maximum execution time of 360 minutes.
|
|
amd64 / test-rosetta-t5x / vit-multi-gpu-multi-node (1, 1)
The job running on runner jumpbox-vc69x-j4nh5 has exceeded the maximum execution time of 360 minutes.
|
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)
The job running on runner jumpbox-vc69x-fp9tr has exceeded the maximum execution time of 360 minutes.
|
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)
The job running on runner jumpbox-vc69x-vxckz has exceeded the maximum execution time of 360 minutes.
|
amd64 / test-rosetta-t5x / single-process-multi-device (1P1G_te-0, 1, --enable-te 0)
The operation was canceled.
|
amd64 / test-rosetta-pax / single-process-multi-device-te (1, 8, 1, 1)
The job running on runner jumpbox-vc69x-lzn4q has exceeded the maximum execution time of 360 minutes.
|
amd64 / test-rosetta-pax / single-process-multi-device-te (1, 8, 1, 1)
The operation was canceled.
|
amd64 / test-rosetta-pax / single-process-evaluation-te (1, 8, 1, 1)
The job running on runner jumpbox-vc69x-swwsp has exceeded the maximum execution time of 360 minutes.
|
amd64 / test-rosetta-pax / single-process-evaluation-te (1, 8, 1, 1)
The operation was canceled.
|
amd64 / test-rosetta-pax / rosetta-pax-multi-node (1, 8, 1, 1)
The job running on runner jumpbox-vc69x-2kz2j has exceeded the maximum execution time of 360 minutes.
|
amd64 / test-rosetta-pax / rosetta-pax-multi-node (1, 8, 1, 1)
The operation was canceled.
|
amd64 / test-rosetta-pax / rosetta-pax-multi-node-te (1DP1FSDP1TP1PP_TE, 1, 1, 1, 1, 4)
The job running on runner jumpbox-vc69x-ffnzb has exceeded the maximum execution time of 360 minutes.
|
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-single-node-dropout-te (1, 8, 1, 1)
The job running on runner jumpbox-vc69x-89xc7 has exceeded the maximum execution time of 360 minutes.
|
amd64 / test-rosetta-pax / rosetta-pax-single-node-dropout-te (1, 8, 1, 1)
The operation was canceled.
|
amd64 / test-upstream-t5x / t5x-multi-node (8G2N_fmha, 8, 2, --enable-fmha 1)
The job running on runner jumpbox-vc69x-zpjwn has exceeded the maximum execution time of 360 minutes.
|
amd64 / test-upstream-t5x / t5x-multi-node (8G2N_fmha, 8, 2, --enable-fmha 1)
The operation was canceled.
|
amd64 / test-upstream-t5x / test-upstream-t5x-outcome
Process completed with exit code 1.
|
amd64 / test-rosetta-t5x / multi-gpu-multi-node (2N2G_te-0, 2, 2, --enable-te 0)
The job running on runner jumpbox-vc69x-8mmnc has exceeded the maximum execution time of 360 minutes.
|
amd64 / test-rosetta-t5x / multi-gpu-multi-node (2N2G_te-0, 2, 2, --enable-te 0)
The operation was canceled.
|
amd64 / test-rosetta-t5x / vit-multi-gpu-multi-node (1, 2)
The job running on runner jumpbox-vc69x-jqx2n has exceeded the maximum execution time of 360 minutes.
|
amd64 / test-rosetta-t5x / vit-multi-gpu-multi-node (1, 2)
The operation was canceled.
|
amd64 / test-rosetta-pax / rosetta-pax-multi-node (4, 2, 1, 2)
The job running on runner jumpbox-vc69x-6fqg4 has exceeded the maximum execution time of 360 minutes.
|
amd64 / test-rosetta-pax / rosetta-pax-multi-node (4, 2, 1, 2)
The operation was canceled.
|
amd64 / test-rosetta-pax / rosetta-pax-multi-node-te (16DP1FSDP1TP1PP_TE, 1, 16, 1, 1, 4)
The job running on runner jumpbox-vc69x-9vm49 has exceeded the maximum execution time of 360 minutes.
|
amd64 / test-rosetta-pax / rosetta-pax-multi-node-te (16DP1FSDP1TP1PP_TE, 1, 16, 1, 1, 4)
The operation was canceled.
|
amd64 / test-rosetta-t5x / vit-multi-gpu-multi-node (8, 2)
The job running on runner jumpbox-vc69x-6w5rb has exceeded the maximum execution time of 360 minutes.
|
amd64 / test-rosetta-t5x / vit-multi-gpu-multi-node (8, 2)
The operation was canceled.
|
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-maxtext / maxtext-multinode (1, 4, 2, 2)
The job running on runner jumpbox-vc69x-7rrjh has exceeded the maximum execution time of 360 minutes.
|
amd64 / test-maxtext / maxtext-multinode (1, 4, 2, 2)
The operation was canceled.
|
amd64 / test-maxtext / test-maxtext-outcome
Process completed with exit code 1.
|
amd64 / test-upstream-pax / pax-multi-node (16DP1FSDP1TP1PP, 1, 16, 1, 1, 4)
The job running on runner jumpbox-vc69x-46kml has exceeded the maximum execution time of 360 minutes.
|
amd64 / test-upstream-pax / pax-multi-node (16DP1FSDP1TP1PP, 1, 16, 1, 1, 4)
The operation was canceled.
|
amd64 / test-upstream-pax / pax-multi-node (2DP1FSDP2TP4PP, 4, 2, 1, 2, 4)
Canceling since a higher priority waiting request for 'CI-olupton/pip-install-nsys-jax' exists
|
amd64 / test-upstream-pax / pax-multi-node (2DP1FSDP2TP4PP, 4, 2, 1, 2, 4)
The operation was canceled.
|
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
|
567 Bytes |
|
artifact-gemma-build-amd64
|
560 Bytes |
|
artifact-jax-build-amd64
|
553 Bytes |
|
artifact-jax-build-arm64
|
553 Bytes |
|
artifact-levanter-build-amd64
|
572 Bytes |
|
artifact-levanter-build-arm64
|
571 Bytes |
|
artifact-maxtext-build-amd64
|
567 Bytes |
|
artifact-maxtext-build-arm64
|
515 Bytes |
|
artifact-maxtext-test
|
654 Bytes |
|
artifact-pax-build-amd64
|
567 Bytes |
|
artifact-pax-build-arm64
|
515 Bytes |
|
artifact-rosetta-build-pax-amd64
|
583 Bytes |
|
artifact-rosetta-build-t5x-amd64
|
584 Bytes |
|
artifact-rosetta-pax-mgmn-test
|
713 Bytes |
|
artifact-rosetta-t5x-mgmn-test
|
640 Bytes |
|
artifact-t5x-build-amd64
|
568 Bytes |
|
artifact-t5x-build-arm64
|
516 Bytes |
|
artifact-triton-build-amd64
|
469 Bytes |
|
artifact-upstream-t5x-mgmn-test
|
634 Bytes |
|
bumped-manifest
|
80.1 KB |
|
gemma-unit-test-A100
|
5.19 KB |
|
gemma-unit-test-V100
|
3.95 KB |
|
jax-unit-test-A100
|
19 KB |
|
jax-unit-test-V100
|
25.4 KB |
|
levanter-unit-test-A100
|
13.1 KB |
|
levanter-unit-test-V100
|
15.1 KB |
|
nsys-jax-unit-test-A100
|
29.1 MB |
|
nsys-jax-unit-test-V100
|
15.9 MB |
|
rosetta-pax-12008367805-1DP2FSDP4TP1PP_single_process_TE
|
112 KB |
|
rosetta-pax-12008367805-1DP8FSDP1TP1PP_TE
|
369 KB |
|
rosetta-pax-12008367805-2DP1FSDP1TP4PP
|
251 KB |
|
rosetta-pax-12008367805-4DP1FSDP2TP1PP
|
409 KB |
|
rosetta-pax-12008367805-4DP1FSDP2TP1PP_TE
|
376 KB |
|
rosetta-pax-12008367805-5B_fused_attn_0
|
420 KB |
|
rosetta-pax-12008367805-5B_fused_attn_1
|
438 KB |
|
rosetta-pax-12008367805-8DP1FSDP1TP1PP_TE
|
377 KB |
|
rosetta-pax-12008367805-LLaMA_eval_TE
|
261 KB |
|
rosetta-t5x-12008367805-1P8G_te-1
|
35.3 KB |
|
rosetta-t5x-vit-12008367805-VIT8G1N
|
32.4 KB |
|
te-2GPU
|
5.37 KB |
|
te-4GPU
|
5.38 KB |
|
te-8GPU
|
5.43 KB |
|
te-multigpu-test
|
528 Bytes |
|
te-unit-test-A100
|
840 KB |
|
te-unit-test-V100
|
679 KB |
|
upstream-maxtext-12008367805-1DP1FSDP1TP1PP
|
14.6 KB |
|
upstream-maxtext-12008367805-1DP1FSDP8TP1PP
|
20.7 KB |
|
upstream-maxtext-12008367805-1DP2FSDP4TP1PP_single_process
|
15 KB |
|
upstream-maxtext-12008367805-1DP4FSDP2TP1PP
|
20.9 KB |
|
upstream-maxtext-12008367805-1DP8FSDP1TP1PP
|
20.7 KB |
|
upstream-maxtext-12008367805-2DP2FSDP2TP1PP
|
21.1 KB |
|
upstream-pax-12008367805-1DP2FSDP4TP1PP_single_process
|
147 KB |
|
upstream-pax-12008367805-1DP8FSDP1TP1PP
|
450 KB |
|
upstream-pax-12008367805-2DP1FSDP1TP4PP
|
325 KB |
|
upstream-pax-12008367805-4DP1FSDP2TP1PP
|
458 KB |
|
upstream-pax-12008367805-8DP1FSDP1TP1PP
|
454 KB |
|
upstream-t5x-12008367805-1P2G_fmha
|
34 KB |
|
upstream-t5x-12008367805-1P8G
|
33.6 KB |
|