[0-size Tensor No.5] Add 0-size Tensor support for allclose (#72820)

co63oc · web-flow · commit 35d04c02ec2f · 2025-05-29T20:40:52.000+08:00
* Fix

* Fix

* Fix

* Fix

* Fix

* Fix

* Fix

* Fix
diff --git a/paddle/fluid/pir/dialect/operator/interface/infer_symbolic_shape/binary_infer_sym.cc b/paddle/fluid/pir/dialect/operator/interface/infer_symbolic_shape/binary_infer_sym.cc
@@ -78,15 +78,33 @@ bool AllcloseOpInferSymbolicShape(
       infer_context->GetShapeOrDataForValue(op->operand_source(0)).shape();
   const auto y_shape =
       infer_context->GetShapeOrDataForValue(op->operand_source(1)).shape();
-  PADDLE_ENFORCE_EQ(x_shape.size(),
-                    y_shape.size(),
-                    common::errors::PreconditionNotMet(
-                        "Input(X) and Input(Y) must have the same "
-                        "dimension size. but got %d vs %d",
-                        x_shape.size(),
-                        y_shape.size()));
+
+  bool size_0 = false;
   for (size_t i = 0; i < x_shape.size(); ++i) {
-    infer_context->AddEqualCstr(x_shape[i], y_shape[i]);
+    if (x_shape[i] == 0) {
+      size_0 = true;
+      break;
+    }
+  }
+  if (!size_0) {
+    for (size_t i = 0; i < y_shape.size(); ++i) {
+      if (y_shape[i] == 0) {
+        size_0 = true;
+        break;
+      }
+    }
+  }
+  if (!size_0) {
+    PADDLE_ENFORCE_EQ(x_shape.size(),
+                      y_shape.size(),
+                      common::errors::PreconditionNotMet(
+                          "Input(X) and Input(Y) must have the same "
+                          "dimension size. but got %d vs %d",
+                          x_shape.size(),
+                          y_shape.size()));
+    for (size_t i = 0; i < x_shape.size(); ++i) {
+      infer_context->AddEqualCstr(x_shape[i], y_shape[i]);
+    }
   }
 
   infer_context->SetShapeOrDataForValue(
diff --git a/paddle/phi/infermeta/binary.cc b/paddle/phi/infermeta/binary.cc
@@ -96,7 +96,9 @@ void AllValueCompareInferMeta(const MetaTensor& x,
                               const MetaTensor& y,
                               MetaTensor* out,
                               MetaConfig config) {
-  detail::BinarySameInputDimsCheck(x, y, config);
+  if (x.numel() != 0 && y.numel() != 0) {
+    detail::BinarySameInputDimsCheck(x, y, config);
+  }
   out->set_dims(common::make_ddim({}));
   out->set_dtype(DataType::BOOL);
 }
diff --git a/paddle/phi/kernels/cpu/allclose_kernel.cc b/paddle/phi/kernels/cpu/allclose_kernel.cc
@@ -31,6 +31,11 @@ void AllCloseKernel(const Context& dev_ctx,
                     const Scalar& atol,
                     bool equal_nan,
                     DenseTensor* out) {
+  if (x.numel() == 0 || y.numel() == 0) {
+    auto* out_data = dev_ctx.template Alloc<bool>(out);
+    *out_data = true;
+    return;
+  }
   double rtol_v = NAN, atol_v = NAN;
   if (rtol.dtype() == DataType::FLOAT64) {
     rtol_v = rtol.to<double>();
diff --git a/paddle/phi/kernels/gpu/allclose_kernel.cu b/paddle/phi/kernels/gpu/allclose_kernel.cu
@@ -21,7 +21,7 @@
 #include "paddle/phi/common/data_type.h"
 #include "paddle/phi/core/enforce.h"
 #include "paddle/phi/core/kernel_registry.h"
-
+#include "paddle/phi/kernels/full_kernel.h"
 namespace phi {
 
 template <typename T>
@@ -66,6 +66,11 @@ void AllCloseKernel(const Context& dev_ctx,
                     const Scalar& atol,
                     bool equal_nan,
                     DenseTensor* out) {
+  if (x.numel() == 0 || y.numel() == 0) {
+    phi::Full<bool, Context>(
+        dev_ctx, phi::IntArray(common::vectorize(out->dims())), true, out);
+    return;
+  }
   double rtol_v, atol_v;
   if (rtol.dtype() == DataType::FLOAT64) {
     rtol_v = rtol.to<double>();
diff --git a/test/legacy_test/test_allclose_op.py b/test/legacy_test/test_allclose_op.py
@@ -463,5 +463,40 @@ def set_args(self):
         self.equal_nan = False
 
 
+class TestAllcloseOp_ZeroSize(OpTest):
+    def set_args(self):
+        self.input = np.random.random((2, 0)).astype("float32")
+        self.other = np.random.random((2, 0)).astype("float32")
+        self.rtol = np.array([1e-05]).astype("float64")
+        self.atol = np.array([1e-08]).astype("float64")
+        self.equal_nan = False
+
+    def setUp(self):
+        self.set_args()
+        self.op_type = "allclose"
+        self.python_api = paddle.allclose
+        self.inputs = {
+            'Input': self.input,
+            'Other': self.other,
+            "Rtol": self.rtol,
+            "Atol": self.atol,
+        }
+        self.attrs = {'equal_nan': self.equal_nan}
+        self.outputs = {
+            'Out': np.array(
+                np.allclose(
+                    self.inputs['Input'],
+                    self.inputs['Other'],
+                    rtol=self.rtol,
+                    atol=self.atol,
+                    equal_nan=self.equal_nan,
+                )
+            )
+        }
+
+    def test_check_output(self):
+        self.check_output(check_pir=True)
+
+
 if __name__ == "__main__":
     unittest.main()
