一个推力问题：如何使用自定义排列顺序将 host_vector 复制到 device_vector？

Question

我在主机中有一个数组，我想以不同的顺序将它传输到设备。

我试过这个玩具代码符合nvc++ test.cpp -stdpar

$ cat test.cpp
#include <iostream>
#include <thrust/iterator/permutation_iterator.h>
#include <thrust/copy.h>
#include <thrust/sequence.h>
#include <thrust/host_vector.h>
#include <thrust/device_vector.h>
#include <array>

using Real = float;

int main(int argc, char* argv[]) {

        std::array<std::size_t,6> idx{0,1,2,3,5,4};


        thrust::host_vector<Real> hvec(6);

        thrust::sequence(hvec.begin(),hvec.end());

        typedef thrust::host_vector<Real>::iterator EleItor;
        typedef std::array<std::size_t,6>::iterator IdxItor;

        thrust::permutation_iterator<EleItor,IdxItor> itor(hvec.begin(),idx.begin());

        thrust::device_vector<Real> test;
        thrust::copy(itor,itor+6,test);  // error
        thrust::copy(itor,itor+6,std::ostream_iterator<Real>(std::cout," ");  

}

问题是thrust::copy不允许从主机复制到设备，我怎样才能绕过这个限制？

Answer 1

根据文档，可以使用thrust::copy从主机复制到设备，但您需要传递设备迭代器：

//-----------------------------vvvvvvvv--
thrust::copy(itor, itor+6, test.begin());

请注意，在此之前，您需要为设备向量分配 memory。
幸运的是， thrust::device_vector有一个构造函数，其大小将分配所需的 memory。
你可以使用类似的东西：

thrust::device_vector<Real> test(host_vector.size());

---

编辑（归功于@paleonix）：
还有另一个采用迭代器的构造函数，即可以在一行中同时进行分配和复制作为初始化，这也具有避免将 memory 不必要地初始化为0.0f的优点。

thrust::device_vector<Real> test(itor, itor+6);