小编tim*_*eng的帖子

CUDA Dot产品

我正在尝试为双精度数组实现经典的点积内核,并对各个块的最终总和进行原子计算.我使用atomicAdd进行双精度编程指南的第116页中所述.可能我做错了.每个块中线程的部分和都是正确计算的,但是之后原子操作似乎没有正常工作因为每次我使用相同的数据运行我的内核,我会收到不同的结果.如果有人能发现错误或提供替代解决方案,我将不胜感激!这是我的内核:

__global__ void cuda_dot_kernel(int *n,double *a, double *b, double *dot_res)
{
    __shared__ double cache[threadsPerBlock]; //thread shared memory
    int global_tid=threadIdx.x + blockIdx.x * blockDim.x;
    int i=0,cacheIndex=0;
    double temp = 0;
    cacheIndex = threadIdx.x;
    while (global_tid < (*n)) {
        temp += a[global_tid] * b[global_tid];
        global_tid += blockDim.x * gridDim.x;
    }
    cache[cacheIndex] = temp;
    __syncthreads();
    for (i=blockDim.x/2; i>0; i>>=1) {
        if (threadIdx.x < i) {
            cache[threadIdx.x] += cache[threadIdx.x + i];
        }
        __syncthreads();
    }
    __syncthreads();
    if (cacheIndex==0) {
        *dot_res=cuda_atomicAdd(dot_res,cache[0]);
    }
}
Run Code Online (Sandbox Code Playgroud)

这是我的设备函数atomicAdd:

__device__ double …
Run Code Online (Sandbox Code Playgroud)

cuda dot-product

6
推荐指数
2
解决办法
6560
查看次数

标签 统计

cuda ×1

dot-product ×1