Skip to content
GitLab
Menu
Projects
Groups
Snippets
Loading...
Help
Help
Support
Community forum
Keyboard shortcuts
?
Submit feedback
Contribute to GitLab
Sign in
Toggle navigation
Menu
Open sidebar
OpenDAS
vllm_cscc
Commits
79f2f1c2
Unverified
Commit
79f2f1c2
authored
Jun 20, 2025
by
Li, Jiang
Committed by
GitHub
Jun 20, 2025
Browse files
[CPU][CI] Fallback sliding window to v0 and fix CPU pooling model tests (#19901)
Signed-off-by:
jiang1.li
<
jiang1.li@intel.com
>
parent
2e3e3c86
Changes
2
Hide whitespace changes
Inline
Side-by-side
Showing
2 changed files
with
13 additions
and
1 deletion
+13
-1
tests/models/language/pooling/test_embedding.py
tests/models/language/pooling/test_embedding.py
+6
-1
vllm/engine/arg_utils.py
vllm/engine/arg_utils.py
+7
-0
No files found.
tests/models/language/pooling/test_embedding.py
View file @
79f2f1c2
# SPDX-License-Identifier: Apache-2.0
# SPDX-FileCopyrightText: Copyright contributors to the vLLM project
import
os
import
pytest
from
vllm.config
import
PoolerConfig
...
...
@@ -33,7 +35,7 @@ def v1(run_with_both_engines):
# To avoid this problem, for now we skip v0 since it will be
# deprecated anyway.
pytest
.
param
(
"ssmits/Qwen2-7B-Instruct-embed-base"
,
marks
=
[
pytest
.
mark
.
skip_v0
]),
marks
=
[
pytest
.
mark
.
skip_v0
,
pytest
.
mark
.
cpu_model
]),
# [Encoder-only]
pytest
.
param
(
"BAAI/bge-base-en-v1.5"
,
marks
=
[
...
...
@@ -58,6 +60,9 @@ def test_models(
model
,
monkeypatch
,
)
->
None
:
if
model
==
"intfloat/e5-mistral-7b-instruct"
and
current_platform
.
is_cpu
(
)
and
os
.
environ
.
get
(
"VLLM_USE_V1"
,
"0"
)
==
"1"
:
pytest
.
skip
(
"CPU V1 doesn't support sliding window"
)
if
model
==
"BAAI/bge-multilingual-gemma2"
and
current_platform
.
is_rocm
():
# ROCm Triton FA does not currently support sliding window attention
...
...
vllm/engine/arg_utils.py
View file @
79f2f1c2
...
...
@@ -1449,6 +1449,13 @@ class EngineArgs:
model_config
=
model_config
)
and
_warn_or_fallback
(
current_platform
.
device_name
):
return
False
if
(
current_platform
.
is_cpu
()
and
model_config
.
get_sliding_window
()
is
not
None
):
_raise_or_fallback
(
feature_name
=
"sliding window (CPU backend)"
,
recommend_to_remove
=
False
)
return
False
#############################################################
return
True
...
...
Write
Preview
Markdown
is supported
0%
Try again
or
attach a new file
.
Attach a file
Cancel
You are about to add
0
people
to the discussion. Proceed with caution.
Finish editing this message first!
Cancel
Please
register
or
sign in
to comment