File tree Expand file tree Collapse file tree 2 files changed +39
-5
lines changed Expand file tree Collapse file tree 2 files changed +39
-5
lines changed Original file line number Diff line number Diff line change 3636 uses : ./.github/workflows/_link_check.yml
3737 with :
3838 ref : ${{ github.sha }}
39+
40+ test-static-hf-llm-qnn-linux :
41+ name : test-static-hf-llm-qnn-linux
42+ uses : pytorch/test-infra/.github/workflows/linux_job_v2.yml@main
43+ permissions :
44+ id-token : write
45+ contents : read
46+ strategy :
47+ matrix :
48+ task : [smollm2_135m]
49+ fail-fast : false
50+ with :
51+ runner : linux.24xlarge
52+ docker-image : ci-image:executorch-ubuntu-22.04-qnn-sdk
53+ submodules : ' recursive'
54+ ref : ${{ github.event_name == 'pull_request' && github.event.pull_request.head.sha || github.sha }}
55+ timeout : 900
56+ script : |
57+ # The generic Linux job chooses to use base env, not the one setup by the image
58+ CONDA_ENV=$(conda env list --json | jq -r ".envs | .[-1]")
59+ conda activate "${CONDA_ENV}"
60+
61+ BUILD_TOOL="cmake"
62+
63+ PYTHON_EXECUTABLE=python bash .ci/scripts/setup-qnn-deps.sh
64+ PYTHON_EXECUTABLE=python bash .ci/scripts/build-qnn-sdk.sh
65+
66+ # Setup executorch
67+ PYTHON_EXECUTABLE=python bash .ci/scripts/setup-linux.sh --build-tool "${BUILD_TOOL}"
68+
69+ # Setup install_requirements for llama
70+ PYTHON_EXECUTABLE=python bash examples/models/llama/install_requirements.sh
71+
72+ PYTHON_EXECUTABLE=python bash .ci/scripts/test_qnn_static_llm.sh ${{ matrix.task }}
Original file line number Diff line number Diff line change @@ -559,22 +559,22 @@ jobs:
559559 # Test llama2
560560 PYTHON_EXECUTABLE=python bash .ci/scripts/test_llama.sh -model stories110M -build_tool "${BUILD_TOOL}" -mode "${MODE}" -dtype "${DTYPE}" -pt2e_quantize "${PT2E_QUANTIZE}"
561561
562- test-static-llm -qnn-linux :
563- name : test-static-llm -qnn-linux
562+ test-static-llama -qnn-linux :
563+ name : test-static-llama -qnn-linux
564564 uses : pytorch/test-infra/.github/workflows/linux_job_v2.yml@main
565565 permissions :
566566 id-token : write
567567 contents : read
568568 strategy :
569569 matrix :
570- task : [stories_110m, stories_260k_bc, smollm2_135m ]
570+ task : [stories_110m, stories_260k_bc]
571571 fail-fast : false
572572 with :
573- runner : linux.24xlarge
573+ runner : linux.2xlarge
574574 docker-image : ci-image:executorch-ubuntu-22.04-qnn-sdk
575575 submodules : ' recursive'
576576 ref : ${{ github.event_name == 'pull_request' && github.event.pull_request.head.sha || github.sha }}
577- timeout : 900
577+ timeout : 180
578578 script : |
579579 # The generic Linux job chooses to use base env, not the one setup by the image
580580 CONDA_ENV=$(conda env list --json | jq -r ".envs | .[-1]")
You can’t perform that action at this time.
0 commit comments