Advanced Search

thrust usage example

/ Published in: C++
Save to your folder(s)

Expand | Embed | Plain Text

Copy this code and paste it in your HTML 
  1. #include <thrust/host_vector.h>
  2. #include <thrust/device_vector.h>
  3. #include <thrust/sort.h>
  4. #include <cstdlib.h>
  5. int main(void)
  6. {
  7.    // generate 32M random numbers on the host
  8.    thrust::host_vector<int> h_vec(32 << 20);
  9.    thrust::generate(h_vec.begin(), h_vec.end(), rand);
  10.    // transfer data to the device
  11.    thrust::device_vector<int> d_vec= h_vec;
  12.    // sort data on the device (846M keys per sec on GeForceGTX 480)
  13.    thrust::sort(d_vec.begin(), d_vec.end());
  14.    // transfer data back to host
  15.    thrust::copy(d_vec.begin(), d_vec.end(), h_vec.begin());
  16.    return 0;
  17. }

Report this snippet


Comments

RSS Icon Subscribe to comments

You need to login to post a comment.