Question

在GPGPU上，用cuda我的问题是：我有一个256个元素的向量，我想创建一个程序，可以提取非零值的位置并将它们复制到另一个向量。

我的代码不起作用：

dev_Hist：是数据源，初始向量;

dev_Xn：是dev_Hist上非零值位置的向量;

nN：是dev_Hist上非零值的数量

1。内核电话：

gpu_Xn<<<1, nN>>>(dev_Hist, nN, dev_Xn) ;

2。设备功能

__global__ void gpu_Xn(int *pHist, int pnN, int* pXn) 
{
    int Tid ;
    Tid = threadIdx.x ;

    __shared__ T tmpXn[256] ;

    tmpXn[Tid] = 0 ;

    __syncthreads() ;

    __shared__ int idx ;

    if(Tid == 0)
        idx = -1  ;

    syncthreads() ;

    if(pHist[Tid] !=0)
    {
        atomicAdd(&idx, 1) ; 
        tmpXn[idx] = Tid ;
    }

    __syncthreads() ;
    if(Tid < pnN)
        pXn[Tid] = tmpXn[Tid] ;
}

Answer 1

此处的问题是您没有正确使用atomicAdd。尽管您以原子方式递增idx的值，但idx的读取对于共享内存的存储不是原子的，这将产生未定义的行为。

你的内核可能应该是这样的：

__global__ void gpu_Xn(int *pHist, int pnN, int* pXn) 
{
    int Tid ;
    Tid = threadIdx.x ;

    __shared__ int tmpXn[256] ;
    __shared__ int idx ;

    tmpXn[Tid] = -1 ;
    if(Tid == 0) idx = 0  ;

    __syncthreads() ;

    if(pHist[Tid] !=0)
    {
        int x = atomicAdd(&idx, 1) ; 
        tmpXn[x] = Tid ;
    }

    __syncthreads() ;
    if(Tid < pnN)
        pXn[Tid] = tmpXn[Tid] ;
}

[免责声明：用浏览器编写，永不编译，自担风险使用]

请注意atomicAdd返回原子更新的位置的先前值。这是加载到共享内存时需要使用的值。

CUDA：将矢量的非零值位置复制到另一个

1 个答案: