cuda 排序算法笔记

x33g5p2x  于2022-03-31 转载在 其他  
字(0.7k)|赞(0)|评价(0)|浏览(349)

判断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

相关文章