cách đúc lực đẩy :: device_vector <int> đến con trỏ thô

Question

Tôi có lực đẩy device_vector. Tôi muốn đưa nó đến một con trỏ thô để tôi có thể truyền nó tới một hạt nhân. Làm thế nào tôi có thể làm như vậy?

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

Answer 1

Bạn có thể thực hiện việc này bằng cách sử dụng thrust::raw_pointer_cast. Lớp vector thiết bị có chức năng thành viên data mà sẽ trả về một thrust::device_ptr vào bộ nhớ được tổ chức bởi các véc tơ, mà có thể được đúc, một cái gì đó như thế này:

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

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

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

(từ chối trách nhiệm: bằng văn bản trong trình duyệt, không bao giờ được biên dịch, không bao giờ thử nghiệm). Có một ví dụ làm việc đầy đủ về việc này bao gồm lực đẩy: unwrap_pointer.cu