Xu Ma
update
1c3c0d9
raw
history blame
423 Bytes
#include <thrust/sort.h>
#include <thrust/device_vector.h>
#include <thrust/copy.h>
#include "device.h"
void sort_on_device(thrust::host_vector<int>& h_vec)
{
// transfer data to the device
thrust::device_vector<int> d_vec = h_vec;
// sort data on the device
thrust::sort(d_vec.begin(), d_vec.end());
// transfer data back to host
thrust::copy(d_vec.begin(), d_vec.end(), h_vec.begin());
}