Skip to content

Commit

Permalink
Add AIR-Top-k reference
Browse files Browse the repository at this point in the history
  • Loading branch information
tfeher committed Nov 30, 2023
1 parent ed272c1 commit b1d628b
Showing 1 changed file with 6 additions and 0 deletions.
6 changes: 6 additions & 0 deletions cpp/include/raft/matrix/detail/select_radix.cuh
Original file line number Diff line number Diff line change
Expand Up @@ -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
Expand Down

0 comments on commit b1d628b

Please sign in to comment.