个人中心
gpt4 book ai didi

c++ - CUDA 直方图遇到非法内存访问 (77)

转载 作者：太空宇宙 更新时间：2023-11-04 07:10:20 24 4
gpt4 key购买 nike

<分区>

所以这是我几乎完整的代码:第一个内核是正常的全局直方图，可以正常工作。但我收到错误“遇到非法内存访问(77)”在计算 shared_histogram 之后的最终 memcpy。我不知道代码有什么问题。似乎共享直方图确实改变了 d_hist2 的大小。我还检查了 bin_count 是否已更改。但它没有。那么我的 shared_histog 内核是错误的还是我在 memCpy 上做错了？？注意:w * h * nc 是我输入图像的大小

__global__ void histog( int *img, int *hist, int bin_count, int n)
{
int x = threadIdx.x + blockDim.x *blockIdx.x;
if(x>=n)   return;

    unsigned char value = img[x];
int bin = value % bin_count;
atomicAdd(&hist[bin],1);

}

__global__ void shared_histog( int *img, int *hist, int n)
{
int x = threadIdx.x + blockDim.x *blockIdx.x;
int indx = threadIdx.x;
if(x>n)   return;

__shared__ int shHist[256];

if (indx < 256)
shHist[indx] =0;
__syncthreads();

unsigned char value = img[x];
__syncthreads();

atomicAdd( (int*)&shHist[value], 1);
__syncthreads();

atomicAdd( (int*)&(hist[indx]), shHist[indx] );

}



int main(int argc, char **argv)
{
cudaDeviceSynchronize();  CUDA_CHECK;


int *imgval  = new int[(size_t)w*h*nc];
for (int i =0; i<w*h*nc; i++)
    imgval[i] = (imgIn[i])*256 + 1;


int bin_count = 256;
int *Histogram  = new int[bin_count];
int *Histogram2  = new int[bin_count];

for (int i =0; i <bin_count; i++)
    Histogram2[i] = 0;

Timer timer; timer.start();
for (int i =0; i <bin_count; i++)
    Histogram[i] = 0;
for (int i =0; i<w*h*nc; i++)
    Histogram[(imgval[i])]++;

showHistogram256("CPU_Histo", Histogram, 100 + w + 40, 100);


timer.end();  float t = timer.get();  // elapsed time in seconds
cout << "CPU time: " << t*1000 << " ms" << endl;


int *d_img = NULL;
int nbytes = w * h * nc * sizeof(int);
cudaMalloc(&d_img, nbytes); CUDA_CHECK;
cudaMemcpy(d_img, imgval, nbytes, cudaMemcpyHostToDevice); CUDA_CHECK;

int *d_hist = NULL;
cudaMalloc(&d_hist, bin_count * sizeof(int)); CUDA_CHECK;
cudaMemset(d_hist, 0, bin_count * sizeof(int)); CUDA_CHECK;

int *d_hist2 = NULL;
cudaMalloc(&d_hist2, bin_count * sizeof(int)); CUDA_CHECK;
cudaMemset(d_hist2, 0, bin_count * sizeof(int)); CUDA_CHECK;

dim3 block = dim3(1024,1,1);
dim3 grid = dim3 ((w*h*nc+block.x-1)/block.x, 1, 1);

Timer timer2; timer2.start();
histog <<<grid, block>>> (d_img, d_hist, bin_count, nbytes);    CUDA_CHECK;
    timer2.end();  float t2 = timer2.get();  // elapsed time in seconds
    cout << "GPU time: " << t2*1000 << " ms" << endl;
cudaMemcpy(Histogram, d_hist,bin_count * sizeof(int), cudaMemcpyDeviceToHost); CUDA_CHECK;
showHistogram256("GPU_Histo", Histogram, 100 + w + 40, 100 + h/2 + 10);


Timer timer3; timer3.start();
shared_histog <<<grid, block>>> (d_img, d_hist2, nbytes); CUDA_CHECK;
timer3.end();  float t3 = timer3.get();  // elapsed time in seconds
    cout << "Shared time: " << t3*1000 << " ms" << endl;

* 错误来了 *

cudaMemcpy(Histogram2, d_hist2, 256 * sizeof(int), cudaMemcpyDeviceToHost);  CUDA_CHECK;

showHistogram256("GPU_Histo_Shared", Histogram2, 100 + w + 40, 100 + h +10);


return 0;
}

24 4 0
文章推荐： c - 为什么在调用 getc 时会出现段错误？
文章推荐： python - 如何在 Python 的列表中扩展列表？
文章推荐： python - 如何连接字符串 for ... in ...？ (Python)
文章推荐： c++ - Realloc 和 glocal new/delete 运算符覆盖
太空宇宙
个人简介

我是一名优秀的程序员,十分优秀！

滴滴打车优惠券免费领取
滴滴打车优惠券
全站热门文章
Copyright 2021 - 2024 cfsdn All Rights Reserved 蜀ICP备2022000587号
广告合作:1813099741@qq.com 6ren.com