Skip to content

Commit b818062

Browse files
cccclaihuydhn
andauthored
add qnn test template job (#7636)
* add qnn test template job (#7636) Summary: Pull Request resolved: #7636 Differential Revision: D68112936 * Increase timeout --------- Co-authored-by: Huy Do <[email protected]>
1 parent 745f17e commit b818062

File tree

1 file changed

+19
-0
lines changed

1 file changed

+19
-0
lines changed

.github/workflows/pull.yml

Lines changed: 19 additions & 0 deletions
Original file line numberDiff line numberDiff 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

0 commit comments

Comments
 (0)