推力/cudpp中求第k最小元素的算法

Question

我正在寻找推力/cudapp中的第k个最小元素算法的实现。我用谷歌搜索了一下，但似乎没有找到。有没有人知道有没有这样的算法？

我看到有重新排序，但它没有说第k个最小。

Answer 1

目前推力不提供选择算法(即std::nth_element )，尽管它在我们的雷达上，并且有good evidence可以在图形处理器上快速完成选择。您现在唯一的办法就是使用thrust::sort或thrust::sort_by_key (或它们的stable_变体)对数据进行排序，然后选择适当的元素。sort中的基本类型排序(例如int，float，char，double)是用一个非常快的基数排序代码实现的，所以绝对性能仍然很好，尽管效率不如specialized selection method。