From b1d628b2ceaa7d339e2d35e224f049ea5a52b9ef Mon Sep 17 00:00:00 2001 From: Tamas Bela Feher Date: Thu, 30 Nov 2023 11:27:52 +0100 Subject: [PATCH] Add AIR-Top-k reference --- cpp/include/raft/matrix/detail/select_radix.cuh | 6 ++++++ 1 file changed, 6 insertions(+) diff --git a/cpp/include/raft/matrix/detail/select_radix.cuh b/cpp/include/raft/matrix/detail/select_radix.cuh index fa12005df2..2fbb7371b0 100644 --- a/cpp/include/raft/matrix/detail/select_radix.cuh +++ b/cpp/include/raft/matrix/detail/select_radix.cuh @@ -1151,6 +1151,12 @@ void radix_topk_one_block(const T* in, * * Note, the output is NOT sorted within the groups of `k` selected elements. * + * Reference: + * Jingrong Zhang, Akira Naruse, Xipeng Li, and Yong Wang. 2023. Parallel Top-K Algorithms on GPU: + * A Comprehensive Study and New Methods. In The International Conference for High Performance + * Computing, Networking, Storage and Analysis (SC ’23), November 12–17, 2023, Denver, CO, USA. + * ACM, New York, NY, USA. https://doi.org/10.1145/3581784.3607062 + * * @tparam T * the type of the keys (what is being compared). * @tparam IdxT