Download examples/cpp_integration/device.cu from camenduru/thrust: direct link, hf CLI and curl.
- Browser
- Download file 423 Bytes
-
https://huggingface.co/camenduru/thrust/resolve/main/examples/cpp_integration/device.cu
- Command line
-
hf download hf://camenduru/thrust/examples/cpp_integration/device.cu
-
curl -L -o device.cu https://huggingface.co/camenduru/thrust/resolve/main/examples/cpp_integration/device.cu
423 Bytes
| 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()); | |
| } | |