Skip to content

Commit c7d082c

Browse files
committed
Add more models to coreml testing
1 parent c16b645 commit c7d082c

File tree

1 file changed

+28
-3
lines changed

1 file changed

+28
-3
lines changed

.github/workflows/trunk.yml

Lines changed: 28 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -18,8 +18,8 @@ concurrency:
1818
cancel-in-progress: true
1919

2020
jobs:
21-
test-models-macos:
22-
name: test-models-macos
21+
test-models-macos-cpu:
22+
name: test-models-macos-cpu
2323
uses: pytorch/test-infra/.github/workflows/macos_job.yml@main
2424
strategy:
2525
matrix:
@@ -55,6 +55,31 @@ jobs:
5555
# Build and test executorch
5656
PYTHON_EXECUTABLE=python ${CONDA_RUN} bash .ci/scripts/test_model.sh "${MODEL_NAME}" "${BUILD_TOOL}" "${BACKEND}"
5757
58+
test-models-macos-coreml:
59+
name: test-models-macos-coreml
60+
uses: pytorch/test-infra/.github/workflows/macos_job.yml@main
61+
strategy:
62+
matrix:
63+
model: [dl3, ic3, mv3, resnet50, vit, w2l, edsr, mobilebert, emformer_join, emformer_transcribe]
64+
backend: [coreml]
65+
fail-fast: false
66+
with:
67+
runner: macos-m1-stable
68+
python-version: '3.11'
69+
submodules: 'recursive'
70+
ref: ${{ github.event_name == 'pull_request' && github.event.pull_request.head.sha || github.sha }}
71+
timeout: 90
72+
script: |
73+
MODEL_NAME=${{ matrix.model }}
74+
BUILD_TOOL=cmake
75+
BACKEND=${{ matrix.backend }}
76+
77+
bash .ci/scripts/setup-conda.sh
78+
# Setup MacOS dependencies as there is no Docker support on MacOS atm
79+
PYTHON_EXECUTABLE=python ${CONDA_RUN} bash .ci/scripts/setup-macos.sh --build-tool "${BUILD_TOOL}"
80+
# Build and test executorch
81+
PYTHON_EXECUTABLE=python ${CONDA_RUN} bash .ci/scripts/test_model.sh "${MODEL_NAME}" "${BUILD_TOOL}" "${BACKEND}"
82+
5883
test-models-linux-aarch64:
5984
name: test-models-linux-aarch64
6085
uses: pytorch/test-infra/.github/workflows/linux_job_v2.yml@main
@@ -589,7 +614,7 @@ jobs:
589614
echo "Finishing installing coreml."
590615
591616
# Build and test coreml model
592-
MODELS=(mv3 ic4 resnet50 edsr mobilebert w2l)
617+
MODELS=(mv3 ic4 resnet50)
593618
for MODEL_NAME in "${MODELS[@]}"; do
594619
echo "::group::Exporting coreml model: $MODEL_NAME"
595620
PYTHON_EXECUTABLE=python ${CONDA_RUN} bash .ci/scripts/test_model.sh "${MODEL_NAME}" "${BUILD_TOOL}" "coreml"

0 commit comments

Comments
 (0)