Spaces:
Runtime error
Runtime error
int main(void) | |
{ | |
size_t N = 10; | |
// obtain raw pointer to device memory | |
int * raw_ptr; | |
cudaMalloc((void **) &raw_ptr, N * sizeof(int)); | |
// wrap raw pointer with a device_ptr | |
thrust::device_ptr<int> dev_ptr = thrust::device_pointer_cast(raw_ptr); | |
// use device_ptr in Thrust algorithms | |
thrust::fill(dev_ptr, dev_ptr + N, (int) 0); | |
// access device memory transparently through device_ptr | |
dev_ptr[0] = 1; | |
// free memory | |
cudaFree(raw_ptr); | |
return 0; | |
} | |