OpenVINO: added CPU-like conditions (#14338)

Signed-off-by: Ilya Lavrenov <[email protected]>
This commit is contained in:
Ilya Lavrenov
2025-03-06 22:24:49 -08:00
committed by GitHub
parent 63137cd922
commit 8ca7a71df7
2 changed files with 3 additions and 2 deletions
+2 -1
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()}
+1 -1
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