Add FMHA PAXML test #3021
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-pallas / run-unit-test
amd64
/
...
/
launch-slurm-runner
1h 15m
amd64
/
...
/
launch-slurm-runner
1h 15m
Matrix: arm64 / test-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
Matrix: amd64 / test-te / run-unit-test
Matrix: amd64 / test-upstream-pax / pax-multi-node
Matrix: amd64 / test-upstream-pax / pax-single-process-evaluation
Matrix: amd64 / test-upstream-pax / 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: 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 / pax-single-process-evaluation
Waiting for pending jobs
Matrix: arm64 / test-upstream-pax / 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
finalize
/
publish-badge
3s
Annotations
22 errors
amd64 / test-pallas / runner / launch-slurm-runner
GitHub Actions has encountered an internal error when running your job.
|
amd64 / test-jax / runner / launch-slurm-runner
GitHub Actions has encountered an internal error when running your job.
|
amd64 / test-maxtext / single-process-multi-device (1, 1, 2, 4)
GitHub Actions has encountered an internal error when running your job.
|
amd64 / test-levanter / runner / launch-slurm-runner
GitHub Actions has encountered an internal error when running your job.
|
amd64 / test-gemma / runner / launch-slurm-runner
GitHub Actions has encountered an internal error when running your job.
|
amd64 / test-te / runner / launch-slurm-runner
GitHub Actions has encountered an internal error when running your job.
|
amd64 / test-te-multigpu / te-multi-gpu (2) / te-2GPU
GitHub Actions has encountered an internal error when running your job.
|
amd64 / test-te-multigpu / te-multi-gpu (8) / te-8GPU
GitHub Actions has encountered an internal error when running your job.
|
amd64 / test-te-multigpu / te-multi-gpu (4) / te-4GPU
GitHub Actions has encountered an internal error when running your job.
|
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# 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# Do not build the AMD GPU backend\nsed -i -e 's|BackendInstaller.copy(\\[\"nvidia\", \"amd\"\\])|BackendInstaller.copy([\"nvidia\"])|g' python/setup.py\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/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# This patch adds two files that are not known to CMake\nif [[ -f \"${XLA_TRITON_PATCHES}/xla_extensions/sparse_dot.patch\" ]]; then\n sed -i -e '/ConvertLayoutOpToLLVM.cpp/a ConvertLayoutOpToLLVM/SharedToSparseDotOperand.cpp' third_party/nvidia/lib/TritonNVIDIAGPUToLLVM/CMakeLists.txt\n sed -i -e '/DotOpToLLVM.cpp/a DotOpToLLVM/Sparse.cpp' third_party/nvidia/lib/TritonNVIDIAGPUToLLVM/CMakeLists.txt\nfi\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-upstream-pax / build-upstream-pax
buildx failed with: ERROR: failed to solve: process "/bin/sh -c <<\"EOF\" bash -exu -o pipefail\npushd ${SRC_PATH_LINGVO}\n\n# Use aarch distribution of protobufs\npatch -p1 <<\"EOFINNER\"\ndiff --git a/lingvo/repo.bzl b/lingvo/repo.bzl\nindex ce65822d2..d9c0277aa 100644\n--- a/lingvo/repo.bzl\n+++ b/lingvo/repo.bzl\n@@ -232,9 +232,9 @@ filegroup(\n )\n \"\"\",\n urls = [\n- \"https://github.com/protocolbuffers/protobuf/releases/download/v21.9/protoc-21.9-linux-x86_64.zip\",\n+ \"https://github.com/protocolbuffers/protobuf/releases/download/v21.9/protoc-21.9-linux-aarch_64.zip\",\n ],\n- sha256 = \"3cd951aff8ce713b94cde55e12378f505f2b89d47bf080508cf77e3934f680b6\",\n+ sha256 = \"a584286dfa8ebb17032ece206ed74d5e9931e2edb9016e427be2a0dab3b21071\",\n )\n\n def icu():\nEOFINNER\n\npip install tensorflow_datasets==4.9.2 auditwheel tensorflow==2.13.0 /opt/tensorflow_text*.whl\nsed -i 's/tensorflow=/#tensorflow=/' docker/dev.requirements.txt\nsed -i 's/tensorflow-text=/#tensorflow-text=/' docker/dev.requirements.txt\nsed -i 's/dataclasses=/#dataclasses=/' docker/dev.requirements.txt\npip install -r docker/dev.requirements.txt\n\n# Some tests are flaky right now, so we skip running the tests.\nSKIP_TESTS=1 PYTHON_MINOR_VERSION=$(python --version | cut -d ' ' -f 2 | cut -d '.' -f 2) pip_package/build.sh\nEOF" did not complete successfully: exit code: 1
|
amd64 / test-maxtext / test-maxtext-outcome
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-metrics
Process completed with exit code 1.
|
amd64 / test-rosetta-t5x / test-t5x-rosetta-outcome
Process completed with exit code 1.
|
amd64 / test-jax / jax-A100-unit-test
This request was automatically failed because there were no enabled runners online to process the request for more than 1 days.
|
amd64 / test-pallas / pallas-A100-unit-test
This request was automatically failed because there were no enabled runners online to process the request for more than 1 days.
|
amd64 / test-gemma / gemma-A100-unit-test
This request was automatically failed because there were no enabled runners online to process the request for more than 1 days.
|
amd64 / test-te / te-A100-unit-test
This request was automatically failed because there were no enabled runners online to process the request for more than 1 days.
|
amd64 / test-levanter / levanter-A100-unit-test
This request was automatically failed because there were no enabled runners online to process the request for more than 1 days.
|
Artifacts
Produced during runtime
Name | Size | |
---|---|---|
artifact-final-report
Expired
|
10.4 KB |
|
artifact-maxtext-test
Expired
|
2.5 KB |
|
artifact-pax-build-arm64
Expired
|
471 Bytes |
|
artifact-rosetta-pax-mgmn-test
Expired
|
3.74 KB |
|
artifact-rosetta-t5x-mgmn-test
Expired
|
679 Bytes |
|
artifact-triton-build-amd64
Expired
|
469 Bytes |
|
artifact-upstream-pax-mgmn-test
Expired
|
2.48 KB |
|
artifact-upstream-t5x-mgmn-test
Expired
|
1.54 KB |
|
artifact-workflow-metadata
Expired
|
277 Bytes |
|
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
|
266 Bytes |
|
final-maxtext
Expired
|
263 Bytes |
|
final-pax
Expired
|
251 Bytes |
|
final-t5x
Expired
|
251 Bytes |
|
final-upstream-pax
Expired
|
276 Bytes |
|
final-upstream-t5x
Expired
|
277 Bytes |
|
mealkit-equinox
Expired
|
272 Bytes |
|
mealkit-gemma
Expired
|
266 Bytes |
|
mealkit-grok
Expired
|
263 Bytes |
|
mealkit-jax
Expired
|
261 Bytes |
|
mealkit-levanter
Expired
|
275 Bytes |
|
mealkit-maxtext
Expired
|
272 Bytes |
|
mealkit-pax
Expired
|
262 Bytes |
|
mealkit-t5x
Expired
|
262 Bytes |
|
mealkit-upstream-pax
Expired
|
285 Bytes |
|
mealkit-upstream-t5x
Expired
|
286 Bytes |
|
rosetta-pax-9471589647-16DP1FSDP1TP1PP_TE
Expired
|
236 KB |
|
rosetta-pax-9471589647-1DP1FSDP1TP1PP_TE
Expired
|
4.83 MB |
|
rosetta-pax-9471589647-1DP2FSDP4TP1PP_single_process_TE
Expired
|
4.31 MB |
|
rosetta-pax-9471589647-1DP8FSDP1TP1PP_TE
Expired
|
4.71 MB |
|
rosetta-pax-9471589647-2DP1FSDP1TP4PP
Expired
|
2.84 MB |
|
rosetta-pax-9471589647-2DP1FSDP2TP4PP
Expired
|
250 KB |
|
rosetta-pax-9471589647-4DP1FSDP2TP1PP
Expired
|
377 KB |
|
rosetta-pax-9471589647-4DP1FSDP2TP1PP_TE
Expired
|
4.7 MB |
|
rosetta-pax-9471589647-5B_fused_attn_0
Expired
|
397 KB |
|
rosetta-pax-9471589647-5B_fused_attn_1
Expired
|
2.13 MB |
|
rosetta-pax-9471589647-8DP1FSDP1TP1PP
Expired
|
373 KB |
|
rosetta-pax-9471589647-8DP1FSDP1TP1PP_TE
Expired
|
4.88 MB |
|
rosetta-pax-9471589647-8DP1FSDP1TP1PP_eval_TE
Expired
|
445 MB |
|
rosetta-pax-9471589647-8DP1FSDP1TP1PP_single_process_TE
Expired
|
4.62 MB |
|
rosetta-pax-9471589647-8DP_TE_dropout
Expired
|
5.84 MB |
|
rosetta-pax-9471589647-LLaMA_eval_TE
Expired
|
1.44 MB |
|
rosetta-pax-metrics-test-log
Expired
|
9.11 KB |
|
rosetta-t5x-9471589647-1P1G_te-0
Expired
|
5.12 MB |
|
rosetta-t5x-9471589647-1P8G_te-1
Expired
|
5.5 MB |
|
rosetta-t5x-9471589647-2N2G_te-0
Expired
|
5.2 MB |
|
rosetta-t5x-9471589647-2N8G-te-1
Expired
|
72.6 KB |
|
rosetta-t5x-vit-9471589647-VIT1G1N
Expired
|
21.7 KB |
|
rosetta-t5x-vit-9471589647-VIT1G2N
Expired
|
23.6 KB |
|
rosetta-t5x-vit-9471589647-VIT1P8G
Expired
|
22 KB |
|
rosetta-t5x-vit-9471589647-VIT8G1N
Expired
|
33.2 KB |
|
rosetta-t5x-vit-9471589647-VIT8G2N
Expired
|
31.2 KB |
|
upstream-maxtext-9471589647-1DP1FSDP1TP1PP
Expired
|
11.2 KB |
|
upstream-maxtext-9471589647-1DP1FSDP8TP1PP
Expired
|
16 KB |
|
upstream-maxtext-9471589647-1DP4FSDP2TP1PP
Expired
|
16 KB |
|
upstream-maxtext-9471589647-1DP8FSDP1TP1PP
Expired
|
16.1 KB |
|
upstream-maxtext-9471589647-2DP2FSDP2TP1PP
Expired
|
16 KB |
|
upstream-maxtext-9471589647-4DP2FSDP2TP1PP
Expired
|
20.5 KB |
|
upstream-maxtext-metrics-test-log
Expired
|
5.27 KB |
|
upstream-pax-9471589647-16DP1FSDP1TP1PP
Expired
|
238 KB |
|
upstream-pax-9471589647-1DP1FSDP1TP1PP
Expired
|
4.75 MB |
|
upstream-pax-9471589647-1DP8FSDP1TP1PP
Expired
|
4.98 MB |
|
upstream-pax-9471589647-2DP1FSDP1TP4PP
Expired
|
2.9 MB |
|
upstream-pax-9471589647-2DP1FSDP2TP4PP
Expired
|
245 KB |
|
upstream-pax-9471589647-4DP1FSDP2TP1PP
Expired
|
5.12 MB |
|
upstream-pax-9471589647-8DP1FSDP1TP1PP
Expired
|
5 MB |
|
upstream-pax-9471589647-8DP1FSDP1TP1PP_eval
Expired
|
2.13 MB |
|
upstream-pax-9471589647-8DP1FSDP1TP1PP_single_process
Expired
|
4.72 MB |
|
upstream-pax-9471589647-8DP2FSDP4TP1PP_single_process
Expired
|
5.08 MB |
|
upstream-pax-9471589647-LLaMA_eval
Expired
|
1.47 MB |
|
upstream-pax-metrics-test-log
Expired
|
6.54 KB |
|
upstream-t5x-9471589647-1P2G_fmha
Expired
|
5.58 MB |
|
upstream-t5x-9471589647-1P8G
Expired
|
5.61 MB |
|
upstream-t5x-9471589647-8G2N
Expired
|
65.7 KB |
|
upstream-t5x-9471589647-8G2N_fmha
Expired
|
66.9 KB |
|
upstream-t5x-metrics-test-log
Expired
|
7.09 KB |
|