Skip to content

Commit

Permalink
[Torch] Skip cuda test if cuda is not available (#2242)
Browse files Browse the repository at this point in the history
### Changes

Skip cuda test if cuda is not available

### Reason for changes

To fix CPU pre-commit

### Tests
precommit_torch_cpu/169/ is finished successfully
  • Loading branch information
daniil-lyakhov authored Nov 6, 2023
1 parent 94d1f9c commit 18c4471
Showing 1 changed file with 4 additions and 0 deletions.
4 changes: 4 additions & 0 deletions tests/torch/ptq/test_reducers_and_aggregators.py
Original file line number Diff line number Diff line change
Expand Up @@ -106,6 +106,8 @@ def all_close(self, val: torch.Tensor, ref) -> bool:
@pytest.mark.parametrize("device", ["cuda", "cpu"])
@pytest.mark.parametrize("size,ref", [(16_000_000, 1_600_000.8750), (17_000_000, 1_700_000.7500)])
def test_quantile_percentile_function(device, size, ref):
if not torch.cuda.is_available() and device == "cuda":
pytest.skip("Cuda is not available in current environment")
tensor = PTNNCFTensor(torch.arange(1, size, 1).float().to(device))
res_quantile = PTNNCFCollectorTensorProcessor.quantile(tensor, [0.1], axis=0)
res_percentile = PTNNCFCollectorTensorProcessor.percentile(tensor, [10], axis=0)
Expand All @@ -118,6 +120,8 @@ def test_quantile_percentile_function(device, size, ref):
@pytest.mark.parametrize("device", ["cuda", "cpu"])
@pytest.mark.parametrize("size,ref", [(16_000_000, 8_000_000), (17_000_000, 8_500_000)])
def test_median_function(device, size, ref):
if not torch.cuda.is_available() and device == "cuda":
pytest.skip("Cuda is not available in current environment")
tensor = PTNNCFTensor(torch.arange(1, size, 1).float().to(device))
res = PTNNCFCollectorTensorProcessor.median(tensor, axis=0)
assert res.tensor == ref
Expand Down

0 comments on commit 18c4471

Please sign in to comment.