cómo lanzar el empuje :: device_vector <int> al puntero sin procesar

Question

Tengo un empuje device_vector. Quiero convertirlo en un puntero sin formato para poder pasarlo a un kernel. ¿Como lo puedo hacer?

thrust::device_vector<int> dv(10); 
//CAST TO RAW 
kernel<<<bl,tpb>>>(pass raw) 

Answer 1

Puedes hacerlo usando thrust::raw_pointer_cast. La clase de vectores dispositivo tiene una función miembro data que devolverá un thrust::device_ptr a la memoria realizada por el vector, que puede ser fundido, algo como esto:

thrust::device_vector<int> dv(10); 

int * dv_ptr = thrust::raw_pointer_cast(dv.data()); 

kernel<<<bl,tpb>>>(dv_ptr) 

(disclaimer: escrito en el navegador, no compilado, nunca se probó) Hay un ejemplo completo de esto incluido con el empuje: unwrap_pointer.cu