From 4f4d42dd2df7a37b1d4126eaebe526b1dc641232 Mon Sep 17 00:00:00 2001
From: qshuihu <humingqing@yq01-sys-hic-v100-box-a225-0106.yq01.baidu.com>
Date: Thu, 26 May 2022 15:06:05 +0800
Subject: [PATCH] fix mini topk bug

---
 paddle/fluid/operators/top_k_op.cu | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/paddle/fluid/operators/top_k_op.cu b/paddle/fluid/operators/top_k_op.cu
index 9d78ee7568d59..3c0398c20ce18 100644
--- a/paddle/fluid/operators/top_k_op.cu
+++ b/paddle/fluid/operators/top_k_op.cu
@@ -66,7 +66,7 @@ __global__ void KernelSortTopK(const size_t num_rows, const T* input_val,
       }
     } else {
       for (int i = 0; i < K; ++i) {
-        int pos = (i + 1) % K;
+        int pos = (i + 1) % num_cols;
         val[i] = in[pos];
         ind[i] = pos;
       }