Skip to content

Fixes Imagen sampling example and updates container #3172

Fixes Imagen sampling example and updates container

Fixes Imagen sampling example and updates container #3172

Triggered via pull request August 2, 2024 23:51
@terrykongterrykong
synchronize #868
imagen-fix
Status Failure
Total duration 2h 36m 46s
Artifacts 108

ci.yaml

on: pull_request
metadata
0s
metadata
bump-manifest
25s
bump-manifest
Matrix: amd64 / test-distribution
Matrix: arm64 / test-distribution
amd64  /  ...  /  build-base
3m 45s
amd64 / build-base / build-base
arm64  /  ...  /  build-base
8m 56s
arm64 / build-base / build-base
amd64  /  ...  /  build-jax
19m 38s
amd64 / build-jax / build-jax
arm64  /  ...  /  build-jax
26m 43s
arm64 / build-jax / build-jax
amd64  /  ...  /  build-upstream-pax
6m 48s
amd64 / build-upstream-pax / build-upstream-pax
Matrix: amd64 / test-jax / run-unit-test
Matrix: amd64 / test-pallas / run-unit-test
amd64  /  ...  /  build-upstream-t5x
6m 10s
amd64 / build-upstream-t5x / build-upstream-t5x
amd64  /  ...  /  launch-slurm-runner
20m 57s
amd64 / test-jax / runner / launch-slurm-runner
amd64  /  ...  /  launch-slurm-runner
5m 41s
amd64 / test-pallas / runner / launch-slurm-runner
amd64  /  ...  /  build-gemma
5m 53s
amd64 / build-gemma / build-gemma
amd64  /  ...  /  build-levanter
3m 48s
amd64 / build-levanter / build-levanter
amd64  /  ...  /  build-maxtext
5m 31s
amd64 / build-maxtext / build-maxtext
amd64  /  ...  /  build-triton
10m 20s
amd64 / build-triton / build-triton
amd64  /  ...  /  build-equinox
2m 54s
amd64 / build-equinox / build-equinox
amd64  /  ...  /  build-grok
2m 56s
amd64 / build-grok / build-grok
arm64  /  ...  /  build-upstream-pax
27m 46s
arm64 / build-upstream-pax / build-upstream-pax
Matrix: arm64 / test-jax / run-unit-test
Waiting for pending jobs
Matrix: arm64 / test-pallas / run-unit-test
Waiting for pending jobs
arm64  /  ...  /  build-upstream-t5x
18m 14s
arm64 / build-upstream-t5x / build-upstream-t5x
arm64  /  ...  /  launch-slurm-runner
arm64 / test-jax / runner / launch-slurm-runner
arm64  /  ...  /  launch-slurm-runner
arm64 / test-pallas / runner / launch-slurm-runner
arm64  /  ...  /  build
arm64 / build-gemma / build
arm64  /  ...  /  build-levanter
5m 13s
arm64 / build-levanter / build-levanter
arm64  /  ...  /  build-maxtext
24m 28s
arm64 / build-maxtext / build-maxtext
arm64  /  ...  /  build
arm64 / build-triton / build
arm64  /  ...  /  build-equinox
3m 38s
arm64 / build-equinox / build-equinox
arm64  /  ...  /  build-grok
3m 47s
arm64 / build-grok / build-grok
amd64  /  ...  /  build-rosetta
6m 31s
amd64 / build-rosetta-pax / build-rosetta
Matrix: amd64 / test-te / run-unit-test
amd64  /  ...  /  launch-slurm-runner
1h 24m
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
1m 39s
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
26m 50s
amd64 / test-gemma / runner / launch-slurm-runner
Matrix: amd64 / test-levanter / run-unit-test
amd64  /  ...  /  launch-slurm-runner
17m 43s
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
amd64  /  ...  /  launch-slurm-runner
1h 31m
amd64 / test-triton / runner / launch-slurm-runner
arm64  /  ...  /  build-rosetta
12m 26s
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
1m 26s
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
20s
amd64 / test-upstream-pax / test-upstream-pax-metrics
amd64  /  ...  /  sitrep
5s
amd64 / test-te-multigpu / sitrep
amd64  /  collect-docker-tags
0s
amd64 / collect-docker-tags
Matrix: amd64 / test-rosetta-t5x / multi-gpu-multi-node
Waiting for pending jobs
Matrix: amd64 / test-rosetta-t5x / single-process-multi-device
Waiting for pending jobs
Matrix: amd64 / test-rosetta-t5x / vit-multi-gpu-multi-node
Waiting for pending jobs
Matrix: amd64 / test-rosetta-t5x / vit-single-process-multi-device
Waiting for pending jobs
amd64  /  ...  /  test-upstream-t5x-summary
0s
amd64 / test-upstream-t5x / test-upstream-t5x-summary
amd64  /  ...  /  test-upstream-t5x-metrics
15s
amd64 / test-upstream-t5x / test-upstream-t5x-metrics
amd64  /  ...  /  test-maxtext-summary
0s
amd64 / test-maxtext / test-maxtext-summary
amd64  /  ...  /  test-maxtext-metrics
13s
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
20s
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
amd64 / test-rosetta-t5x / test-t5x-rosetta-summary
amd64  /  ...  /  test-t5x-rosetta-metrics
amd64 / test-rosetta-t5x / test-t5x-rosetta-metrics
amd64  /  ...  /  sitrep
9s
amd64 / test-upstream-t5x / test-upstream-t5x-sitrep / sitrep
amd64  /  ...  /  sitrep
5s
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
52s
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
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
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
4s
make-publish-configs
merge-new-manifest
0s
merge-new-manifest
Matrix: publish-containers
finalize  /  workflow-badge
2m 0s
finalize / workflow-badge
finalize  /  report
16s
finalize / report
finalize  /  upload-badge
9s
finalize / upload-badge
finalize  /  publish-badge
2s
finalize / publish-badge
Fit to window
Zoom out
Zoom in

Annotations

5 errors and 6 warnings
amd64 / build-rosetta-t5x / build-rosetta
buildx failed with: ERROR: failed to solve: process "/bin/sh -c <<\"EOF\" bash -exu\nMANIFEST_DIR=$(dirname ${MANIFEST_FILE})\nif [[ \"${UPDATE_PATCHES}\" != \"true\" && \"${UPDATE_PATCHES}\" != \"false\" ]]; then\n echo \"UPDATE_PATCHES can only be true or false\"\n exit 1\nfi\nif [[ \"${UPDATE_PATCHES}\" == \"true\" ]]; then\n cp -r /mnt/jax-toolbox/.github/container/patches ${MANIFEST_DIR}/\n cp /mnt/jax-toolbox/.github/container/manifest.yaml ${MANIFEST_DIR}/manifest.yaml\n cp /mnt/jax-toolbox/.github/container/create-distribution.sh ${MANIFEST_DIR}/create-distribution.sh\n # TODO: remove\n cp /mnt/jax-toolbox/.github/container/pip-finalize.sh /usr/local/bin/\nfi\ncp -r /mnt/jax-toolbox/rosetta /opt/rosetta\n\nif [[ -n \"${UPDATED_TE_REF}\" ]]; then\n TE_INSTALL_DIR=/opt/transformer-engine\n yq e \".transformer-engine.latest_verified_commit = \\\"${UPDATED_TE_REF}\\\"\" -i $MANIFEST_FILE\n # Install from source instead of pre-built wheel\n sed -i -E 's@( file:///opt/transformer-engine)/dist/[^ ]*@\\1@' /opt/pip-tools.d/requirements-te.in\n git -C $TE_INSTALL_DIR fetch -a\n if [[ \"${UPDATED_TE_REF}\" =~ ^pull/ ]]; then\n PR_ID=$(cut -d/ -f2 <<<\"${UPDATED_TE_REF}\")\n git -C $TE_INSTALL_DIR fetch origin ${UPDATED_TE_REF}:PR-${PR_ID}\n git -C $TE_INSTALL_DIR checkout PR-${PR_ID}\n else\n git -C $TE_INSTALL_DIR checkout ${UPDATED_TE_REF}\n fi\nfi\n\n# Setting the username/email is required to author commits from patches\ngit config --global user.email \"${GIT_USER_EMAIL}\"\ngit config --global user.name \"${GIT_USER_NAME}\"\n\nbash ${MANIFEST_DIR}/create-distribution.sh \\\n --manifest ${MANIFEST_FILE} \\\n --package t5x\nbash ${MANIFEST_DIR}/create-distribution.sh \\\n --manifest ${MANIFEST_FILE} \\\n --package flax\n# Remove .gitconfig to avoid end-user authoring commits as the \"build user\"\nrm -f ~/.gitconfig\n\necho \"--extra-index-url https://developer.download.nvidia.com/compute/redist\" >> /opt/pip-tools.d/requirements-t5x.in\necho \"-e file:///opt/rosetta\" >> /opt/pip-tools.d/requirements-t5x.in\nEOF" did not complete successfully: exit code: 1
arm64 / build-rosetta-t5x / build-rosetta
buildx failed with: ERROR: failed to solve: process "/bin/sh -c <<\"EOF\" bash -exu\nMANIFEST_DIR=$(dirname ${MANIFEST_FILE})\nif [[ \"${UPDATE_PATCHES}\" != \"true\" && \"${UPDATE_PATCHES}\" != \"false\" ]]; then\n echo \"UPDATE_PATCHES can only be true or false\"\n exit 1\nfi\nif [[ \"${UPDATE_PATCHES}\" == \"true\" ]]; then\n cp -r /mnt/jax-toolbox/.github/container/patches ${MANIFEST_DIR}/\n cp /mnt/jax-toolbox/.github/container/manifest.yaml ${MANIFEST_DIR}/manifest.yaml\n cp /mnt/jax-toolbox/.github/container/create-distribution.sh ${MANIFEST_DIR}/create-distribution.sh\n # TODO: remove\n cp /mnt/jax-toolbox/.github/container/pip-finalize.sh /usr/local/bin/\nfi\ncp -r /mnt/jax-toolbox/rosetta /opt/rosetta\n\nif [[ -n \"${UPDATED_TE_REF}\" ]]; then\n TE_INSTALL_DIR=/opt/transformer-engine\n yq e \".transformer-engine.latest_verified_commit = \\\"${UPDATED_TE_REF}\\\"\" -i $MANIFEST_FILE\n # Install from source instead of pre-built wheel\n sed -i -E 's@( file:///opt/transformer-engine)/dist/[^ ]*@\\1@' /opt/pip-tools.d/requirements-te.in\n git -C $TE_INSTALL_DIR fetch -a\n if [[ \"${UPDATED_TE_REF}\" =~ ^pull/ ]]; then\n PR_ID=$(cut -d/ -f2 <<<\"${UPDATED_TE_REF}\")\n git -C $TE_INSTALL_DIR fetch origin ${UPDATED_TE_REF}:PR-${PR_ID}\n git -C $TE_INSTALL_DIR checkout PR-${PR_ID}\n else\n git -C $TE_INSTALL_DIR checkout ${UPDATED_TE_REF}\n fi\nfi\n\n# Setting the username/email is required to author commits from patches\ngit config --global user.email \"${GIT_USER_EMAIL}\"\ngit config --global user.name \"${GIT_USER_NAME}\"\n\nbash ${MANIFEST_DIR}/create-distribution.sh \\\n --manifest ${MANIFEST_FILE} \\\n --package t5x\nbash ${MANIFEST_DIR}/create-distribution.sh \\\n --manifest ${MANIFEST_FILE} \\\n --package flax\n# Remove .gitconfig to avoid end-user authoring commits as the \"build user\"\nrm -f ~/.gitconfig\n\necho \"--extra-index-url https://developer.download.nvidia.com/compute/redist\" >> /opt/pip-tools.d/requirements-t5x.in\necho \"-e file:///opt/rosetta\" >> /opt/pip-tools.d/requirements-t5x.in\nEOF" did not complete successfully: exit code: 1
amd64 / test-upstream-t5x / test-upstream-t5x-outcome
Process completed with exit code 1.
amd64 / test-upstream-pax / test-upstream-pax-outcome
Process completed with exit code 1.
amd64 / test-rosetta-pax / test-pax-rosetta-outcome
Process completed with exit code 1.
publish-containers (grok, mock-jax-mealkit, 500, ghcr.io/nvidia/jax-toolbox-internal:10223441272-... / publish
Failed to download action 'https://api.github.com/repos/docker/login-action/tarball/9780b0c442fbb1117ed29e0efdff1e18412f7567'. Error: The request was canceled due to the configured HttpClient.Timeout of 100 seconds elapsing.
publish-containers (grok, mock-jax-mealkit, 500, ghcr.io/nvidia/jax-toolbox-internal:10223441272-... / publish
Failed to download action 'https://api.github.com/repos/docker/login-action/tarball/9780b0c442fbb1117ed29e0efdff1e18412f7567'. Error: The request was canceled due to the configured HttpClient.Timeout of 100 seconds elapsing.
finalize / workflow-badge
Failed to download action 'https://api.github.com/repos/actions/upload-artifact/tarball/89ef406dd8d7e03cfd12d9e0a4a378f454709029'. Error: The request was canceled due to the configured HttpClient.Timeout of 100 seconds elapsing.
finalize / workflow-badge
Back off 16.739 seconds before retry.

Artifacts

Produced during runtime
Name Size
artifact-base-build-amd64
566 Bytes
artifact-base-build-arm64
567 Bytes
artifact-equinox-build-amd64
569 Bytes
artifact-equinox-build-arm64
568 Bytes
artifact-final-report
10.1 KB
artifact-gemma-build-amd64
560 Bytes
artifact-grok-build-amd64
558 Bytes
artifact-grok-build-arm64
557 Bytes
artifact-jax-build-amd64
554 Bytes
artifact-jax-build-arm64
554 Bytes
artifact-levanter-build-amd64
572 Bytes
artifact-levanter-build-arm64
571 Bytes
artifact-maxtext-build-amd64
568 Bytes
artifact-maxtext-build-arm64
567 Bytes
artifact-maxtext-test
2.74 KB
artifact-pax-build-amd64
569 Bytes
artifact-pax-build-arm64
568 Bytes
artifact-rosetta-build-pax-amd64
583 Bytes
artifact-rosetta-build-pax-arm64
584 Bytes
artifact-rosetta-build-t5x-amd64
489 Bytes
artifact-rosetta-build-t5x-arm64
489 Bytes
artifact-rosetta-pax-mgmn-test
2.8 KB
artifact-t5x-build-amd64
568 Bytes
artifact-t5x-build-arm64
567 Bytes
artifact-triton-build-amd64
566 Bytes
artifact-upstream-pax-mgmn-test
2.46 KB
artifact-upstream-t5x-mgmn-test
2.09 KB
artifact-workflow-metadata
277 Bytes
bumped-manifest
71.5 KB
final-base
254 Bytes
final-equinox
263 Bytes
final-gemma
257 Bytes
final-grok
254 Bytes
final-jax
251 Bytes
final-levanter
266 Bytes
final-maxtext
262 Bytes
final-pax
251 Bytes
final-triton
259 Bytes
final-upstream-pax
276 Bytes
final-upstream-t5x
276 Bytes
gemma-unit-test-A100
4.23 KB
gemma-unit-test-V100
2.88 KB
jax-unit-test-A100
16.8 KB
jax-unit-test-V100
17.2 KB
levanter-unit-test-A100
9.81 KB
levanter-unit-test-V100
9.35 KB
mealkit-equinox
272 Bytes
mealkit-gemma
265 Bytes
mealkit-grok
263 Bytes
mealkit-jax
261 Bytes
mealkit-levanter
274 Bytes
mealkit-maxtext
271 Bytes
mealkit-pax
261 Bytes
mealkit-triton
267 Bytes
mealkit-upstream-pax
285 Bytes
mealkit-upstream-t5x
286 Bytes
pallas-unit-test-A100
7.25 KB
pallas-unit-test-V100
7.14 KB
rosetta-pax-10223441272-16DP1FSDP1TP1PP_TE
587 KB
rosetta-pax-10223441272-1DP1FSDP1TP1PP_TE
86.1 KB
rosetta-pax-10223441272-1DP2FSDP4TP1PP_single_process_TE
105 KB
rosetta-pax-10223441272-1DP8FSDP1TP1PP_TE
328 KB
rosetta-pax-10223441272-2DP1FSDP1TP4PP
199 KB
rosetta-pax-10223441272-2DP1FSDP2TP4PP
354 KB
rosetta-pax-10223441272-4DP1FSDP2TP1PP
357 KB
rosetta-pax-10223441272-4DP1FSDP2TP1PP_TE
319 KB
rosetta-pax-10223441272-5B_fused_attn_0
200 KB
rosetta-pax-10223441272-5B_fused_attn_1
205 KB
rosetta-pax-10223441272-8DP1FSDP1TP1PP
360 KB
rosetta-pax-10223441272-8DP1FSDP1TP1PP_TE
321 KB
rosetta-pax-10223441272-8DP1FSDP1TP1PP_eval_TE
76.4 KB
rosetta-pax-10223441272-8DP1FSDP1TP1PP_single_process_TE
105 KB
rosetta-pax-10223441272-8DP_TE_dropout
331 KB
rosetta-pax-10223441272-LLaMA_eval_TE
223 KB
rosetta-pax-metrics-test-log
8.55 KB
te-2GPU
5.94 KB
te-4GPU
6.01 KB
te-8GPU
6.06 KB
te-multigpu-test
524 Bytes
te-unit-test-A100
377 KB
te-unit-test-V100
376 KB
triton-unit-test-A100
4.81 KB
triton-unit-test-V100
4.76 KB
upstream-maxtext-10223441272-1DP1FSDP1TP1PP
12.7 KB
upstream-maxtext-10223441272-1DP1FSDP8TP1PP
17.7 KB
upstream-maxtext-10223441272-1DP2FSDP4TP1PP_single_process
12.9 KB
upstream-maxtext-10223441272-1DP4FSDP2TP1PP
17.9 KB
upstream-maxtext-10223441272-1DP8FSDP1TP1PP
17.9 KB
upstream-maxtext-10223441272-2DP2FSDP2TP1PP
17.9 KB
upstream-maxtext-10223441272-4DP2FSDP2TP1PP
24.1 KB
upstream-maxtext-metrics-test-log
4.55 KB
upstream-pax-10223441272-16DP1FSDP1TP1PP
716 KB
upstream-pax-10223441272-1DP1FSDP1TP1PP
141 KB
upstream-pax-10223441272-1DP2FSDP4TP1PP_single_process
141 KB
upstream-pax-10223441272-1DP8FSDP1TP1PP
410 KB
upstream-pax-10223441272-2DP1FSDP1TP4PP
197 KB
upstream-pax-10223441272-2DP1FSDP2TP4PP
347 KB
upstream-pax-10223441272-4DP1FSDP2TP1PP
411 KB
upstream-pax-10223441272-8DP1FSDP1TP1PP
419 KB
upstream-pax-10223441272-8DP1FSDP1TP1PP_eval
107 KB
upstream-pax-10223441272-8DP1FSDP1TP1PP_single_process
141 KB
upstream-pax-10223441272-LLaMA_eval
235 KB
upstream-pax-metrics-test-log
6.75 KB
upstream-t5x-10223441272-1P2G_fmha
5.17 MB
upstream-t5x-10223441272-1P8G
5.27 MB
upstream-t5x-10223441272-8G2N
5.38 MB
upstream-t5x-10223441272-8G2N_fmha
5.29 MB
upstream-t5x-metrics-test-log
8.73 KB