[0-size Tensor Job2 No.14] Add 0-size Tensor support for gammaincc (#73207)

co63oc · web-flow · commit 5d5fb53f9fbe · 2025-06-11T12:17:36.000+08:00
* Fix

* Fix
diff --git a/paddle/phi/kernels/impl/gammaincc_grad_kernel_impl.h b/paddle/phi/kernels/impl/gammaincc_grad_kernel_impl.h
@@ -49,6 +49,10 @@ void GammainccGradKernel(const Context& dev_ctx,
                          const DenseTensor& y,
                          const DenseTensor& d_out,
                          DenseTensor* d_y) {
+  if (d_y && d_y->numel() == 0) {
+    dev_ctx.template Alloc<T>(d_y);
+    return;
+  }
   auto numel = d_out.numel();
   auto* dout_data = d_out.data<T>();
   auto* x_data = x.data<T>();
diff --git a/paddle/phi/kernels/impl/gammaincc_kernel_impl.h b/paddle/phi/kernels/impl/gammaincc_kernel_impl.h
@@ -132,6 +132,10 @@ void GammainccKernel(const Context& dev_ctx,
                      const DenseTensor& x,
                      const DenseTensor& y,
                      DenseTensor* out) {
+  if (out && out->numel() == 0) {
+    dev_ctx.template Alloc<T>(out);
+    return;
+  }
   auto numel = x.numel();
   auto* x_data = x.data<T>();
   auto* y_data = y.data<T>();
diff --git a/test/legacy_test/test_gammaincc_op.py b/test/legacy_test/test_gammaincc_op.py
@@ -32,13 +32,16 @@ def setUp(self):
         self.op_type = 'gammaincc'
         self.python_api = paddle.gammaincc
         self.init_dtype_type()
-        self.shape = (3, 40)
+        self.init_shape()
         self.x = np.random.random(self.shape).astype(self.dtype) + 1
         self.y = np.random.random(self.shape).astype(self.dtype) + 1
         self.inputs = {'x': self.x, 'y': self.y}
         out = ref_gammaincc(self.x, self.y)
         self.outputs = {'out': out}
 
+    def init_shape(self):
+        self.shape = (3, 40)
+
     def init_dtype_type(self):
         self.dtype = np.float64
 
@@ -127,5 +130,17 @@ def init_dtype_type(self):
         self.dtype = "float32"
 
 
+class TestGammainccOp_ZeroSize(TestGammainccOp):
+
+    def init_shape(self):
+        self.shape = (0, 40)
+
+
+class TestGammainccOp_ZeroSize2(TestGammainccOp):
+
+    def init_shape(self):
+        self.shape = (0, 0)
+
+
 if __name__ == "__main__":
     unittest.main()
