[0-size Tensor Job2 No.25] Add 0-size Tensor support for increment (#73382)

co63oc · web-flow · commit 3c69eb3a035a · 2025-06-18T16:34:18.000+08:00
diff --git a/paddle/phi/infermeta/unary.cc b/paddle/phi/infermeta/unary.cc
@@ -2164,12 +2164,14 @@ void IdentityLossInferMeta(const MetaTensor& x,
 }
 
 void IncrementInferMeta(const MetaTensor& x, float value, MetaTensor* out) {
-  PADDLE_ENFORCE_EQ(
-      product(x.dims()),
-      1UL,
-      errors::InvalidArgument("The number of elements in Input(X) should be 1."
-                              "Now the number is %d.",
-                              product(x.dims())));
+  if (x.numel() != 0) {
+    PADDLE_ENFORCE_EQ(product(x.dims()),
+                      1UL,
+                      errors::InvalidArgument(
+                          "The number of elements in Input(X) should be 1."
+                          "Now the number is %d.",
+                          product(x.dims())));
+  }
   out->set_dims(x.dims());
   out->share_lod(x);
   out->set_layout(x.layout());
diff --git a/paddle/phi/kernels/impl/increment_kernel_impl.h b/paddle/phi/kernels/impl/increment_kernel_impl.h
@@ -26,6 +26,9 @@ void IncrementKernel(const Context& dev_ctx,
                      float value,
                      DenseTensor* out) {
   dev_ctx.template Alloc<T>(out);
+  if (x.numel() == 0) {
+    return;
+  }
   auto& dev = *dev_ctx.eigen_device();
   funcs::EigenAdd<std::decay_t<decltype(dev)>, T>::Eval(
       dev,
diff --git a/paddle/phi/kernels/xpu/increment_kernel.cc b/paddle/phi/kernels/xpu/increment_kernel.cc
@@ -25,6 +25,10 @@ void IncrementKernel(const Context& dev_ctx,
                      const DenseTensor& x,
                      float value,
                      DenseTensor* out) {
+  if (x.numel() == 0) {
+    dev_ctx.template Alloc<T>(out);
+    return;
+  }
   // check input
   PADDLE_ENFORCE_EQ(x.numel(),
                     1,
diff --git a/test/legacy_test/test_increment.py b/test/legacy_test/test_increment.py
@@ -76,5 +76,15 @@ def test_increment(self):
             self.assertEqual(a[0], 1)
 
 
+class TestIncrement_ZeroSize(unittest.TestCase):
+
+    def test_api(self):
+        with base.dygraph.guard():
+            input = paddle.randn(shape=[0]).astype('int64')
+            expected_result = np.random.random([0]).astype('int64')
+            output = paddle.tensor.math.increment(input, value=1)
+            self.assertEqual((output.numpy() == expected_result).all(), True)
+
+
 if __name__ == "__main__":
     unittest.main()
