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:
395
395
# Test llama2
396
396
PYTHON_EXECUTABLE=python bash .ci/scripts/test_llama.sh -model stories110M -build_tool "${BUILD_TOOL}" -mode "${MODE}" -dtype "${DTYPE}" -pt2e_quantize "${PT2E_QUANTIZE}"
397
397
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 : 90
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
+
398
417
test-phi-3-mini-runner-linux :
399
418
name : test-phi-3-mini-runner-linux
400
419
uses : pytorch/test-infra/.github/workflows/linux_job.yml@main
You can’t perform that action at this time.
0 commit comments