File tree Expand file tree Collapse file tree 1 file changed +10
-7
lines changed Expand file tree Collapse file tree 1 file changed +10
-7
lines changed Original file line number Diff line number Diff line change 9
9
pre-commit :
10
10
runs-on : ubuntu-latest
11
11
steps :
12
- - uses : actions/checkout@v4
13
- - uses : actions/setup-python@v5
12
+ - uses : actions/checkout@v4 # Keep this as v4
13
+ - uses : actions/setup-python@v5 # Keep this as v5
14
14
with :
15
15
python-version : " 3.12"
16
16
- run : echo "::add-matcher::.github/workflows/matchers/actionlint.json"
17
17
- run : echo "::add-matcher::.github/workflows/matchers/mypy.json"
18
- - name : Set VLLM_TARGET_DEVICE to CPU for pre-commit checks
19
- run : echo "VLLM_TARGET_DEVICE=cpu" >> $GITHUB_ENV
18
+
19
+ - name : Set VLLM_TARGET_DEVICE and VLLM_CMAKE_ARGS for CPU build
20
+ run : |
21
+ echo "VLLM_TARGET_DEVICE=cpu" >> $GITHUB_ENV
22
+ # Crucial: Tell CMake to explicitly disable CUDA usage.
23
+ # This should override PyTorch/Caffe2's internal checks for CUDA libraries.
24
+ echo "VLLM_CMAKE_ARGS=-DUSE_CUDA=OFF" >> $GITHUB_ENV
20
25
- name : Install PyTorch for CPU
21
26
run : pip install torch==2.6.0 --index-url https://download.pytorch.org/whl/cpu
22
27
- name : Install vLLM Build Dependencies (excluding torch and comments)
23
- # This command filters out lines with 'torch==' AND lines starting with '#'
24
28
run : pip install $(grep -v 'torch==' requirements/build.txt | grep -v '^#' | tr '\n' ' ')
25
29
- name : Install vLLM Project and Dev Dependencies
26
30
run : pip install -e ".[dev]"
27
-
28
- - uses : pre-commit/action@v3.0.1
31
+ - uses : pre-commit/action@v3.0.1 # Keep this as v3.0.1
29
32
with :
30
33
extra_args : --all-files --hook-stage manual
You can’t perform that action at this time.
0 commit comments