[0-size Tensor No.20、22、54、56、242、301、186、212、178、226] Add 0-size Tensor support for polygamma/bitwise_not (#72977)

co63oc · web-flow · commit e9ad101d969e · 2025-06-03T16:37:14.000+08:00
diff --git a/test/legacy_test/test_activation_op_zero_size.py b/test/legacy_test/test_activation_op_zero_size.py
@@ -25,10 +25,15 @@
     TestCeil,
     TestCos,
     TestCosh,
+    TestExpFp32_Prim,
+    TestExpm1,
     TestFloor,
+    TestHardSwish,
+    TestLeakyRelu,
     TestLogSigmoid,
     TestReciprocal,
     TestRelu,
+    TestRelu6,
     TestRsqrt,
     TestSigmoid,
     TestSilu,
@@ -37,6 +42,7 @@
     TestSoftsign,
     TestSqrt,
     TestSquare,
+    TestSwish,
     TestTan,
     TestTanh,
     TestTanhshrink,
@@ -96,7 +102,12 @@ def test_check_grad(self):
 create_test_zero_size_class(TestLogSigmoid)
 create_test_zero_size_class(TestFloor)
 create_test_zero_size_class(TestCeil)
-
+create_test_zero_size_class(TestExpFp32_Prim)
+create_test_zero_size_class(TestExpm1)
+create_test_zero_size_class(TestLeakyRelu)
+create_test_zero_size_class(TestRelu6)
+create_test_zero_size_class(TestHardSwish)
+create_test_zero_size_class(TestSwish)
 
 if __name__ == "__main__":
     unittest.main()
diff --git a/test/legacy_test/test_bitwise_op.py b/test/legacy_test/test_bitwise_op.py
@@ -392,6 +392,11 @@ def init_shape(self):
         self.x_shape = []
 
 
+class TestBitwiseNot_ZeroSize(TestBitwiseNot):
+    def init_shape(self):
+        self.x_shape = [0, 3, 4, 5]
+
+
 class TestBitwiseNotUInt8(TestBitwiseNot):
     def init_dtype(self):
         self.dtype = np.uint8
diff --git a/test/legacy_test/test_polygamma_op.py b/test/legacy_test/test_polygamma_op.py
@@ -217,5 +217,25 @@ def test_check_grad(self):
         )
 
 
+class TestPolygammaOp_ZeroSize(TestPolygammaOp):
+
+    def init_config(self):
+        self.dtype = np.float64
+        self.order = 1
+        rand_case = np.random.randn(0).astype(self.dtype)
+        int_case = np.random.randint(low=1, high=100, size=0).astype(self.dtype)
+        self.case = np.concatenate([rand_case, int_case])
+        self.inputs = {'x': self.case}
+        self.attrs = {'n': self.order}
+        self.target = ref_polygamma(self.inputs['x'], self.order)
+
+    def test_check_grad(self):
+        self.check_grad(
+            ['x'],
+            'out',
+            check_pir=True,
+        )
+
+
 if __name__ == "__main__":
     unittest.main()
