一個推力問題：如何使用自定義排列順序將 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);