码迷,mamicode.com
首页 > 其他好文 > 详细

向Kernel函数传递thrust vector的方法

时间:2014-07-31 09:31:45      阅读:180      评论:0      收藏:0      [点我收藏+]

标签:http   color   os   io   for   cti   代码   ar   

废话不说,直接上代码

C/C++ code?
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
__global__ 
void Kernel(int* dv)
{
    int i = threadIdx.x;
    dv[i] = i;
}
 
int main()
{
    thrust::device_vector<int> dv(10);
     
    Kernel<<<1,10>>>(thrust::raw_pointer_cast(&dv[0])); 
 
 
    thrust::host_vector<int> hv = dv;
 
    for (int i=0; i<10; ++i)
    {
        printf("%d\n", hv[i]);
    }
 
    return 0;
}



运行结果:

0
1
2
3
4
5
6
7
8
9

向Kernel函数传递thrust vector的方法,布布扣,bubuko.com

向Kernel函数传递thrust vector的方法

标签:http   color   os   io   for   cti   代码   ar   

原文地址:http://www.cnblogs.com/iasd923/p/3879843.html

(0)
(0)
   
举报
评论 一句话评论(0
登录后才能评论!
© 2014 mamicode.com 版权所有  联系我们:gaon5@hotmail.com
迷上了代码!