Skipping CUDA tests for CPU setup (#2191)

### Changes Add check `torch.cuda.is_available` to `TestTorchCudaFBCAlgorithm` test. ### Related tickets 122537
openvinotoolkit · Oct 13, 2023 · bd9e843 · bd9e843
1 parent 1100a94
commit bd9e843
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 2 deletions.
diff --git 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
@@ -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):