带迭代器的CUB排序

Question

我想转换价值观并一次排序，如下所示：

thrust::vector<int> dataIn  = ...
thrust::vector<int> dataOut = ...
auto iterIn = cub::TransformInputIterator<int, Ftor, int*>(dataIn.begin(), Ftor());
cub::DeviceRadixSort::SortKeys(dTemp, tempBytes, iterIn, dataOut.begin(), numElems);

但是，SortKeys需要原始指针而不是迭代器。但是，是否可以使用迭代器来完成这项工作？我知道这是可能的推力，但我想使用幼崽。

谢谢你的建议。

Answer 1

很抱歉让人失望，但是AFAIK的幼崽不支持这个。从理论上讲，它可以有更深层次的温度，但事实并非如此。

您可以从cub中提取代码，也可以使用额外的模板参数修改代码。这将是一个令人头痛的问题，但如果您只想通过一个设备端函数通过某种转换传递输入值，这是可行的。