From eda8df7133680ba56e7ab4cc762d265759621a6b Mon Sep 17 00:00:00 2001 From: RuohengMa <120699764+RuohengMa@users.noreply.github.com> Date: Sat, 6 May 2023 14:11:52 +0800 Subject: [PATCH] [XPU] substitute new api kernel for combinatorial adaptive avg_pool2d_grad kernel (#53528) --- paddle/phi/kernels/xpu/pool_grad_kernel.cc | 24 ---------------------- 1 file changed, 24 deletions(-) diff --git a/paddle/phi/kernels/xpu/pool_grad_kernel.cc b/paddle/phi/kernels/xpu/pool_grad_kernel.cc index 4eeef936216..b03be5dd944 100644 --- a/paddle/phi/kernels/xpu/pool_grad_kernel.cc +++ b/paddle/phi/kernels/xpu/pool_grad_kernel.cc @@ -112,30 +112,6 @@ void Pool2dGradKernel(const Context& ctx, true); } else if (pooling_type == "avg") { - // When output dim is 1 * 1 (1 * 1 * 1 in pool_3d), use scale - // and broadcast kernels to get same output, but better performance. - if (out_h == 1 && out_w == 1 && std::is_same::value) { - xpu::ctx_guard RAII_GUARD(ctx.x_context()); - float scale = 1.0 / (in_h * in_w); - float* scaled_dy = RAII_GUARD.alloc_l3_or_gm(n * c); - r = xpu::scale(ctx.x_context(), - dout.data(), - scaled_dy, - n * c, - true, - scale, - 0.0); - PADDLE_ENFORCE_XDNN_SUCCESS(r, "scale"); - - r = xpu::broadcast(ctx.x_context(), - scaled_dy, - dx->data(), - {n, c, 1, 1}, - {n, c, in_h, in_w}); - PADDLE_ENFORCE_XDNN_SUCCESS(r, "broadcast"); - - return; - } r = xpu::adaptive_avg_pool2d_grad( ctx.x_context(), reinterpret_cast(dout.data()), -- GitLab