fix

Author: enkilee

paddle/phi/kernels/impl/unfold_grad_kernel_impl.h

@@ -17,7 +17,6 @@
 #include <vector>
 
 #include "paddle/phi/core/dense_tensor.h"
-#include "paddle/phi/kernels/full_kernel.h"
 #include "paddle/phi/kernels/funcs/im2col.h"
 #include "paddle/phi/kernels/funcs/math_function.h"
 #include "paddle/phi/kernels/funcs/unfold_functor.h"
@@ -33,14 +32,8 @@ void UnfoldGradKernel(const Context& dev_ctx,
                       const std::vector<int>& paddings,
                       const std::vector<int>& dilations,
                       DenseTensor* x_grad) {
-  if (out_grad.numel() == 0) {
-    if (x_grad) {
-      dev_ctx.template Alloc<T>(x_grad);
-      phi::Full<T, Context>(
-          dev_ctx, phi::IntArray(common::vectorize(x.dims())), 0, x_grad);
-    }
-    return;
-  }
+  dev_ctx.template Alloc<T>(x_grad);
+  if (x_grad->numel() == 0) return;
 
   if (!x_grad) return;
   const auto& x_dims = x_grad->dims();

paddle/phi/kernels/impl/unfold_kernel_impl.h

@@ -17,7 +17,6 @@
 #include <vector>
 
 #include "paddle/phi/core/dense_tensor.h"
-#include "paddle/phi/kernels/full_kernel.h"
 #include "paddle/phi/kernels/funcs/im2col.h"
 #include "paddle/phi/kernels/funcs/math_function.h"
 #include "paddle/phi/kernels/funcs/unfold_functor.h"
@@ -33,11 +32,8 @@ void UnfoldKernel(const Context& dev_ctx,
                   const std::vector<int>& dilations,
                   DenseTensor* out) {
   const int batch_size = static_cast<int>(x.dims()[0]);
-  if (x && x.numel() == 0) {
-    phi::Full<T, Context>(
-        dev_ctx, phi::IntArray(common::vectorize(out->dims())), 0, out);
-    return;
-  }
+  dev_ctx.template Alloc<T>(out);
+  if (out->numel() == 0) return;
   phi::funcs::Im2ColFunctor<phi::funcs::ColFormat::kCFO, Context, T> im2col;
   const auto& x_dims = x.dims();
 

paddle/phi/kernels/xpu/unfold_grad_kernel.cc

@@ -16,7 +16,6 @@
 
 #include "paddle/phi/backends/xpu/enforce_xpu.h"
 #include "paddle/phi/core/kernel_registry.h"
-#include "paddle/phi/kernels/full_kernel.h"
 #include "paddle/phi/kernels/funcs/unfold_functor.h"
 
 namespace phi {
@@ -31,14 +30,9 @@ void UnfoldGradKernel(const Context& dev_ctx,
                       const std::vector<int>& dilations_,
                       DenseTensor* x_grad) {
   using XPUType = typename XPUTypeTrait<T>::Type;
-  if (out_grad.numel() == 0) {
-    if (x_grad) {
-      dev_ctx.template Alloc<T>(x_grad);
-      phi::Full<T, Context>(
-          dev_ctx, phi::IntArray(common::vectorize(x.dims())), 0, x_grad);
-    }
-    return;
-  }
+  dev_ctx.template Alloc<T>(x_grad);
+  if (x_grad->numel() == 0) return;
+
   const std::string data_format = common::DataLayoutToString(x.layout());
   bool is_nchw = data_format == "NCHW";
   PADDLE_ENFORCE_EQ(is_nchw,

paddle/phi/kernels/xpu/unfold_kernel.cc

@@ -16,7 +16,6 @@
 
 #include "paddle/phi/backends/xpu/enforce_xpu.h"
 #include "paddle/phi/core/kernel_registry.h"
-#include "paddle/phi/kernels/full_kernel.h"
 #include "paddle/phi/kernels/funcs/unfold_functor.h"
 
 namespace phi {
@@ -30,11 +29,8 @@ void UnfoldKernel(const Context& dev_ctx,
                   const std::vector<int>& dilations_,
                   DenseTensor* out) {
   using XPUType = typename XPUTypeTrait<T>::Type;
-  if (x.numel() == 0) {
-    phi::Full<T, Context>(
-        dev_ctx, phi::IntArray(common::vectorize(out->dims())), 0, out);
-    return;
-  }
+  dev_ctx.template Alloc<T>(out);
+  if (out->numel() == 0) return;
   const std::string data_format = common::DataLayoutToString(x.layout());
   bool is_nchw = data_format == "NCHW";
   PADDLE_ENFORCE_EQ(is_nchw,