并行缩减#5展开最后一次扭曲

时间:2017-05-19 10:01:03

标签: cuda

我正在研究马克哈里斯对着名幻灯片的减少。特别是,我实现了优化步骤#5但我得到了错误的结果:17而不是41.我使用了幻灯片中显示的相同数字序列。我在代码中省略了" extern"对于共享数组,因为内核和主机代码在同一个.cu文件中。

#include <stdio.h>
#include <cuda_runtime.h>

#define THREAD_PER_BLOCK 16

__global__ void reduce5(int *g_idata, int *g_odata) {
    __shared__ int sdata[THREAD_PER_BLOCK];
    // perform first level of reduction,
    // reading from global memory, writing to shared memory
    unsigned int tid = threadIdx.x;
    unsigned int i = blockIdx.x*(blockDim.x*2) + threadIdx.x;
    sdata[tid] = g_idata[i] + g_idata[i+blockDim.x];
    __syncthreads();
    // do reduction in shared mem
    for (unsigned int s=blockDim.x/2; s>32; s>>=1) {
        if (tid < s) sdata[tid] += sdata[tid + s];
        __syncthreads();
    }

    if (tid < 32)
    {
        sdata[tid] += sdata[tid + 32];
        sdata[tid] += sdata[tid + 16];
        sdata[tid] += sdata[tid + 8];
        sdata[tid] += sdata[tid + 4];
        sdata[tid] += sdata[tid + 2];
        sdata[tid] += sdata[tid + 1];
    }
    // write result for this block to global mem
    if (tid == 0) g_odata[blockIdx.x] = sdata[0];
}

int main()
{
    int inputLength=16;
    int hostInput[16]={10,1,8,-1,0,-2,3,5,-2,-3,2,7,0,11,0,2};
    int hostOutput=0;
    int *deviceInput;
    int *deviceOutput;

    cudaMalloc((void **)&deviceInput, inputLength * sizeof(int));
    cudaMalloc((void **)&deviceOutput, sizeof(int));

    cudaMemcpy(deviceInput, hostInput, inputLength * sizeof(int),cudaMemcpyHostToDevice);

    reduce5<<<1,16>>>(deviceInput, deviceOutput);

    cudaDeviceSynchronize();

    cudaMemcpy(&hostOutput, deviceOutput,sizeof(int), cudaMemcpyDeviceToHost);

    printf("%d\n",hostOutput);

    cudaFree(deviceInput);
    cudaFree(deviceOutput);

    return 0;
}

2 个答案:

答案 0 :(得分:2)

在此代码中,THREAD_PER_BLOCK必须是32的倍数,并且至少为64.输入数据的长度也必须是块和网格大小的两倍。

您没有看到它(因为您没有执行任何类型的错误检查),但由于超出范围的共享内存和全局内存访问,线程和warp减少将失败。

另请注意,extern __shared__与内核和其他代码是否在同一文件中无关。这表示该变量的共享内存将在运行时动态分配,而不是在编译时静态分配。分配的大小作为内核启动语法中的第三个参数传递。

答案 1 :(得分:1)

我遇到了同样的问题,发现如果变量未声明为volatile,则线程实际上并不同步。

在声明volatile时,只需添加sdata即可解决问题。 请参阅我的帖子:cuda Threads in A Warp appear to be not in synchronization