[0-size Tensor No.80、307] Add 0-size Tensor support for imag API. (#72936)

DanielSun11 · web-flow · commit d38fcd9f7833 · 2025-05-28T11:07:42.000+08:00
* fix 0 size Tensor for expand kernel in onednn

* format

* expand_grad support 0 size Tensor

* fix bug and add  unittest

* modify the unittest for onednn kernel

* modify expand unittest to force enable onednn op

* add unittest case
diff --git a/test/legacy_test/test_real_imag_op.py b/test/legacy_test/test_real_imag_op.py
@@ -275,8 +275,26 @@ def setUp(self):
             self.places.append(paddle.CPUPlace())
         if paddle.is_compiled_with_cuda():
             self.places.append(paddle.CUDAPlace(0))
+        self.init_shape()
+
+    def init_shape(self):
         self._shape = [2, 20, 2, 3]
 
 
+class TestImagAPIZeroSize(TestImagAPI):
+    def init_shape(self):
+        self._shape = [2, 0, 2, 3]
+
+
+class TestImagAPIZeroSize1(TestImagAPI):
+    def init_shape(self):
+        self._shape = [2, 0, 0, 3]
+
+
+class TestImagAPIZeroSize2(TestImagAPI):
+    def init_shape(self):
+        self._shape = [0, 0, 0, 0]
+
+
 if __name__ == "__main__":
     unittest.main()
