判断nan的方法:
https://blog.csdn.net/jacke121/article/details/123836382
Thrust是cuda自带的c++库,cuda安装好之后,这个库也默认安装好了。
这个库基本是采用类似STL的接口方式,因此对于开发者非常友好,开发者不再需要关注内存与显存相关的问题了。
#include <thrust/host_vector.h>
#include <thrust/device_vector.h>
#include <thrust/generate.h>
#include <thrust/sort.h>
#include <thrust/copy.h>
#include <algorithm>
#include <vector>
#include <time.h>
int main(void)
{
thrust::host_vector<int> h_vec(1024*1024);
std::generate(h_vec.begin(), h_vec.end(), rand);
std::vector<int> vec(h_vec.size());
thrust::copy(h_vec.begin(), h_vec.end(), vec.begin());
thrust::device_vector<int> d_vec = h_vec;
clock_t time1,time2
版权说明 : 本文为转载文章, 版权归原作者所有 版权申明
原文链接 : https://blog.csdn.net/jacke121/article/details/123836020
内容来源于网络,如有侵权,请联系作者删除!