Skip to content

Commit 93f6d84

Browse files
committed
Move HF LLM to nightly
1 parent 3062972 commit 93f6d84

File tree

2 files changed

+39
-5
lines changed

2 files changed

+39
-5
lines changed

.github/workflows/nightly.yml

Lines changed: 34 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -36,3 +36,37 @@ jobs:
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 }}

.github/workflows/pull.yml

Lines changed: 5 additions & 5 deletions
Original file line numberDiff line numberDiff 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]")

0 commit comments

Comments
 (0)