diff --git a/tests/conftest.py b/tests/conftest.py index 57a33ad08c94..4fbb4132d385 100644 --- a/tests/conftest.py +++ b/tests/conftest.py @@ -257,7 +257,8 @@ def wrap_device(self, x: _T, device: Optional[str] = None) -> _T: 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()} diff --git a/tests/utils.py b/tests/utils.py index 5a97636eec64..ba490cc38450 100644 --- a/tests/utils.py +++ b/tests/utils.py @@ -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