fix

PaddlePaddle · Nov 8, 2023 · 54fde6d · 54fde6d
1 parent 4c14a36
commit 54fde6d
Showing 1 changed file with 17 additions and 12 deletions.
diff --git a/test/legacy_test/test_activation_op.py b/test/legacy_test/test_activation_op.py
@@ -1523,21 +1523,26 @@ def test_check_grad(self):
         if self.dtype == np.float16:
             return
         if self.dtype not in [np.complex64, np.complex128]:
-          self.check_grad(
-              ['X'],
-              'Out',
-              check_prim=True,
-              check_pir=True,
-              check_prim_pir=True,
-          )
+            self.check_grad(
+                ['X'],
+                'Out',
+                check_prim=True,
+                check_pir=True,
+                check_prim_pir=True,
+            )
         else:
-          self.check_grad(
-              ['X'],
-              'Out',
-          )   
+            self.check_grad(
+                ['X'],
+                'Out',
+            )
 
     def test_check_output(self):
-        self.check_output(check_prim=True, check_pir=True, check_prim_pir=True)
+        if self.dtype not in [np.complex64, np.complex128]:
+            self.check_output(
+                check_prim=True, check_pir=True, check_prim_pir=True
+            )
+        else:
+            self.check_output()
 
 
 class TestSqrtComplex64(TestSqrt):