Skip to content

test-jax.sh: avoid instantiating TPU tests, which breaks on arm #4374

test-jax.sh: avoid instantiating TPU tests, which breaks on arm

test-jax.sh: avoid instantiating TPU tests, which breaks on arm #4374

Triggered via pull request May 16, 2025 15:04
Status Failure
Total duration 4h 48m 54s
Artifacts 42

ci.yaml

on: pull_request
metadata
0s
metadata
bump-manifest
17s
bump-manifest
Matrix: amd64 / test-distribution
Matrix: arm64 / test-distribution
amd64  /  ...  /  build-base
3m 24s
amd64 / build-base / build-base
arm64  /  ...  /  build-base
3m 37s
arm64 / build-base / build-base
amd64  /  ...  /  build-jax
13m 2s
amd64 / build-jax / build-jax
arm64  /  ...  /  build-jax
20m 36s
arm64 / build-jax / build-jax
Matrix: amd64 / test-jax / run-unit-test
Matrix: amd64 / test-te-h100 / te-test-eks
Matrix: amd64 / test-te-unit-a100 / run-unit-test
amd64  /  ...  /  launch-slurm-runner
26m 28s
amd64 / test-jax / runner / launch-slurm-runner
amd64  /  test-nsys-jax-eks
12m 49s
amd64 / test-nsys-jax-eks
amd64  /  ...  /  launch-slurm-runner
51m 8s
amd64 / test-te-unit-a100 / runner / launch-slurm-runner
amd64  /  ...  /  build-maxtext
9m 29s
amd64 / build-maxtext / build-maxtext
amd64  /  ...  /  build-upstream-t5x
6m 55s
amd64 / build-upstream-t5x / build-upstream-t5x
amd64  /  ...  /  build-axlearn
5m 20s
amd64 / build-axlearn / build-axlearn
Matrix: amd64 / test-nsys-jax / run-unit-test
amd64  /  ...  /  build-equinox
4m 7s
amd64 / build-equinox / build-equinox
amd64  /  ...  /  launch-slurm-runner
37m 9s
amd64 / test-nsys-jax / runner / launch-slurm-runner
Matrix: arm64 / test-jax / run-unit-test
Waiting for pending jobs
Matrix: arm64 / test-te-h100 / te-test-eks
Waiting for pending jobs
Matrix: arm64 / test-te-unit-a100 / run-unit-test
Waiting for pending jobs
arm64  /  test-nsys-jax-eks
0s
arm64 / test-nsys-jax-eks
arm64  /  ...  /  launch-slurm-runner
arm64 / test-jax / runner / launch-slurm-runner
arm64  /  ...  /  launch-slurm-runner
arm64 / test-te-unit-a100 / runner / launch-slurm-runner
arm64  /  ...  /  build-maxtext
11m 18s
arm64 / build-maxtext / build-maxtext
arm64  /  ...  /  build-upstream-t5x
9m 13s
arm64 / build-upstream-t5x / build-upstream-t5x
arm64  /  ...  /  build-axlearn
7m 41s
arm64 / build-axlearn / build-axlearn
Matrix: arm64 / test-nsys-jax / run-unit-test
Waiting for pending jobs
arm64  /  ...  /  build-equinox
6m 56s
arm64 / build-equinox / build-equinox
arm64  /  ...  /  launch-slurm-runner
arm64 / test-nsys-jax / runner / launch-slurm-runner
Matrix: amd64 / test-maxtext / maxtext-multinode
Matrix: amd64 / test-maxtext / single-process-multi-device
amd64  /  ...  /  build-rosetta
15m 9s
amd64 / build-rosetta-t5x / build-rosetta
amd64  /  test-axlearn-eks
4h 20m
amd64 / test-axlearn-eks
amd64  /  test-axlearn-fuji-models-eks
1h 58m
amd64 / test-axlearn-fuji-models-eks
Matrix: amd64 / test-nsys-jax-archive
Matrix: arm64 / test-maxtext / maxtext-multinode
Waiting for pending jobs
Matrix: arm64 / test-maxtext / single-process-multi-device
Waiting for pending jobs
arm64  /  ...  /  build-rosetta
17m 31s
arm64 / build-rosetta-t5x / build-rosetta
arm64  /  test-axlearn-eks
0s
arm64 / test-axlearn-eks
arm64  /  test-axlearn-fuji-models-eks
0s
arm64 / test-axlearn-fuji-models-eks
Matrix: arm64 / test-nsys-jax-archive
amd64  /  ...  /  test-maxtext-metrics
15s
amd64 / test-maxtext / test-maxtext-metrics
amd64  /  collect-docker-tags
0s
amd64 / collect-docker-tags
Matrix: amd64 / test-rosetta-t5x / vit-multi-gpu-multi-node
arm64  /  ...  /  test-maxtext-metrics
arm64 / test-maxtext / test-maxtext-metrics
arm64  /  collect-docker-tags
0s
arm64 / collect-docker-tags
Matrix: arm64 / test-rosetta-t5x / vit-multi-gpu-multi-node
Waiting for pending jobs
amd64  /  ...  /  sitrep
5s
amd64 / test-maxtext / test-maxtext-sitrep / sitrep
amd64  /  ...  /  test-t5x-rosetta-summary
0s
amd64 / test-rosetta-t5x / test-t5x-rosetta-summary
amd64  /  ...  /  test-t5x-rosetta-metrics
23s
amd64 / test-rosetta-t5x / test-t5x-rosetta-metrics
arm64  /  ...  /  sitrep
arm64 / test-maxtext / test-maxtext-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
amd64  /  ...  /  test-maxtext-outcome
0s
amd64 / test-maxtext / test-maxtext-outcome
amd64  /  ...  /  sitrep
20s
amd64 / test-rosetta-t5x / test-t5x-rosetta-sitrep / sitrep
arm64  /  ...  /  test-maxtext-outcome
arm64 / test-maxtext / test-maxtext-outcome
arm64  /  ...  /  sitrep
arm64 / test-rosetta-t5x / test-t5x-rosetta-sitrep / sitrep
amd64  /  ...  /  test-t5x-rosetta-outcome
0s
amd64 / test-rosetta-t5x / test-t5x-rosetta-outcome
arm64  /  ...  /  test-t5x-rosetta-outcome
arm64 / test-rosetta-t5x / test-t5x-rosetta-outcome
make-publish-configs
2s
make-publish-configs
merge-new-manifest
0s
merge-new-manifest
Matrix: publish-containers
finalize  /  workflow-badge
3s
finalize / workflow-badge
finalize  /  report
20s
finalize / report
finalize  /  upload-badge
9s
finalize / upload-badge
finalize  /  publish-badge
3s
finalize / publish-badge
Fit to window
Zoom out
Zoom in

Annotations

1 error and 1 warning
amd64 / test-axlearn-eks
Process completed with exit code 1.
amd64 / test-axlearn-eks
This job failure may be caused by using an out of date version of GitHub runner on your self-hosted runner. You are currently using GitHub runner version 2.323.0. Please update to the latest version 2.324.0

Artifacts

Produced during runtime
Name Size Digest
artifact-axlearn-build-amd64
566 Bytes
sha256:5473a2e892c7c8ec9a63c482fca258f18afbb18fa678c09b253cfcc1cc082656
artifact-axlearn-build-arm64
566 Bytes
sha256:8c9afda28e66d18116a15598ed501e8cce1511732e14c28e9b575f1a22fc44b3
artifact-axlearn-test
103 KB
sha256:216ceb5a51a63e2eb9a4909accf0790b5044d79b9fab838422417229934857e9
artifact-base-build-amd64
567 Bytes
sha256:b99f67b36f3f767a20df2538921037d6f54f20785ca704e58145208aa3590802
artifact-base-build-arm64
568 Bytes
sha256:9f6852d12ba322cc142ddf50b301d95db49e0191568778340598d59d83099aab
artifact-equinox-build-amd64
570 Bytes
sha256:bd5ec4ad0c867a8388620ab8edb7d27a2ab95375336b2b9a77408e6b1324eae5
artifact-equinox-build-arm64
568 Bytes
sha256:e9ad382538784246cb8605ac5ca8f0d4e5fbb433836c66fd19cef5d89a0cd20a
artifact-final-report
3.92 KB
sha256:8c7b4cd57c310e2948bca48d847712a01d5a59a7d313aebc3078a29444a08135
artifact-jax-build-amd64
554 Bytes
sha256:090238aebd2489f3080825d1ea744446fe1a6624e9881f6d756c7f50ec6766f9
artifact-jax-build-arm64
554 Bytes
sha256:97e393f5352ee70ca526a2f8cdfad5a00b4ed9517706147f41baee24025b53d8
artifact-maxtext-build-amd64
568 Bytes
sha256:1e8cfad501e049b123bdcf64760ad0ffbfa95b7a6d20d6e59c5016853963408d
artifact-maxtext-build-arm64
567 Bytes
sha256:9473cc4c6a3df0ed8a74d25be131cd4d2376215453630163aecd2277dc7a13ef
artifact-maxtext-test
1.84 KB
sha256:34f6f621cdbf9be3bbaa7b77e4881f46b01ba972c5beb648868480e120243c2f
artifact-multigpu-test-transformerengine-15071513017-8gpu-unittest
584 Bytes
sha256:13910779f176239625347a4107a0b930fab816f3ff39508a456cccd5ad00e723
artifact-rosetta-build-t5x-amd64
583 Bytes
sha256:8f35234c319bd6e3786731eb112a27cfe011091a46f87a07ae982b4cc443ba0a
artifact-rosetta-build-t5x-arm64
584 Bytes
sha256:596f9811392936b2c2124dc0d1057dd68bcdf142f6204961c6d89aab29740815
artifact-rosetta-t5x-mgmn-test
1.28 KB
sha256:2663bf10613c5a4097912bcd3fa292d380fdf9fff16d9ddddc30aca64b9c8d2d
artifact-t5x-build-amd64
567 Bytes
sha256:f428d83188e22dfe8b9d5b300d05e6ce62c7b0ff3af294d7ca949814db64b91d
artifact-t5x-build-arm64
567 Bytes
sha256:ba86ae0177da0bec8797ae3dc7f0a5140ee6fbe21bba35f5d1ab0d87907a18d3
artifact-workflow-metadata
277 Bytes
sha256:ced95e0a6b1e52010ad66bbbc52ef059573de5d3f8e5890a1abcb801eafeb9f1
bumped-manifest
46.9 KB
sha256:38222caa760eb724afeaeaf0631fec81053f68f83e708edbab8b5dc5085eb41e
final-axlearn
263 Bytes
sha256:e28c46ca2b0b866d8ec1850ee9cf6ce07c3cc8f47088ebd4a2d7aec01d925d14
final-base
254 Bytes
sha256:a62f540b9e858acd3ede27c2ad5b38deaa12c65c53466cefe0de342106867c71
final-equinox
263 Bytes
sha256:3548d9685bed31048c932bcc70e73286c2f628cccaa8efa23e0fe7094bbcc06e
final-jax
251 Bytes
sha256:34df8bd371b345ddf1ce8b8bd07fd3a5a554ab065615e9ecee3e73336f55a0cf
final-maxtext
263 Bytes
sha256:b212ae9f382eca29ac77d14c5afc2ac08465f716585eb3414e67f998821bc8f1
final-t5x
251 Bytes
sha256:0482d9d3b2d0fe6e3d715ef9b8ed00b61811d7db7b60dd08e3bdf56f799c65fa
final-upstream-t5x
276 Bytes
sha256:29ace8fc61351c45ff8bd1f794b73df5293eec775a4bb43cc12659e7926ece42
jax-unit-test-A100
19.4 KB
sha256:4e341ed14a2b77f3c7bf3cdd2effa9c747f3751095cb7043fa0a723f43d31229
mealkit-axlearn
272 Bytes
sha256:455e9af91e2c8de7162255bfd0f91c9ee1c19cc65b031594b12073feed42e66a
mealkit-equinox
273 Bytes
sha256:c8f09c92a3dcf64d899d5dfdab8ba322d6a763860e5aeed9581159abb25a7be0
mealkit-jax
261 Bytes
sha256:9ef2d917d16e29322ed1a8b3236d99bcfc5ac454cd507d86112ec32c4629e245
mealkit-maxtext
272 Bytes
sha256:a6a97fc582766fd730ac4e90792dbc803e2cb5f2eb50639b00dbe0d2eb5d454e
mealkit-t5x
261 Bytes
sha256:0563a0882ca86bd6a3fb9381306a6805132ff667ede6281f6370454c0cc234bd
mealkit-upstream-t5x
286 Bytes
sha256:9d697a83c1e3e6a30e15d6829141902ceee082096db635c4610b5f6d7d80dd93
nsys-jax-unit-test-A100
33.1 MB
sha256:939d1e3bdb228836689b8183dc25239115310a1e7d6d508d549a7d726fa6bba2
rosetta-t5x-metrics-test-log
1.03 KB
sha256:ee6804fcb519a30329913cef6d318ba467b8ed0e77173c02a3c95095f9691c7f
rosetta-t5x-vit-15071513017-VIT8G1N
32.4 KB
sha256:37808f8e0929e02d1a93a989f904849786c5001e2e8184778cf08760827b67fe
te-unit-test-A100
939 KB
sha256:bc8ff0b9e62c85bd31665299565bfacfa471cce8c077c98961eef11b6b5db810
upstream-maxtext-15071513017-1DP2FSDP4TP1PP_single_process
20 KB
sha256:c2558b59a8f9b5ad92c191f36fe54a1c8d769cd19edf593619cd7f468bbfdfe4
upstream-maxtext-15071513017-2DP2FSDP2TP1PP
30.2 KB
sha256:2c0da7ea0a326263a57545797be5ba8b300ce2d8d70dd3690dca4784c67991f0
upstream-maxtext-metrics-test-log
1.83 KB
sha256:a9cfb9fc1cea0639f5990318abb6c3548063388841ee504cb5a3c16bf7a33d29