bugfix

open-mmlab · Dec 8, 2022 · 15ac0eb · 15ac0eb
1 parent 5477510
commit 15ac0eb
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/tests/test_ops/test_fused_bias_leakyrelu.py b/tests/test_ops/test_fused_bias_leakyrelu.py
@@ -22,7 +22,7 @@ def setup_class(cls):
             cls.input_tensor = torch.randn((2, 2, 2, 2),
                                            requires_grad=True).cuda()
             cls.bias = torch.zeros(2, requires_grad=True).cuda()
-        else:
+        elif IS_NPU_AVAILABLE:
             cls.input_tensor = torch.randn((2, 2, 2, 2),
                                            requires_grad=True).npu()
             cls.bias = torch.zeros(2, requires_grad=True).npu()
@@ -47,7 +47,7 @@ def test_gradient(self, device):
                     self.input_tensor,
                     delta=1e-4,
                     pt_atol=1e-3)
-            else:
+            elif IS_NPU_AVAILABLE:
                 gradcheck(
                     FusedBiasLeakyReLU(2).npu(),
                     self.input_tensor,
@@ -60,7 +60,7 @@ def test_gradient(self, device):
                     self.input_tensor,
                     eps=1e-4,
                     atol=1e-3)
-            else:
+            elif IS_NPU_AVAILABLE:
                 gradcheck(
                     FusedBiasLeakyReLU(2).npu(),
                     self.input_tensor,
@@ -86,7 +86,7 @@ def test_gradgradient(self, device):
                 self.input_tensor,
                 eps=1e-4,
                 atol=1e-3)
-        else:
+        elif IS_NPU_AVAILABLE:
             gradcheck(
                 FusedBiasLeakyReLU(2).npu(),
                 self.input_tensor,