@@ -273,47 +273,47 @@ jobs:
273
273
# Test selective build
274
274
PYTHON_EXECUTABLE=python bash examples/portable/scripts/test_demo_backend_delegation.sh "${BUILD_TOOL}"
275
275
276
- # test-arm-backend:
277
- # name: test-arm-backend
278
- # uses: pytorch/test-infra/.github/workflows/linux_job_v2.yml@main
279
- # permissions:
280
- # id-token: write
281
- # contents: read
282
- # strategy:
283
- # matrix:
284
- # include:
285
- # - test_arm_baremetal: test_pytest_ops_ethosu_fvp
286
- # - test_arm_baremetal: test_pytest_models_ethosu_fvp
287
- # - test_arm_baremetal: test_run_ethosu_fvp
288
- # - test_arm_baremetal: test_models_tosa
289
- # - test_arm_baremetal: test_models_ethos-u55
290
- # - test_arm_baremetal: test_models_ethos-u85
291
- # - test_arm_baremetal: test_smaller_stories_llama
292
- # fail-fast: false
293
- # with:
294
- # runner: linux.2xlarge.memory
295
- # docker-image: ci-image:executorch-ubuntu-22.04-arm-sdk
296
- # submodules: 'recursive'
297
- # ref: ${{ github.event_name == 'pull_request' && github.event.pull_request.head.sha || github.sha }}
298
- # timeout: 120
299
- # script: |
300
- # # The generic Linux job chooses to use base env, not the one setup by the image
301
- # CONDA_ENV=$(conda env list --json | jq -r ".envs | .[-1]")
302
- # conda activate "${CONDA_ENV}"
276
+ test-arm-backend :
277
+ name : test-arm-backend
278
+ uses : pytorch/test-infra/.github/workflows/linux_job_v2.yml@main
279
+ permissions :
280
+ id-token : write
281
+ contents : read
282
+ strategy :
283
+ matrix :
284
+ include :
285
+ - test_arm_baremetal : test_pytest_ops_ethosu_fvp
286
+ - test_arm_baremetal : test_pytest_models_ethosu_fvp
287
+ - test_arm_baremetal : test_run_ethosu_fvp
288
+ - test_arm_baremetal : test_models_tosa
289
+ - test_arm_baremetal : test_models_ethos-u55
290
+ - test_arm_baremetal : test_models_ethos-u85
291
+ - test_arm_baremetal : test_smaller_stories_llama
292
+ fail-fast : false
293
+ with :
294
+ runner : linux.2xlarge.memory
295
+ docker-image : ci-image:executorch-ubuntu-22.04-arm-sdk
296
+ submodules : ' recursive'
297
+ ref : ${{ github.event_name == 'pull_request' && github.event.pull_request.head.sha || github.sha }}
298
+ timeout : 120
299
+ script : |
300
+ # The generic Linux job chooses to use base env, not the one setup by the image
301
+ CONDA_ENV=$(conda env list --json | jq -r ".envs | .[-1]")
302
+ conda activate "${CONDA_ENV}"
303
303
304
- # source .ci/scripts/utils.sh
305
- # install_executorch "--use-pt-pinned-commit"
304
+ source .ci/scripts/utils.sh
305
+ install_executorch "--use-pt-pinned-commit"
306
306
307
- # .ci/scripts/setup-arm-baremetal-tools.sh
307
+ .ci/scripts/setup-arm-baremetal-tools.sh
308
308
309
- # # Increase number of files user can monitor to bypass buck failures.
310
- # # Hopefully this is high enough for this setup.
311
- # sudo sysctl fs.inotify.max_user_watches=1048576 # 1024 * 1024
309
+ # Increase number of files user can monitor to bypass buck failures.
310
+ # Hopefully this is high enough for this setup.
311
+ sudo sysctl fs.inotify.max_user_watches=1048576 # 1024 * 1024
312
312
313
- # ARM_TEST=${{ matrix.test_arm_baremetal }}
313
+ ARM_TEST=${{ matrix.test_arm_baremetal }}
314
314
315
- # # Test test_arm_baremetal.sh with test
316
- # backends/arm/test/test_arm_baremetal.sh "${ARM_TEST}"
315
+ # Test test_arm_baremetal.sh with test
316
+ backends/arm/test/test_arm_baremetal.sh "${ARM_TEST}"
317
317
318
318
test-arm-cortex-m-size-test :
319
319
name : test-arm-cortex-m-size-test
@@ -666,71 +666,71 @@ jobs:
666
666
PYTHON_EXECUTABLE=python bash .ci/scripts/build-qnn-sdk.sh
667
667
PYTHON_EXECUTABLE=python bash .ci/scripts/test_model.sh ${{ matrix.model }} "cmake" "qnn"
668
668
669
- # test-models-macos-coreml:
670
- # name: test-models-macos-coreml
671
- # uses: pytorch/test-infra/.github/workflows/macos_job.yml@main
672
- # strategy:
673
- # matrix:
674
- # model: [dl3, edsr, efficient_sam, emformer_join, emformer_transcribe, ic3, ic4, mobilebert, mv2, mv3, resnet50, vit, w2l]
675
- # fail-fast: false
676
- # with:
677
- # runner: macos-m1-stable
678
- # python-version: '3.11'
679
- # submodules: 'recursive'
680
- # ref: ${{ github.event_name == 'pull_request' && github.event.pull_request.head.sha || github.sha }}
681
- # timeout: 90
682
- # script: |
683
- # MODEL_NAME=${{ matrix.model }}
684
- # BUILD_TOOL=cmake
685
- # BACKEND="coreml-pybind"
669
+ test-models-macos-coreml :
670
+ name : test-models-macos-coreml
671
+ uses : pytorch/test-infra/.github/workflows/macos_job.yml@main
672
+ strategy :
673
+ matrix :
674
+ model : [dl3, edsr, efficient_sam, emformer_join, emformer_transcribe, ic3, ic4, mobilebert, mv2, mv3, resnet50, vit, w2l]
675
+ fail-fast : false
676
+ with :
677
+ runner : macos-m1-stable
678
+ python-version : ' 3.11'
679
+ submodules : ' recursive'
680
+ ref : ${{ github.event_name == 'pull_request' && github.event.pull_request.head.sha || github.sha }}
681
+ timeout : 90
682
+ script : |
683
+ MODEL_NAME=${{ matrix.model }}
684
+ BUILD_TOOL=cmake
685
+ BACKEND="coreml-pybind"
686
686
687
687
688
- # # Set model specific overrides
689
- # if [[ "${MODEL_NAME}" == "mobilebert" ]]; then
690
- # # See https://github.com/pytorch/executorch/issues/12907
691
- # # mobilebert has nan output on FP16, and high MSE on fp32, so we disable runtime test now
692
- # BACKEND="coreml"
693
- # fi
688
+ # Set model specific overrides
689
+ if [[ "${MODEL_NAME}" == "mobilebert" ]]; then
690
+ # See https://github.com/pytorch/executorch/issues/12907
691
+ # mobilebert has nan output on FP16, and high MSE on fp32, so we disable runtime test now
692
+ BACKEND="coreml"
693
+ fi
694
694
695
- # if [[ "${MODEL_NAME}" == "efficient_sam" ]]; then
696
- # # See https://github.com/pytorch/executorch/issues/12906
697
- # # efficient_sam fails to run on CoreML
698
- # BACKEND="coreml"
699
- # fi
695
+ if [[ "${MODEL_NAME}" == "efficient_sam" ]]; then
696
+ # See https://github.com/pytorch/executorch/issues/12906
697
+ # efficient_sam fails to run on CoreML
698
+ BACKEND="coreml"
699
+ fi
700
700
701
- # bash .ci/scripts/setup-conda.sh
701
+ bash .ci/scripts/setup-conda.sh
702
702
703
- # # Setup MacOS dependencies as there is no Docker support on MacOS atm
704
- # PYTHON_EXECUTABLE=python ${CONDA_RUN} bash .ci/scripts/setup-macos.sh --build-tool "${BUILD_TOOL}"
705
- # PYTHON_EXECUTABLE=python ${CONDA_RUN} bash backends/apple/coreml/scripts/install_requirements.sh
706
- # echo "Finishing installing coreml."
703
+ # Setup MacOS dependencies as there is no Docker support on MacOS atm
704
+ PYTHON_EXECUTABLE=python ${CONDA_RUN} bash .ci/scripts/setup-macos.sh --build-tool "${BUILD_TOOL}"
705
+ PYTHON_EXECUTABLE=python ${CONDA_RUN} bash backends/apple/coreml/scripts/install_requirements.sh
706
+ echo "Finishing installing coreml."
707
707
708
- # PYTHON_EXECUTABLE=python ${CONDA_RUN} bash .ci/scripts/test_model.sh "${MODEL_NAME}" "${BUILD_TOOL}" "${BACKEND}"
708
+ PYTHON_EXECUTABLE=python ${CONDA_RUN} bash .ci/scripts/test_model.sh "${MODEL_NAME}" "${BUILD_TOOL}" "${BACKEND}"
709
709
710
- # test-models-macos-mps:
711
- # name: test-models-macos-mps
712
- # uses: pytorch/test-infra/.github/workflows/macos_job.yml@main
713
- # strategy:
714
- # fail-fast: false
715
- # with:
716
- # runner: macos-m1-stable
717
- # python-version: '3.11'
718
- # submodules: 'recursive'
719
- # ref: ${{ github.event_name == 'pull_request' && github.event.pull_request.head.sha || github.sha }}
720
- # timeout: 90
721
- # script: |
722
- # BUILD_TOOL=cmake
723
- # bash .ci/scripts/setup-conda.sh
710
+ test-models-macos-mps :
711
+ name : test-models-macos-mps
712
+ uses : pytorch/test-infra/.github/workflows/macos_job.yml@main
713
+ strategy :
714
+ fail-fast : false
715
+ with :
716
+ runner : macos-m1-stable
717
+ python-version : ' 3.11'
718
+ submodules : ' recursive'
719
+ ref : ${{ github.event_name == 'pull_request' && github.event.pull_request.head.sha || github.sha }}
720
+ timeout : 90
721
+ script : |
722
+ BUILD_TOOL=cmake
723
+ bash .ci/scripts/setup-conda.sh
724
724
725
- # # Setup MacOS dependencies as there is no Docker support on MacOS atm
726
- # PYTHON_EXECUTABLE=python ${CONDA_RUN} bash .ci/scripts/setup-macos.sh --build-tool "${BUILD_TOOL}"
725
+ # Setup MacOS dependencies as there is no Docker support on MacOS atm
726
+ PYTHON_EXECUTABLE=python ${CONDA_RUN} bash .ci/scripts/setup-macos.sh --build-tool "${BUILD_TOOL}"
727
727
728
- # # Build and test mps model
729
- # for MODEL_NAME in mv3 ic4 resnet50 edsr mobilebert w2l; do
730
- # echo "::group::Exporting mps model: $MODEL_NAME"
731
- # PYTHON_EXECUTABLE=python ${CONDA_RUN} bash .ci/scripts/test_model.sh "${MODEL_NAME}" "${BUILD_TOOL}" "mps"
732
- # echo "::endgroup::"
733
- # done
728
+ # Build and test mps model
729
+ for MODEL_NAME in mv3 ic4 resnet50 edsr mobilebert w2l; do
730
+ echo "::group::Exporting mps model: $MODEL_NAME"
731
+ PYTHON_EXECUTABLE=python ${CONDA_RUN} bash .ci/scripts/test_model.sh "${MODEL_NAME}" "${BUILD_TOOL}" "mps"
732
+ echo "::endgroup::"
733
+ done
734
734
735
735
test-huggingface-transformers-xnnpack :
736
736
# NB: Don't run this on fork PRs because they won't have access to the secret and would fail anyway
0 commit comments