Skip to content

Commit

Permalink
gha: test multiple vLLM versions
Browse files Browse the repository at this point in the history
  • Loading branch information
dtrifiro committed Jul 2, 2024
1 parent 6b136ae commit f9b46cf
Showing 1 changed file with 17 additions and 0 deletions.
17 changes: 17 additions & 0 deletions .github/workflows/tests.yaml
Original file line number Diff line number Diff line change
Expand Up @@ -11,6 +11,11 @@ on:

env:
FORCE_COLOR: "1"
# facilitate testing by building vLLM for CPU when needed
VLLM_CPU_DISABLE_AVX512: "true"
VLLM_TARGET_DEVICE: "cpu"
# prefer torch cpu version
PIP_EXTRA_INDEX_URL: "https://download.pytorch.org/whl/cpu"

concurrency:
group: ${{ github.workflow }}-${{ github.head_ref || github.run_id }}
Expand All @@ -25,6 +30,10 @@ jobs:
matrix:
os: [ubuntu-latest]
pyv: ["3.11"]
vllm_version:
- "pypi"
- "git+https://github.com/vllm-project/vllm"
- "git+https://github.com/opendatahub-io/vllm"

steps:
- name: Check out the repository
Expand All @@ -50,6 +59,14 @@ jobs:
pip --version
nox --version
- name: Set custom vllm version
if: ${{ matrix.vllm_version != 'pypi' }}
run: |
vllm_version="vllm@${{matrix.vllm_version}}"
echo "Using vllm@${vllm_version}"
sed -i "s|\"vllm.*\",|\"${vllm_version}\",|g" pyproject.toml
- name: Lint code and check dependencies
run: nox -s lint-${{ matrix.pyv }}

Expand Down

0 comments on commit f9b46cf

Please sign in to comment.