update

PaddlePaddle · Apr 15, 2024 · 172319b · 172319b
1 parent 1eb8f80
commit 172319b
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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);