如何使用 CUDA/Thrust 对两个数组/向量根据其中一个数组中的值进行排序

Question

这是一个关于编程的概念问题。

总而言之，我有两个数组/向量，我需要对一个数组/向量进行排序，并将更改传播到另一个数组/向量中，这样，如果我对 arrayOne 进行排序，则对于排序中的每个交换 - arrayTwo 都会发生同样的情况。现在，我知道 std::sort 允许您定义一个比较函数（对于我假设的自定义对象），并且我正在考虑定义一个比较函数来同时交换 arrayTwo。

所以我想要的是 - 使用 CUDA 根据其中一个向量中的值对两个向量进行排序。

这就是我的不确定性上升的地方，本质上我想使用 Thrust 库来进行排序。它是否支持自定义比较函数的定义？如果是这样，我仍然没有弄清楚如何传播 arrayTwo 中的更改（因为它将基于 CUDA）。

我确实没有时间在 CUDA 上实现自定义并行快速排序，尽管我应该/想要这样做。

原因

本质上，我需要对一堆变量数组与单个数组执行排序和计算（想想回归树）。当然，我需要尽快完成，基于 CPU 的排序不够快。

#UPDATE

我应该强调，我在主机上对两者进行排序没有问题，我正在寻找使用CUDA的解决方案。谢谢。

#UPDATE 2

我想我真的很幸运，自从我发布问题后找到了解决方案， 事实证明，Thrust 实际上提供了我默认所需的内容：

#include <thrust/sort.h>
  ...
  const int N = 6;
  int    keys[N] = {  1,   4,   2,   8,   5,   7};
  char values[N] = {'a', 'b', 'c', 'd', 'e', 'f'};
  thrust::sort_by_key(keys, keys + N, values);
  // keys is now   {  1,   2,   4,   5,   7,   8}
  // values is now {'a', 'c', 'b', 'e', 'f', 'd'}

*取自 http ://code.google.com/p/thrust/wiki/QuickStartGuide#Fancy_Iterators*

所以，现在我所要做的就是从两个数组中获取两个 Thrust::device_vectors （我必须摆脱二维数组）。快乐的。

Answer 1

创建一个整数索引向量，初始化为 { 0, 1, 2, 3, etc }。每个整数代表向量中的一个位置。使用自定义比较函数对索引向量进行排序，该函数使用索引来引用向量 1。完成后，您可以使用排序索引对向量 1 和 向量 2 进行重新排序。

但我认为你不能就地进行重新排序，所以无论如何你都必须从一个向量复制到另一个向量，所以我认为 Kerrek 的建议也很好。