diff --git a/paddle/phi/kernels/gpu/top_k_kernel.cu b/paddle/phi/kernels/gpu/top_k_kernel.cu index a791812432a06bba944fc0bf0be514fecc1e2094..bef328ec21a2030077da469aae02eb01b5d446f2 100644 --- a/paddle/phi/kernels/gpu/top_k_kernel.cu +++ b/paddle/phi/kernels/gpu/top_k_kernel.cu @@ -102,7 +102,7 @@ void TopkKernel(const Context& dev_ctx, // The conclusion is drawn from the data through multiple sets of // statistics - if (input_width >= 128 && k >= input_width * 0.75) { + if (input_width >= 128 && k >= input_width * 0.25) { auto* ctx = reinterpret_cast(&dev_ctx); if (phi::funcs::SortTopk(*ctx, input,