From 2a992178ea713c8d4503534b57636e9cfa3a6e8a Mon Sep 17 00:00:00 2001
From: haosicheng <47998305+HarperCy@users.noreply.github.com>
Date: Fri, 26 Aug 2022 10:54:32 +0800
Subject: [PATCH] fix reduce mean grad bug *test=kunlun (#45401)

* add temporal shift and grad *test=kunlun

* fix reduce mean grad bug *test=kunlun
---
 paddle/fluid/operators/reduce_ops/reduce_mean_op_xpu.cc | 1 +
 1 file changed, 1 insertion(+)

diff --git a/paddle/fluid/operators/reduce_ops/reduce_mean_op_xpu.cc b/paddle/fluid/operators/reduce_ops/reduce_mean_op_xpu.cc
index 2a79d031dcb..108bba3b452 100644
--- a/paddle/fluid/operators/reduce_ops/reduce_mean_op_xpu.cc
+++ b/paddle/fluid/operators/reduce_ops/reduce_mean_op_xpu.cc
@@ -112,6 +112,7 @@ class ReduceMeanGradXPUKernel : public framework::OpKernel<T> {
         d = d + xdims.size();
       }
       reduce_numel *= xdims[d];
+      ydims.insert(ydims.begin() + d, 1);
     }
 
     float val = 1.0f / static_cast<float>(reduce_numel);
-- 
GitLab