cuda - 使用常量内存打印地址而不是 cuda 中的值

Question

我正在尝试在代码中使用常量内存，并从内核分配常量内存值，而不是使用 cudacopytosymbol。

 #include <iostream>
    using namespace std;
    #define N 10
    //__constant__ int constBuf_d[N];
    __constant__ int *constBuf;

__global__ void foo( int *results )
{
    int tdx = threadIdx.x;
    int idx = blockIdx.x * blockDim.x + tdx;


    if( idx < N )
    {
        constBuf[idx]=1;
         results[idx] = constBuf[idx];
    }
}

// main routine that executes on the host
int main(int argc, char* argv[])
{
    int *results_h = new int[N];
    int *results_d;


    cudaMalloc((void **)&results_d, N*sizeof(int));

    foo <<< 1, 10 >>> ( results_d );

    cudaMemcpy(results_h, results_d, N*sizeof(int), cudaMemcpyDeviceToHost);

    for( int i=0; i < N; ++i )
        printf("%i ", results_h[i] );
        delete(results_h);
}

输出显示

6231808 6226116 0 0 0 0 0 0 0 0 

我希望程序通过代码中的 kenel 打印分配给常量内存的值。

Answer 1

顾名思义，常量内存对于设备代码来说是常量/只读的。你试图做的事情是非法的，不能让它工作。

要在常量内存中设置值，您目前有两种选择：

1. cudaMemcpyToSymbol通过API 调用（或其等效项）从主机代码中设置值
2. 在编译时使用静态初始化

在后一种情况下，这样的事情会起作用：

__constant__ int constBuf[N] = { 16, 2, 77, 40, 12, 3, 5, 3, 6, 6 };

__global__ void foo( int *results )
{
    int tdx = threadIdx.x;
    int idx = blockIdx.x * blockDim.x + tdx;


    if( tdx < N )
    {
        results[idx] = constBuf[tdx]; // Note changes here!
    }
}