Skip to content

Commit 59d5d2c

Browse files
authored
[CI/Build] Skip prompt embeddings tests on V1-only CPU backend (vllm-project#24721)
Signed-off-by: jiang1.li <[email protected]>
1 parent d21a36f commit 59d5d2c

File tree

1 file changed

+6
-0
lines changed

1 file changed

+6
-0
lines changed

tests/models/language/generation/test_common.py

Lines changed: 6 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -119,6 +119,12 @@ def test_models(hf_runner, vllm_runner, example_prompts, model: str,
119119
# in parts of the operators
120120
pytest.skip(f"Skipping '{model}' model test with AITER kernel.")
121121

122+
# Note: can be removed when
123+
# https://github.com/vllm-project/vllm/pull/24278 finished
124+
if current_platform.is_cpu() and use_prompt_embeds:
125+
pytest.skip("Skipping use_prompt_embeds=True with "
126+
"V1-only CPU backend.")
127+
122128
with hf_runner(model) as hf_model:
123129
hf_outputs = hf_model.generate_greedy_logprobs_limit(
124130
example_prompts, max_tokens, num_logprobs)

0 commit comments

Comments
 (0)