add PADDLE_ENFORCE_LE for unbind (#74224)

Author: Difers

paddle/phi/kernels/impl/unbind_kernel_impl.h

@@ -24,6 +24,13 @@ void UnbindKernel(const Context& dev_ctx,
                   const DenseTensor& x,
                   int axis,
                   std::vector<DenseTensor*> outs) {
+  int64_t unbind_numel = x.numel();
+  PADDLE_ENFORCE_LE(unbind_numel,
+                    std::numeric_limits<int>::max(),
+                    ::common::errors::PreconditionNotMet(
+                        "The number of proposals in unbind should be less than "
+                        "2^31, but got %lld. Please check the input tensor. ",
+                        unbind_numel));
   auto x_dims = x.dims();
   axis = axis < 0 ? x_dims.size() + axis : axis;
 

paddle/phi/kernels/stride/unbind_kernel.cc

@@ -32,6 +32,13 @@ void UnbindStridedKernel(const Context& dev_ctx,
         "FLAGS_use_stride_kernel is closed. Strided kernel "
         "be called, something wrong has happened!"));
   }
+  int64_t unbind_numel = x.numel();
+  PADDLE_ENFORCE_LE(unbind_numel,
+                    std::numeric_limits<int>::max(),
+                    ::common::errors::PreconditionNotMet(
+                        "The number of proposals in unbind should be less than "
+                        "2^31, but got %lld. Please check the input tensor. ",
+                        unbind_numel));
   int64_t num = static_cast<int64_t>(outs.size());
   int64_t start = 0;