CUDA|关于thrust的使用的一些总结
void gpuInit()
{ cudaSetDevice(0);
thrust::host_vector m(5);
//主机向量
thrust::sequence(m.begin(),m.end());
//顺序插入数0 1 2 3 ....
thrust::device_vector d_v(5);
//设备向量
thrust::device_vector d_vv(5);
//二维设备向量 double A[5][4];
for(int i = 0;
i < 5;
++i)
{
for(int j = 0;
j < 4;
++j)
{
A[i][j] = i + j;
}
} for(int i = 0;
i < 5;
++i)
{
d_vv[i] = A[i];
//给二维向量设备向量赋值 }
std::cout << "device_vector is size = " << d_vv.size() << std::endl;
for(int i = 0;
i < 5;
++i)
{
for(int j = 0;
j < 4;
++j)
{
std::cout << "d_vv[" << i << "][" << j << "] = " << d_vv[i][j]<< " ";
}
std::cout << std::endl;
} d_v = m;
//内存数据复制到显存里，一维数据比较好用
for(int i = 0;
i < m.size() ;
++i)
{
std::cout << "m[i] = " << m[i] << "";
}
std::cout << std::endl;
for(int i = 0;
i < 5 ;
++i)
{
std::cout << "d_v = " << d_v[i] << "";
}
std::cout << std::endl;
thrust::host_vector > h(5,thrust::host_vector(5));
//主机的二维向量可以这样申请，设备的就不行尝试了好多都不行 std::cout <<"host_vector to device:" << std::endl;
std::cout << "device_vector is size = " << d_vv.size() << std::endl;
for(int i = 0;
i < 5;
++i)
{
for(int j = 0;
j < 5;
++j)
{
std::cout << "d_vv[" << i << "][" << j << "] = " << d_vv[i][j]<< " ";
}
std::cout << std::endl;
} std::cout << "host_vector is size = " << h.size() << std::endl;
for(int i = 0;
i < 5;
++i)
{
for(int j = 0;
j < 5;
++j)
{
std::cout << h[i][j] << "";
}
std::cout << std::endl;
}
} 【CUDA|关于thrust的使用的一些总结】
推荐阅读
- 关于QueryWrapper|关于QueryWrapper,实现MybatisPlus多表关联查询方式
- 四首关于旅行记忆的外文歌曲
- 醒不来的梦
- 关于自我为中心的一点感想
- 「按键精灵安卓版」关于全分辨率脚本的一些理解（非游戏app）
- 关于Ruby的杂想
- 关于读书的思考
- 关于this的一些问题（1）
- 《声之形》
- 关于如何沟通的1/2/3