スラストデバイスイテレータを生ポインタに変換する

Question

thrust::host_vector<int>::iterator h_temp_iterator = h_temp.begin();とthrust::device_vector<int>::iterator d_temp_iterator = d_temp.begin();を未処理のポインタに変換する次の単純なコードを検討しています。

この目的のために、私は&(h_temp_iterator[0])と&(d_temp_iterator[0])をそれぞれ関数とカーネルに渡しています。前者（CPUケース）はコン​​パイルし、後者（GPUケース）はコン​​パイルしません。

Error 1 error : no suitable conversion function from "thrust::device_ptr<int>" to "int *" exists  

構成は以下のとおりです：

1. Windows 7、Visual Studio 2010、CUDA 7.5、3.5用にコンパイル私は、エラーメッセージの理由を理解していないので、2つのケースは、原則として対称にする必要があります建築。
2. Windows 10,Visual Studio 2013,CUDA 8.0、5.2アーキテクチャのためにコンパイルする。 talonmiesコメント続き

CODE

#include <thrust\host_vector.h> 
#include <thrust\device_vector.h> 

__global__ void testKernel(int *a, const int N) 
{ 
    int i = threadIdx.x; 

    if (i >= N) return; 

    a[i] = 2; 
} 

void testFunction(int *a, const int N) 
{ 
    for (int i = 0; i < N; i++) a[i] = 2; 
} 

int main() 
{ 
    const int N = 10; 

    thrust::host_vector<int> h_temp(N); 
    thrust::device_vector<int> d_temp(N); 

    thrust::host_vector<int>::iterator h_temp_iterator = h_temp.begin(); 
    thrust::device_vector<int>::iterator d_temp_iterator = d_temp.begin(); 

    testFunction(&(h_temp_iterator[0]), N); 
    testKernel<<<1, N>>>(&(d_temp_iterator[0]), N); 

    for (int i = 0; i < N; i++) printf("%i %i\n", i, h_temp[i]); 

    return 0; 
} 

Answer 1

、解決策は、完全に動作するコードを以下に

thrust::raw_pointer_cast(&d_temp_iterator[0]) 

なく

&d_temp_iterator[0] 

を渡すことです

#include <thrust\host_vector.h> 
#include <thrust\device_vector.h> 

__global__ void testKernel(int *a, const int N) 
{ 
    int i = threadIdx.x; 

    if (i >= N) return; 

    a[i] = 2; 

    printf("GPU %i %i\n", i, a[i]); 
} 

void testFunction(int *a, const int N) 
{ 
    for (int i = 0; i < N; i++) { 
     a[i] = 2; 
     printf("CPU %i %i\n", i, a[i]); 
    } 
} 

int main() 
{ 
    const int N = 10; 

    thrust::host_vector<int> h_temp(N); 
    thrust::device_vector<int> d_temp(N); 

    thrust::host_vector<int>::iterator h_temp_iterator = h_temp.begin(); 
    thrust::device_vector<int>::iterator d_temp_iterator = d_temp.begin(); 

    int *temp = thrust::raw_pointer_cast(&d_temp_iterator[0]); 

    testFunction(&(h_temp_iterator[0]), N); 
    testKernel<<<1, N>>>(temp, N); 

    return 0; 
}