581
社区成员
发帖
与我相关
我的任务
分享
__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;
}