come trasmettere thrust :: device_vector <int> al puntatore raw

Question

Ho una spinta device_vector. Voglio lanciarlo su un puntatore raw in modo che possa passarlo a un kernel. Come posso farlo?

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

Answer 1

È possibile farlo utilizzando thrust::raw_pointer_cast. La classe di unità vettore ha una funzione di membro data che restituirà un thrust::device_ptr alla memoria detenuta dal vettore, che può essere gettato, qualcosa di simile:

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

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

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

(disclaimer: scritto in del browser, mai compilato, mai testato). C'è un esempio funzionante completo di questo incluso con la spinta: unwrap_pointer.cu