diff --git a/paddle/phi/kernels/gpu/argsort_kernel.cu b/paddle/phi/kernels/gpu/argsort_kernel.cu
index 3b51a3fceddbb3..54193d88cc47c2 100644
--- a/paddle/phi/kernels/gpu/argsort_kernel.cu
+++ b/paddle/phi/kernels/gpu/argsort_kernel.cu
@@ -260,7 +260,7 @@ void ArgsortKernel(const Context& dev_ctx,
                                    out_data,
                                    out_data + size,
                                    ids_data,
-                                   thrust::greater<KeyT>());
+                                   thrust::greater<T>());
       } else {
         thrust::stable_sort_by_key(
             thrust::device, out_data, out_data + size, ids_data);