Merge 1adc4c8 into d31b221

open-mmlab · Mar 19, 2023 · cffb4e4 · cffb4e4
2 parents d31b221 + 1adc4c8
commit cffb4e4
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/tests/test_ops/test_filtered_lrelu.py b/tests/test_ops/test_filtered_lrelu.py
@@ -1,6 +1,7 @@
 # Copyright (c) OpenMMLab. All rights reserved.
 import pytest
 import torch
+from mmengine.utils import digit_version
 
 from mmcv.ops import filtered_lrelu
 
@@ -113,7 +114,10 @@ def test_filtered_lrelu_cpu(self):
             self.input_tensor, bias=self.bias, flip_filter=True)
         assert out.shape == (1, 3, 16, 16)
 
-    @pytest.mark.skipif(not torch.cuda.is_available(), reason='requires cuda')
+    @pytest.mark.skipif(
+        not torch.cuda.is_available()
+        or digit_version(torch.version.cuda) < digit_version('10.2'),
+        reason='requires cuda>=10.2')
     def test_filtered_lrelu_cuda(self):
         out = filtered_lrelu(self.input_tensor.cuda(), bias=self.bias.cuda())
         assert out.shape == (1, 3, 16, 16)