xinyun/vllm
1
0
Fork 0
mirror of https://git.datalinker.icu/vllm-project/vllm.git synced 2025-12-09 22:25:32 +08:00
Code Issues Packages Projects Releases Wiki Activity

OpenVINO: added CPU-like conditions (#14338)

Browse Source 
Signed-off-by: Ilya Lavrenov <ilya.lavrenov@intel.com>
This commit is contained in:
Ilya Lavrenov 2025-03-07 10:24:49 +04:00 committed by GitHub
parent 63137cd922
commit 8ca7a71df7
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
2 changed files with 3 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
tests/conftest.py
View File

@ -257,7 +257,8 @@ class HfRunner:
return x
if device is None:
device = "cpu" if current_platform.is_cpu() else "cuda"
device = "cpu" if current_platform.is_cpu(
) or current_platform.is_openvino() else "cuda"
if isinstance(x, dict):
return {k: self.wrap_device(v, device) for k, v in x.items()}

2
tests/utils.py
View File

@ -698,7 +698,7 @@ def large_gpu_mark(min_gb: int) -> pytest.MarkDecorator:
without enough resources, or called when filtering tests to run directly.
"""
try:
if current_platform.is_cpu():
if current_platform.is_cpu() or current_platform.is_openvino():
memory_gb = 0
else:
memory_gb = current_platform.get_device_total_memory() / GB_bytes