Skip to content

nsys-jax: re-work to be more pip-install-able #3562

nsys-jax: re-work to be more pip-install-able

nsys-jax: re-work to be more pip-install-able #3562

You are viewing an older attempt in the history of this workflow run. View latest attempt.
Triggered via pull request November 22, 2024 15:14
Status Failure
Total duration 23h 53m 52s
Artifacts

ci.yaml

on: pull_request
metadata
3s
metadata
bump-manifest
35s
bump-manifest
Matrix: amd64 / test-distribution
Matrix: arm64 / test-distribution
amd64  /  ...  /  build-base
4m 27s
amd64 / build-base / build-base
arm64  /  ...  /  build-base
4m 46s
arm64 / build-base / build-base
amd64  /  ...  /  build-jax
8m 58s
amd64 / build-jax / build-jax
arm64  /  ...  /  build-jax
15m 27s
arm64 / build-jax / build-jax
amd64  /  ...  /  build-upstream-pax
7m 1s
amd64 / build-upstream-pax / build-upstream-pax
Matrix: amd64 / test-jax / run-unit-test
Matrix: amd64 / test-nsys-jax / run-unit-test
amd64  /  ...  /  build-upstream-t5x
6m 44s
amd64 / build-upstream-t5x / build-upstream-t5x
amd64  /  ...  /  launch-slurm-runner
3h 25m
amd64 / test-jax / runner / launch-slurm-runner
amd64  /  ...  /  launch-slurm-runner
3h 7m
amd64 / test-nsys-jax / runner / launch-slurm-runner
amd64  /  ...  /  build-gemma
5m 49s
amd64 / build-gemma / build-gemma
amd64  /  ...  /  build-levanter
4m 39s
amd64 / build-levanter / build-levanter
amd64  /  ...  /  build-maxtext
8m 24s
amd64 / build-maxtext / build-maxtext
amd64  /  ...  /  build-triton
9m 10s
amd64 / build-triton / build-triton
amd64  /  ...  /  build-equinox
3m 57s
amd64 / build-equinox / build-equinox
arm64  /  ...  /  build-upstream-pax
15m 12s
arm64 / build-upstream-pax / build-upstream-pax
Matrix: arm64 / test-jax / run-unit-test
Waiting for pending jobs
Matrix: arm64 / test-nsys-jax / run-unit-test
Waiting for pending jobs
arm64  /  ...  /  build-upstream-t5x
12m 47s
arm64 / build-upstream-t5x / build-upstream-t5x
arm64  /  ...  /  launch-slurm-runner
arm64 / test-jax / runner / launch-slurm-runner
arm64  /  ...  /  launch-slurm-runner
arm64 / test-nsys-jax / runner / launch-slurm-runner
arm64  /  ...  /  build
arm64 / build-gemma / build
arm64  /  ...  /  build-levanter
6m 37s
arm64 / build-levanter / build-levanter
arm64  /  ...  /  build-maxtext
10m 2s
arm64 / build-maxtext / build-maxtext
arm64  /  ...  /  build
arm64 / build-triton / build
arm64  /  ...  /  build-equinox
4m 37s
arm64 / build-equinox / build-equinox
amd64  /  ...  /  build-rosetta
7m 3s
amd64 / build-rosetta-pax / build-rosetta
Matrix: amd64 / test-te / run-unit-test
amd64  /  ...  /  launch-slurm-runner
5h 22m
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
8m 23s
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
3h 40m
amd64 / test-gemma / runner / launch-slurm-runner
Matrix: amd64 / test-levanter / run-unit-test
amd64  /  ...  /  launch-slurm-runner
3h 39m
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
Waiting for pending jobs
amd64  /  ...  /  launch-slurm-runner
amd64 / test-triton / runner / launch-slurm-runner
arm64  /  ...  /  build-rosetta
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
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
0s
amd64 / test-upstream-pax / test-upstream-pax-metrics
amd64  /  ...  /  sitrep
4s
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
0s
amd64 / test-upstream-t5x / test-upstream-t5x-metrics
amd64  /  ...  /  test-maxtext-summary
0s
amd64 / test-maxtext / test-maxtext-summary
amd64  /  ...  /  test-maxtext-metrics
0s
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
0s
amd64 / test-rosetta-pax / test-pax-rosetta-metrics
amd64  /  ...  /  sitrep
9s
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
0s
amd64 / test-rosetta-t5x / test-t5x-rosetta-metrics
amd64  /  ...  /  sitrep
7s
amd64 / test-upstream-t5x / test-upstream-t5x-sitrep / sitrep
amd64  /  ...  /  sitrep
7s
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
7s
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
12s
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
3s
make-publish-configs
merge-new-manifest
0s
merge-new-manifest
Matrix: publish-containers
finalize  /  workflow-badge
3s
finalize / workflow-badge
finalize  /  report
22s
finalize / report
finalize  /  upload-badge
16s
finalize / upload-badge
finalize  /  publish-badge
3s
finalize / publish-badge
Fit to window
Zoom out
Zoom in

Annotations

40 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-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
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
amd64 / test-levanter / levanter-V100-unit-test
Process completed with exit code 1.
amd64 / test-nsys-jax / nsys-jax-A100-unit-test
Process completed with exit code 1.
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-pktbb 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-rosetta-t5x / single-process-multi-device (1P1G_te-0, 1, --enable-te 0)
The job running on runner jumpbox-vc69x-qrztw has exceeded the maximum execution time of 360 minutes.
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-bf9ml 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-cgfm4 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-s46wd 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-upstream-t5x / t5x-multi-node (8G2N_fmha, 8, 2, --enable-fmha 1)
The job running on runner jumpbox-vc69x-lm9px has exceeded the maximum execution time of 360 minutes.
amd64 / test-upstream-t5x / test-upstream-t5x-outcome
Process completed with exit code 1.
amd64 / test-rosetta-t5x / vit-multi-gpu-multi-node (1, 2)
The job running on runner jumpbox-vc69x-zd29f 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-t5x / multi-gpu-multi-node (2N2G_te-0, 2, 2, --enable-te 0)
The job running on runner jumpbox-vc69x-9lwpw has exceeded the maximum execution time of 360 minutes.
amd64 / test-rosetta-pax / rosetta-pax-multi-node (4, 2, 1, 2)
The job running on runner jumpbox-vc69x-wf92m 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-96gsn has exceeded the maximum execution time of 360 minutes.
amd64 / test-maxtext / maxtext-multinode (1, 4, 2, 2)
The job running on runner jumpbox-vc69x-wct7p 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-vjwz9 has exceeded the maximum execution time of 360 minutes.
amd64 / test-rosetta-pax / test-pax-rosetta-outcome
Process completed with exit code 1.
amd64 / test-rosetta-t5x / vit-multi-gpu-multi-node (8, 2)
The job running on runner jumpbox-vc69x-dzzd8 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-upstream-pax / pax-multi-node (2DP1FSDP2TP4PP, 4, 2, 1, 2, 4)
The job running on runner jumpbox-vc69x-78pt2 has exceeded the maximum execution time of 360 minutes.
amd64 / test-upstream-pax / test-upstream-pax-outcome
Process completed with exit code 1.

Artifacts

Produced during runtime
Name Size
artifact-base-build-amd64
567 Bytes
artifact-base-build-arm64
566 Bytes
artifact-equinox-build-amd64
569 Bytes
artifact-equinox-build-arm64
569 Bytes
artifact-final-report
4.8 KB
artifact-gemma-build-amd64
561 Bytes
artifact-jax-build-amd64
553 Bytes
artifact-jax-build-arm64
553 Bytes
artifact-levanter-build-amd64
573 Bytes
artifact-levanter-build-arm64
572 Bytes
artifact-maxtext-build-amd64
568 Bytes
artifact-maxtext-build-arm64
515 Bytes
artifact-maxtext-test
653 Bytes
artifact-pax-build-amd64
567 Bytes
artifact-pax-build-arm64
514 Bytes
artifact-rosetta-build-pax-amd64
583 Bytes
artifact-rosetta-build-t5x-amd64
585 Bytes
artifact-rosetta-pax-mgmn-test
737 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-pax-mgmn-test
688 Bytes
artifact-upstream-t5x-mgmn-test
634 Bytes
artifact-workflow-metadata
277 Bytes
bumped-manifest
80.1 KB
final-base
254 Bytes
final-equinox
263 Bytes
final-gemma
257 Bytes
final-jax
251 Bytes
final-levanter
266 Bytes
final-maxtext
263 Bytes
final-pax
251 Bytes
final-t5x
251 Bytes
final-upstream-pax
276 Bytes
final-upstream-t5x
276 Bytes
gemma-unit-test-A100
5.18 KB
gemma-unit-test-V100
3.94 KB
jax-unit-test-A100
18.6 KB
jax-unit-test-V100
24.1 KB
levanter-unit-test-A100
13.2 KB
levanter-unit-test-V100
15.1 KB
mealkit-equinox
272 Bytes
mealkit-gemma
265 Bytes
mealkit-jax
261 Bytes
mealkit-levanter
275 Bytes
mealkit-maxtext
272 Bytes
mealkit-pax
262 Bytes
mealkit-t5x
262 Bytes
mealkit-upstream-pax
286 Bytes
mealkit-upstream-t5x
286 Bytes
nsys-jax-unit-test-A100
29.3 MB
nsys-jax-unit-test-V100
15.8 MB
rosetta-pax-11975232090-1DP1FSDP1TP1PP_TE
92.5 KB
rosetta-pax-11975232090-1DP2FSDP4TP1PP_single_process_TE
112 KB
rosetta-pax-11975232090-1DP8FSDP1TP1PP_TE
376 KB
rosetta-pax-11975232090-2DP1FSDP1TP4PP
251 KB
rosetta-pax-11975232090-4DP1FSDP2TP1PP
399 KB
rosetta-pax-11975232090-4DP1FSDP2TP1PP_TE
371 KB
rosetta-pax-11975232090-5B_fused_attn_0
420 KB
rosetta-pax-11975232090-5B_fused_attn_1
427 KB
rosetta-pax-11975232090-8DP1FSDP1TP1PP
394 KB
rosetta-pax-11975232090-8DP1FSDP1TP1PP_TE
372 KB
rosetta-pax-11975232090-8DP1FSDP1TP1PP_eval_TE
84.1 KB
rosetta-pax-11975232090-8DP1FSDP1TP1PP_single_process_TE
112 KB
rosetta-pax-11975232090-8DP_TE_dropout
383 KB
rosetta-pax-11975232090-LLaMA_eval_TE
259 KB
rosetta-t5x-11975232090-1P8G_te-1
35.3 KB
rosetta-t5x-vit-11975232090-VIT8G1N
32.7 KB
te-2GPU
5.35 KB
te-4GPU
5.41 KB
te-8GPU
5.44 KB
te-multigpu-test
529 Bytes
te-unit-test-A100
840 KB
te-unit-test-V100
679 KB
upstream-maxtext-11975232090-1DP1FSDP1TP1PP
2.53 KB
upstream-maxtext-11975232090-1DP1FSDP8TP1PP
4.38 KB
upstream-maxtext-11975232090-1DP2FSDP4TP1PP_single_process
2.63 KB
upstream-maxtext-11975232090-1DP4FSDP2TP1PP
4.32 KB
upstream-maxtext-11975232090-1DP8FSDP1TP1PP
4.38 KB
upstream-maxtext-11975232090-2DP2FSDP2TP1PP
4.44 KB
upstream-pax-11975232090-1DP1FSDP1TP1PP
147 KB
upstream-pax-11975232090-1DP2FSDP4TP1PP_single_process
147 KB
upstream-pax-11975232090-1DP8FSDP1TP1PP
446 KB
upstream-pax-11975232090-2DP1FSDP1TP4PP
318 KB
upstream-pax-11975232090-4DP1FSDP2TP1PP
455 KB
upstream-pax-11975232090-8DP1FSDP1TP1PP
454 KB
upstream-pax-11975232090-8DP1FSDP1TP1PP_eval
114 KB
upstream-pax-11975232090-8DP1FSDP1TP1PP_single_process
147 KB
upstream-pax-11975232090-LLaMA_eval
268 KB
upstream-t5x-11975232090-1P2G_fmha
34 KB
upstream-t5x-11975232090-1P8G
33.6 KB