Merge branch 'nanmedian' of https://github.com/thunder95/Paddle into …

…nanmedian
PaddlePaddle · May 20, 2022 · 5f14183 · 5f14183
2 parents b0c9471 + 718fcdb
commit 5f14183
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/paddle/phi/kernels/cpu/nanmedian_kernel.cc b/paddle/phi/kernels/cpu/nanmedian_kernel.cc
@@ -52,7 +52,7 @@ void CalcMedianFunc(const Context& dev_ctx,
   bool is_ori_odd = stride & 1;
   if (should_ignore_nan) {
     for (i = 0; i < pre_dim; i++) {
-      offset = i * pre_dim;
+      offset = i * sort_k;
       if (nan_counts[i] == stride) {
         m_ptr[i * 2] = -1;
         m_ptr[i * 2 + 1] = -1;