fix roi_align roi_pool to static num 0 (#55342)

paddle/phi/kernels/cpu/roi_align_kernel.cc

@@ -192,6 +192,11 @@ void RoiAlignKernel(const Context& dev_ctx,
   int width = in_dims[3];
   int rois_num = boxes.dims()[0];
 
+  if (rois_num == 0) {
+    dev_ctx.template Alloc<T>(out);
+    return;
+  }
+
   auto in_stride = phi::stride(in_dims);
   auto roi_stride = phi::stride(boxes.dims());
   auto out_stride = phi::stride(out->dims());

paddle/phi/kernels/cpu/roi_pool_kernel.cc

@@ -37,6 +37,11 @@ void RoiPoolKernel(const Context& dev_ctx,
   int width = x_dims[3];
   int rois_num = boxes.dims()[0];
 
+  if (rois_num == 0) {
+    dev_ctx.template Alloc<T>(out);
+    return;
+  }
+
   auto in_stride = phi::stride(x_dims);
   auto arg_max_stride = phi::stride(arg_max->dims());
   auto box_stride = phi::stride(boxes.dims());

paddle/phi/kernels/gpu/roi_align_kernel.cu

@@ -153,7 +153,10 @@ void RoiAlignKernel(const Context& dev_ctx,
 
   int rois_num = boxes.dims()[0];
 
-  if (rois_num == 0) return;
+  if (rois_num == 0) {
+    dev_ctx.template Alloc<T>(out);
+    return;
+  }
 
   int output_size = out->numel();
   int blocks = NumBlocks(output_size);

paddle/phi/kernels/gpu/roi_pool_kernel.cu

@@ -118,7 +118,10 @@ void RoiPoolKernel(const Context& dev_ctx,
 
   int rois_num = boxes.dims()[0];
 
-  if (rois_num == 0) return;
+  if (rois_num == 0) {
+    dev_ctx.template Alloc<T>(out);
+    return;
+  }
 
   int output_size = out->numel();
   int blocks = NumBlocks(output_size);

paddle/phi/kernels/xpu/roi_align_kernel.cc

@@ -40,7 +40,10 @@ void RoiAlignKernel(const Context& dev_ctx,
 
   int rois_num = boxes.dims()[0];
 
-  if (rois_num == 0) return;
+  if (rois_num == 0) {
+    dev_ctx.template Alloc<T>(out);
+    return;
+  }
 
   DenseTensor roi_batch_id_list;
   roi_batch_id_list.Resize({rois_num});