diff --git a/tests/torch/ptq/test_fast_bias_correction.py b/tests/torch/ptq/test_fast_bias_correction.py index 7f5639aaeba..e5d3b937376 100644 --- a/tests/torch/ptq/test_fast_bias_correction.py +++ b/tests/torch/ptq/test_fast_bias_correction.py @@ -11,6 +11,7 @@ from typing import List +import pytest import torch from nncf.common.factory import NNCFGraphFactory @@ -61,6 +62,7 @@ def check_bias(model: NNCFNetwork, ref_bias: list): raise ValueError("Not found node with bias") +@pytest.mark.skipif(not torch.cuda.is_available(), reason="Skipping for CPU-only setups") class TestTorchCudaFBCAlgorithm(TestTorchFBCAlgorithm): @staticmethod def list_to_backend_type(data: List) -> torch.Tensor: diff --git a/tests/torch/test_tensor.py b/tests/torch/test_tensor.py index eb4d907022b..35402c44891 100644 --- a/tests/torch/test_tensor.py +++ b/tests/torch/test_tensor.py @@ -22,11 +22,10 @@ def to_tensor(x): return torch.tensor(x) +@pytest.mark.skipif(not torch.cuda.is_available(), reason="Skipping for CPU-only setups") class TestCudaPTNNCFTensorOperators(TemplateTestNNCFTensorOperators): @staticmethod def to_tensor(x): - if not torch.cuda.is_available(): - pytest.skip("Skipping for CPU-only setups") return torch.tensor(x).cuda() def test_device(self):