Skip to content

Commit 8ca7a71

Browse files
OpenVINO: added CPU-like conditions (#14338)
Signed-off-by: Ilya Lavrenov <[email protected]>
1 parent 63137cd commit 8ca7a71

File tree

2 files changed

+3
-2
lines changed

2 files changed

+3
-2
lines changed

tests/conftest.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -257,7 +257,8 @@ def wrap_device(self, x: _T, device: Optional[str] = None) -> _T:
257257
return x
258258

259259
if device is None:
260-
device = "cpu" if current_platform.is_cpu() else "cuda"
260+
device = "cpu" if current_platform.is_cpu(
261+
) or current_platform.is_openvino() else "cuda"
261262

262263
if isinstance(x, dict):
263264
return {k: self.wrap_device(v, device) for k, v in x.items()}

tests/utils.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -698,7 +698,7 @@ def large_gpu_mark(min_gb: int) -> pytest.MarkDecorator:
698698
without enough resources, or called when filtering tests to run directly.
699699
"""
700700
try:
701-
if current_platform.is_cpu():
701+
if current_platform.is_cpu() or current_platform.is_openvino():
702702
memory_gb = 0
703703
else:
704704
memory_gb = current_platform.get_device_total_memory() / GB_bytes

0 commit comments

Comments
 (0)