Unverified Commit 8ca7a71d authored by Ilya Lavrenov's avatar Ilya Lavrenov Committed by GitHub
Browse files

OpenVINO: added CPU-like conditions (#14338)


Signed-off-by: default avatarIlya Lavrenov <ilya.lavrenov@intel.com>
parent 63137cd9
......@@ -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()}
......
......@@ -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
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment