From 718fcdbce2f01be3af6887b1e4eb2d415b571a55 Mon Sep 17 00:00:00 2001
From: thunder95 <290844930@qq.com>
Date: Thu, 12 May 2022 07:56:33 +0000
Subject: [PATCH] typo pre_dim

---
 paddle/phi/kernels/cpu/nanmedian_kernel.cc | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/paddle/phi/kernels/cpu/nanmedian_kernel.cc b/paddle/phi/kernels/cpu/nanmedian_kernel.cc
index e207366bd04cc..ed38405c9179f 100644
--- 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;