@@ -273,47 +273,47 @@ jobs:
273273 # Test selective build
274274 PYTHON_EXECUTABLE=python bash examples/portable/scripts/test_demo_backend_delegation.sh "${BUILD_TOOL}"
275275
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}"
303303
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"
306306
307- # .ci/scripts/setup-arm-baremetal-tools.sh
307+ .ci/scripts/setup-arm-baremetal-tools.sh
308308
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
312312
313- # ARM_TEST=${{ matrix.test_arm_baremetal }}
313+ ARM_TEST=${{ matrix.test_arm_baremetal }}
314314
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}"
317317
318318 test-arm-cortex-m-size-test :
319319 name : test-arm-cortex-m-size-test
@@ -666,71 +666,71 @@ jobs:
666666 PYTHON_EXECUTABLE=python bash .ci/scripts/build-qnn-sdk.sh
667667 PYTHON_EXECUTABLE=python bash .ci/scripts/test_model.sh ${{ matrix.model }} "cmake" "qnn"
668668
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"
686686
687687
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
694694
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
700700
701- # bash .ci/scripts/setup-conda.sh
701+ bash .ci/scripts/setup-conda.sh
702702
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."
707707
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}"
709709
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
724724
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}"
727727
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
734734
735735 test-huggingface-transformers-xnnpack :
736736 # NB: Don't run this on fork PRs because they won't have access to the secret and would fail anyway
0 commit comments