Fix (#72837)

co63oc · web-flow · commit e5dc602af5a3 · 2025-05-22T09:45:26.000+08:00
diff --git a/paddle/phi/kernels/impl/angle_grad_kernel_impl.h b/paddle/phi/kernels/impl/angle_grad_kernel_impl.h
@@ -29,6 +29,10 @@ void AngleGradKernel(const Context& dev_ctx,
   auto* dout_data = out_grad.data<phi::dtype::Real<T>>();
   auto* x_data = x.data<T>();
   x_grad->Resize(out_grad.dims());
+  if (x_grad->numel() == 0) {
+    dev_ctx.template Alloc<T>(x_grad);
+    return;
+  }
   auto* dx_data = dev_ctx.template Alloc<T>(x_grad);
 
   phi::funcs::ForRange<Context> for_range(dev_ctx, numel);
diff --git a/paddle/phi/kernels/impl/angle_kernel_impl.h b/paddle/phi/kernels/impl/angle_kernel_impl.h
@@ -27,6 +27,10 @@ void AngleKernel(const Context& dev_ctx,
   auto numel = x.numel();
   auto* x_data = x.data<T>();
   out->Resize(x.dims());
+  if (out->numel() == 0) {
+    dev_ctx.template Alloc<phi::dtype::Real<T>>(out);
+    return;
+  }
   auto* out_data = dev_ctx.template Alloc<phi::dtype::Real<T>>(out);
 
   funcs::ForRange<Context> for_range(dev_ctx, numel);
diff --git a/test/legacy_test/test_angle_op.py b/test/legacy_test/test_angle_op.py
@@ -167,5 +167,17 @@ def test_static(self):
         np.testing.assert_allclose(self.out, out_np, rtol=1e-05)
 
 
+class TestZeroSize(unittest.TestCase):
+    def setUp(self):
+        self.x = np.random.randn(2, 0) + 1j * np.random.randn(2, 0)
+        self.out = np.angle(self.x)
+
+    def test_0size(self):
+        with dygraph.guard():
+            x = paddle.to_tensor(self.x)
+            out_np = paddle.angle(x).numpy()
+        np.testing.assert_allclose(self.out, out_np, rtol=1e-05)
+
+
 if __name__ == "__main__":
     unittest.main()
