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: ...@@ -257,7 +257,8 @@ class HfRunner:
return x return x
if device is None: 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): if isinstance(x, dict):
return {k: self.wrap_device(v, device) for k, v in x.items()} 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: ...@@ -698,7 +698,7 @@ def large_gpu_mark(min_gb: int) -> pytest.MarkDecorator:
without enough resources, or called when filtering tests to run directly. without enough resources, or called when filtering tests to run directly.
""" """
try: try:
if current_platform.is_cpu(): if current_platform.is_cpu() or current_platform.is_openvino():
memory_gb = 0 memory_gb = 0
else: else:
memory_gb = current_platform.get_device_total_memory() / GB_bytes 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