From b1536d25540ed1089f8ed4ca652ca38a7927cee5 Mon Sep 17 00:00:00 2001 From: Alexander Dokuchaev Date: Wed, 11 Oct 2023 22:10:03 +0300 Subject: [PATCH] Skipping CUDA tests for CPU setup --- tests/torch/ptq/test_fast_bias_correction.py | 5 +++++ 1 file changed, 5 insertions(+) diff --git a/tests/torch/ptq/test_fast_bias_correction.py b/tests/torch/ptq/test_fast_bias_correction.py index 7f5639aaeba..7d4779044a1 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 @@ -64,10 +65,14 @@ def check_bias(model: NNCFNetwork, ref_bias: list): class TestTorchCudaFBCAlgorithm(TestTorchFBCAlgorithm): @staticmethod def list_to_backend_type(data: List) -> torch.Tensor: + if not torch.cuda.is_available(): + pytest.skip("Skipping for CPU-only setups") return torch.Tensor(data).cuda() @staticmethod def backend_specific_model(model: bool, tmp_dir: str): + if not torch.cuda.is_available(): + pytest.skip("Skipping for CPU-only setups") return get_nncf_network(model.cuda(), model.INPUT_SIZE) @staticmethod