Skip to content

Commit 23a6d11

Browse files
authored
Add tests on graviton instances (#9207)
- **Create Arm64 Graviton instance docker image** #9199 - **Test models on linux aarch64 graviton instances** In the next follow-up PR, I will trim macos jobs
1 parent 5402716 commit 23a6d11

File tree

1 file changed

+31
-0
lines changed

1 file changed

+31
-0
lines changed

.github/workflows/trunk.yml

Lines changed: 31 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -63,6 +63,37 @@ jobs:
6363
# Build and test executorch
6464
PYTHON_EXECUTABLE=python ${CONDA_RUN} bash .ci/scripts/test_model.sh "${MODEL_NAME}" "${BUILD_TOOL}" "${BACKEND}" "${DEMO_BACKEND_DELEGATION}"
6565
66+
test-models-linux-aarch64:
67+
name: test-models-linux-aarch64
68+
uses: pytorch/test-infra/.github/workflows/linux_job_v2.yml@main
69+
permissions:
70+
id-token: write
71+
contents: read
72+
strategy:
73+
matrix:
74+
model: [linear, add, add_mul, ic3, ic4, mv2, mv3, resnet18, resnet50, vit, w2l, mobilebert, emformer_join, emformer_transcribe]
75+
backend: [portable, xnnpack-quantization-delegation]
76+
runner: [linux.arm64.2xlarge]
77+
fail-fast: false
78+
with:
79+
runner: ${{ matrix.runner }}
80+
docker-image: executorch-ubuntu-22.04-gcc11-aarch64
81+
submodules: 'true'
82+
ref: ${{ github.event_name == 'pull_request' && github.event.pull_request.head.sha || github.sha }}
83+
timeout: 90
84+
script: |
85+
# The generic Linux job chooses to use base env, not the one setup by the image
86+
CONDA_ENV=$(conda env list --json | jq -r ".envs | .[-1]")
87+
conda activate "${CONDA_ENV}"
88+
89+
MODEL_NAME=${{ matrix.model }}
90+
BUILD_TOOL="cmake"
91+
BACKEND=${{ matrix.backend }}
92+
93+
PYTHON_EXECUTABLE=python bash .ci/scripts/setup-linux.sh --build-tool "${BUILD_TOOL}"
94+
# Build and test ExecuTorch
95+
PYTHON_EXECUTABLE=python bash .ci/scripts/test_model.sh "${MODEL_NAME}" "${BUILD_TOOL}" "${BACKEND}"
96+
6697
test-custom-ops-macos:
6798
name: test-custom-ops-macos
6899
uses: pytorch/test-infra/.github/workflows/macos_job.yml@main

0 commit comments

Comments
 (0)