Skip to content

Commit 851c23e

Browse files
russellbshreyankg
authored andcommitted
[CI/Build] Use spawn multiprocessing mode for V1 test pipeline (vllm-project#14243)
Signed-off-by: Russell Bryant <[email protected]>
1 parent 221d1f1 commit 851c23e

File tree

1 file changed

+3
-0
lines changed

1 file changed

+3
-0
lines changed

.buildkite/test-pipeline.yaml

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -78,6 +78,7 @@ steps:
7878
- tests/basic_correctness/test_preemption
7979
- tests/basic_correctness/test_cumem.py
8080
commands:
81+
- export VLLM_WORKER_MULTIPROC_METHOD=spawn
8182
- pytest -v -s basic_correctness/test_cumem.py
8283
- pytest -v -s basic_correctness/test_basic_correctness.py
8384
- pytest -v -s basic_correctness/test_cpu_offload.py
@@ -112,6 +113,7 @@ steps:
112113
- tests/entrypoints/test_chat_utils
113114
- tests/entrypoints/offline_mode
114115
commands:
116+
- export VLLM_WORKER_MULTIPROC_METHOD=spawn
115117
- pytest -v -s entrypoints/llm --ignore=entrypoints/llm/test_lazy_outlines.py --ignore=entrypoints/llm/test_generate.py --ignore=entrypoints/llm/test_generate_multiple_loras.py --ignore=entrypoints/llm/test_guided_generate.py --ignore=entrypoints/llm/test_collective_rpc.py
116118
- pytest -v -s entrypoints/llm/test_lazy_outlines.py # it needs a clean process
117119
- pytest -v -s entrypoints/llm/test_generate.py # it needs a clean process
@@ -196,6 +198,7 @@ steps:
196198
- tests/v1
197199
commands:
198200
# split the test to avoid interference
201+
- export VLLM_WORKER_MULTIPROC_METHOD=spawn
199202
- VLLM_USE_V1=1 pytest -v -s v1/core
200203
- VLLM_USE_V1=1 pytest -v -s v1/engine
201204
- VLLM_USE_V1=1 pytest -v -s v1/sample

0 commit comments

Comments
 (0)