c++ - CUDA在设备上静态分配数据

我一直在尝试分配每个内核函数都可以访问的变量。
我的尝试是下面附上的代码，但由于内核无法查看dArray，因此无法编译。在C++中，您可以将变量放在顶部或声明static，以在整个程序的每个范围中进行访问。

__global__ void StoreThreadNumber()
{
    dArray[threadIdx.x] = threadIdx.x;
}

int main( int argc, char** argv)
{
    unsigned __int8 Array[16] = { 0 };
    unsigned __int8 dArray[16];

    for( __int8 Position = 0; Position < 16; Position++)
        cout << Array[Position] << " ";
    cout << endl;

    cudaMalloc((void**) dArray, 16*sizeof(__int8));
    cudaMemcpy( dArray, Array, 16*sizeof(__int8), cudaMemcpyHostToDevice);

    StoreThreadNumber<<<1, 16>>>();

    cudaMemcpy( Array, dArray, 16*sizeof(__int8), cudaMemcpyDeviceToHost);

    for( __int8 Position = 0; Position < 16; Position++)
        cout << Array[Position] << " ";
    cout << endl;

    cudaFree(dArray);
}

最佳答案

您可以在CUDA中使用__device__或__constant__类型的全局变量。因此，例如，如果使用__constant__将cudaMemcpyToSymbol()指针变量初始化为设备指针的地址，则可以通过__constant__变量访问该指针:

__constant__ int* dArrayPtr;

__global__ void StoreThreadNumber()
{
    dArrayPtr[threadIdx.x] = threadIdx.x;
}

只需确保在运行内核之前从主机代码正确初始化了dArrayPtr。