Skip to content

Commit 7023d86

Browse files
committed
Update on "[llm] Add CMake for llama runner"
Summary: As titled. Also add a CI job. Will run a model in next PR. Test Plan: Reviewers: Subscribers: Tasks: Tags: [ghstack-poisoned]
1 parent ac6b874 commit 7023d86

File tree

1 file changed

+24
-0
lines changed

1 file changed

+24
-0
lines changed

.github/workflows/pull.yml

Lines changed: 24 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -213,3 +213,27 @@ jobs:
213213
uses: ./.github/workflows/_unittest.yml
214214
with:
215215
docker-image: executorch-ubuntu-22.04-clang12
216+
217+
test-llama-runner-cmake:
218+
name: test-llama-runner
219+
uses: pytorch/test-infra/.github/workflows/linux_job.yml@main
220+
strategy:
221+
matrix:
222+
include:
223+
- build-tool: cmake
224+
fail-fast: false
225+
with:
226+
runner: linux.2xlarge
227+
docker-image: executorch-ubuntu-22.04-clang12
228+
submodules: 'true'
229+
ref: ${{ github.event_name == 'pull_request' && github.event.pull_request.head.sha || github.sha }}
230+
timeout: 90
231+
script: |
232+
# The generic Linux job chooses to use base env, not the one setup by the image
233+
CONDA_ENV=$(conda env list --json | jq -r ".envs | .[-1]")
234+
conda activate "${CONDA_ENV}"
235+
236+
BUILD_TOOL=${{ matrix.build-tool }}
237+
PYTHON_EXECUTABLE=python bash .ci/scripts/setup-linux.sh "${BUILD_TOOL}"
238+
# Test selective build
239+
PYTHON_EXECUTABLE=python bash examples/models/llama2/test_llama_runner.sh "${BUILD_TOOL}"

0 commit comments

Comments
 (0)