File tree Expand file tree Collapse file tree 1 file changed +19
-0
lines changed Expand file tree Collapse file tree 1 file changed +19
-0
lines changed Original file line number Diff line number Diff line change @@ -395,6 +395,25 @@ jobs:
395395 # Test llama2
396396 PYTHON_EXECUTABLE=python bash .ci/scripts/test_llama.sh -model stories110M -build_tool "${BUILD_TOOL}" -mode "${MODE}" -dtype "${DTYPE}" -pt2e_quantize "${PT2E_QUANTIZE}"
397397
398+ test-qnn-models-linux :
399+ name : test-qnn-models-linux
400+ uses : pytorch/test-infra/.github/workflows/linux_job.yml@main
401+ strategy :
402+ fail-fast : false
403+ with :
404+ runner : linux.2xlarge
405+ docker-image : executorch-ubuntu-22.04-qnn-sdk
406+ submodules : ' true'
407+ ref : ${{ github.event_name == 'pull_request' && github.event.pull_request.head.sha || github.sha }}
408+ timeout : 180
409+ script : |
410+ # The generic Linux job chooses to use base env, not the one setup by the image
411+ CONDA_ENV=$(conda env list --json | jq -r ".envs | .[-1]")
412+ conda activate "${CONDA_ENV}"
413+
414+ # placeholder for running test_qnn_delegate.py, can use matrix such that we can trigger different jobs, refers to test-llama-runner-qnn-linux
415+ # reminder: make sure each job runs fast
416+
398417 test-phi-3-mini-runner-linux :
399418 name : test-phi-3-mini-runner-linux
400419 uses : pytorch/test-infra/.github/workflows/linux_job.yml@main
You can’t perform that action at this time.
0 commit comments