如何在cuda內核中調用推力函數

Question

當我運行這段代碼時，編譯器說我正在從設備調用一個主機函數。我不太明白如何。

__global__ void kernel(thrust::device_vector<float*> d_V) { 

     float *var = d_V[0]; 
    } 

    int main() { 

     thrust::host_vector<float*> V; 
     thrust::host_vector<float*> d_V; 

     float f[10]; 
     for (int i = 0; i < 10; i++) { 
      f[i] = i; 
     } 
     V.push_back(f); 
     d_V = V; 
     kernel<<<1, 1>>>(d_V); 

     return 0;  
    } 

Answer 1

推力功能/方法被設計用於主機（CPU）側。它們不能在CUDA內核的設備（GPU）端被調用。

您在代碼中演示的內容實際上是將一些數據傳遞給內核。數據可以被內核參數列表中除推力容器以外的原始指針引用。

__global__ void kernel(float* p) 
{ 
    float *var = p; 
} 

int main() 
{ 
    thrust::device_vector<float> d_v(
     thrust::make_counting_iterator((float)0), 
     thrust::make_counting_iterator((float)0)+10); 
    kernel<<<1,1>>>(thrust::raw_pointer_cast(&d_v[0])); 
}