Skip to content

Commit f240557

Browse files
committed
[Backend Tester] Run Core ML tests in nightly CI
ghstack-source-id: a27976c ghstack-comment-id: 3190288611 Pull-Request: #13446
1 parent 78d5401 commit f240557

File tree

2 files changed

+47
-0
lines changed

2 files changed

+47
-0
lines changed

.ci/scripts/backend-test-macos.sh

Lines changed: 24 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,24 @@
1+
#!/usr/bin/env bash
2+
# Copyright (c) Meta Platforms, Inc. and affiliates.
3+
# All rights reserved.
4+
#
5+
# This source code is licensed under the BSD-style license found in the
6+
# LICENSE file in the root directory of this source tree.
7+
set -eux
8+
9+
SUITE=$1
10+
FLOW=$2
11+
ARTIFACT_DIR=$3
12+
13+
echo "Running backend test job for suite $SUITE, flow $FLOW."
14+
echo "Saving job artifacts to $ARTIFACT_DIR."
15+
16+
${CONDA_RUN} --no-capture-output pip install awscli==1.37.21
17+
18+
bash .ci/scripts/setup-conda.sh
19+
eval "$(conda shell.bash hook)"
20+
21+
PYTHON_EXECUTABLE=python
22+
${CONDA_RUN} --no-capture-output .ci/scripts/setup-macos.sh --build-tool cmake --build-mode Release
23+
24+
${CONDA_RUN} --no-capture-output python -m executorch.backends.test.suite.runner $SUITE --flow $FLOW --report "$ARTIFACT_DIR/test_results.csv"

.github/workflows/nightly.yml

Lines changed: 23 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -57,3 +57,26 @@ jobs:
5757
# TODO (gjcomer) Remove this when jobs are stable.
5858
.ci/scripts/backend-test-linux.sh "${{ matrix.suite }}" "${{ matrix.flow }}" "${RUNNER_ARTIFACT_DIR}" || EXIT_CODE=$?
5959
echo "Test run complete with exit code $EXIT_CODE."
60+
61+
backend-test-macos:
62+
uses: pytorch/test-infra/.github/workflows/macos_job.yml@main
63+
strategy:
64+
fail-fast: false
65+
matrix:
66+
flow: [coreml, coreml_static_int8]
67+
suite: [models, operators]
68+
with:
69+
ref: ${{ github.event_name == 'pull_request' && github.event.pull_request.head.sha || github.sha }}
70+
runner: macos-latest-xlarge
71+
python-version: 3.12
72+
submodules: recursive
73+
timeout: 90
74+
upload-artifact: test-report-${{ matrix.flow }}-${{ matrix.suite }}
75+
script: |
76+
set -eux
77+
78+
# This is needed to get the prebuilt PyTorch wheel from S3
79+
${CONDA_RUN} --no-capture-output pip install awscli==1.37.21
80+
81+
.ci/scripts/backend-test-macos.sh "${{ matrix.suite }}" "${{ matrix.flow }}" "${RUNNER_ARTIFACT_DIR}" || EXIT_CODE=$?
82+
echo "Test run complete with exit code $EXIT_CODE."

0 commit comments

Comments
 (0)