@@ -463,7 +463,7 @@ jobs:
463463 strategy :
464464 matrix :
465465 dtype : [fp32]
466- model : [dl3, mv3, mv2, ic4, ic3, vit, mb, w2l]
466+ model : [dl3, mv3, mv2, ic4, ic3, vit, mb, w2l, cvt, deit, dit, efficientnet, focalnet, mobilevit_v1, mobilevit_v2, pvt, swin, albert, bert, distilbert, roberta ]
467467 fail-fast : false
468468 with :
469469 runner : linux.2xlarge
@@ -480,32 +480,6 @@ jobs:
480480 PYTHON_EXECUTABLE=python bash .ci/scripts/build-qnn-sdk.sh
481481 PYTHON_EXECUTABLE=python bash .ci/scripts/test_model.sh ${{ matrix.model }} "cmake" "qnn"
482482
483- test-qnn-optimum-model :
484- name : test-qnn-optimum-model
485- uses : pytorch/test-infra/.github/workflows/linux_job_v2.yml@main
486- permissions :
487- id-token : write
488- contents : read
489- strategy :
490- matrix :
491- dtype : [fp32]
492- model : [cvt, deit, dit, efficientnet, focalnet, mobilevit_v1, mobilevit_v2, pvt, swin, albert, bert, distilbert, roberta] # eurobert requires transfomer >= 4.48.0, skip for now
493- fail-fast : false
494- with :
495- runner : linux.2xlarge.memory
496- docker-image : executorch-ubuntu-22.04-qnn-sdk
497- submodules : ' recursive'
498- ref : ${{ github.event_name == 'pull_request' && github.event.pull_request.head.sha || github.sha }}
499- timeout : 90
500- script : |
501- # The generic Linux job chooses to use base env, not the one setup by the image
502- CONDA_ENV=$(conda env list --json | jq -r ".envs | .[-1]")
503- conda activate "${CONDA_ENV}"
504- PYTHON_EXECUTABLE=python bash .ci/scripts/setup-linux.sh --build-tool cmake
505- PYTHON_EXECUTABLE=python bash .ci/scripts/setup-qnn-deps.sh
506- PYTHON_EXECUTABLE=python bash .ci/scripts/build-qnn-sdk.sh
507- PYTHON_EXECUTABLE=python bash .ci/scripts/test_model.sh ${{ matrix.model }} "cmake" "qnn"
508-
509483 test-apple-model :
510484 name : test-apple-model
511485 uses : pytorch/test-infra/.github/workflows/macos_job.yml@main
0 commit comments