堆栈内存溢出
  繁体   English   中英

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

[英]A thrust problem: How can I copy a host_vector to device_vector with a customized permutation order?

 batman216  2023-01-19 11:03:45  51   1    c++/ gpu/ thrust

提示:本站为国内最大中英文翻译问答网站，提供中英文对照查看，鼠标放在中文字句上可显示英文原文。

问题描述

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

我试过这个玩具代码符合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不允许从主机复制到设备，我怎样才能绕过这个限制？

1 条回复

1楼
 wohlstad  3 已采纳  2023-01-19 11:14:38

根据文档，可以使用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);
问题未解决？试试搜索： 一个推力问题：如何使用自定义排列顺序将 host_vector 复制到 device_vector？。
暂无
暂无

声明:本站的技术帖子网页，遵循CC BY-SA 4.0协议，如果您需要转载，请注明本站网址或者原文地址。任何问题请咨询:yoyou2525@163.com.

相关问题 如何防止将推力的 device_vector 复制到设备 如何通过推力将主机矢量复制到设备矢量 如何投射二维推力:: device_vector <thrust::device_vector<int> &gt;指向原始指针 在device_vector上使用推力:: iterator_adaptor时出现问题 2D推力:: device_vector到内核 通过引用将推力::device_vector 传递给 function 确定最大推力长度::device_vector 将thrust :: device_vector放在全局范围内 计算推力:: device_vector上的梯度 类的CUDA推力:: device_vector | 错误
相关标签
 
粤ICP备18138465号  © 2020-2023 STACKOOM.COM