Thrust :: sort和transform_iterator

时间:2012-06-13 10:00:48

标签: cuda transform thrust stable-sort

我想对整数值列表进行排序,但在对它们进行排序之前,我应该将它们除以数字N.所以我会有一些重复的键,我会在列表中使用这个重复的stable_sort。

我的问题是,哪个更好地划分所有值并将划分的值存储在列表中然后执行排序,还是使用transform_iterator?使用transform_iterator会将排序算法从radix_sort更改为merge_sort,因为它们有很大的时差。

例如:

//already sorted according to another parameter
thrust::device_vector<int> myvalues...

//we want to group them..
thrust::transform(myvalues.begin(), myvalues.end(), groups.begin(), divide_by_n(N));
thrust::stable_sort_by_key(groups.begin(), groups.end(), myvalues.begin();

first = thrust::make_transform_iterator(myvalues.begin(), divide_by_n(N));
last  = thrust::make_transform_iterator(myvalues.end(),   divide_by_n(N));
thrust::stable_sort_by_key(first, last, myvalues.begin());

由于

1 个答案:

答案 0 :(得分:2)

根据这篇文章(来自@JaredHoberock的评论),第二个不起作用。 How to sort with less precision on keys with Thrust library