add PADDLE_ENFORCE_LE for unbind

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 FPN 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 FPN 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;