Inv support 0-size (#72262)

aquagull · web-flow · commit 528d8c17b68a · 2025-04-17T10:48:32.000+08:00
* update

* add test

* fix

* empty
diff --git a/paddle/phi/infermeta/unary.cc b/paddle/phi/infermeta/unary.cc
@@ -2289,7 +2289,7 @@ void InverseInferMeta(const MetaTensor& x, MetaTensor* out) {
           input_dims));
   for (int i = 0; i < input_rank; ++i) {
     PADDLE_ENFORCE_EQ(
-        (input_dims[i] == -1) || (input_dims[i] > 0),
+        (input_dims[i] == -1) || (input_dims[i] >= 0),
         true,
         errors::InvalidArgument(
             "Each dimension of input tensor is expected to be -1 or a "
diff --git a/paddle/phi/kernels/impl/inverse_grad_kernel_impl.h b/paddle/phi/kernels/impl/inverse_grad_kernel_impl.h
@@ -31,7 +31,9 @@ void InverseGradKernel(const Context& dev_ctx,
                        DenseTensor* in_grad) {
   if (in_grad) {
     dev_ctx.template Alloc<T>(in_grad);
-
+    if (out_grad.numel() == 0) {
+      return;
+    }
     auto blas = phi::funcs::GetBlas<Context, T>(dev_ctx);
 
     DenseTensor tmp_out;
diff --git a/paddle/phi/kernels/impl/inverse_kernel_impl.h b/paddle/phi/kernels/impl/inverse_kernel_impl.h
@@ -28,7 +28,9 @@ void InverseKernel(const Context& dev_ctx,
                    const DenseTensor& x,
                    DenseTensor* out) {
   dev_ctx.template Alloc<T>(out);
-
+  if (x.numel() == 0) {
+    return;
+  }
   phi::funcs::MatrixInverseFunctor<Context, T> mat_inv;
   mat_inv(dev_ctx, x, out);
 }
diff --git a/test/legacy_test/test_inverse_op.py b/test/legacy_test/test_inverse_op.py
@@ -60,6 +60,20 @@ def config(self):
         self.python_api = paddle.tensor.math.inverse
 
 
+class TestInverseOpZeroSize(TestInverseOp):
+    def config(self):
+        self.matrix_shape = [0, 0]
+        self.dtype = "float64"
+        self.python_api = paddle.tensor.math.inverse
+
+
+class TestInverseOpBatchedZeroSize(TestInverseOp):
+    def config(self):
+        self.matrix_shape = [7, 0, 0]
+        self.dtype = "float64"
+        self.python_api = paddle.tensor.math.inverse
+
+
 class TestInverseOpLarge(TestInverseOp):
     def config(self):
         self.matrix_shape = [32, 32]
