From f9c1cdc13966135448345e6183db6fa0fca0a740 Mon Sep 17 00:00:00 2001
From: Leo Guo <58431564+ZibinGuo@users.noreply.github.com>
Date: Mon, 17 Oct 2022 13:56:46 +0800
Subject: [PATCH] Fix the bug of PHI kernel of reduce_sum in kunlun when using
 eager mode. (#47004)

test=kunlun
---
 paddle/phi/kernels/xpu/reduce_sum_grad_kernel.cc | 3 +++
 1 file changed, 3 insertions(+)

diff --git a/paddle/phi/kernels/xpu/reduce_sum_grad_kernel.cc b/paddle/phi/kernels/xpu/reduce_sum_grad_kernel.cc
index 9dc1fe92fac..218eb25c5c9 100644
--- a/paddle/phi/kernels/xpu/reduce_sum_grad_kernel.cc
+++ b/paddle/phi/kernels/xpu/reduce_sum_grad_kernel.cc
@@ -32,6 +32,9 @@ void ReduceSumGradKernel(const Context& dev_ctx,
   dev_ctx.template Alloc<XPUType>(x_grad);
   const auto* out_data = out_grad.data<XPUType>();
   auto* x_grad_data = x_grad->data<XPUType>();
+  if (dims_arr.size() == 0) {
+    reduce_all = true;
+  }
   const auto& input_dim_size = x.dims().size();
   std::vector<int> true_dims;
   for (size_t i = 0; i < dims.size(); ++i) {
-- 
GitLab