C语言 在 CUDA 中使用设备 malloc() 获得分段错误

Question

在调用random_ints(b)时，代码在访问数组中的N-1位置后遇到了分割错误，与N的值无关。尽管调用random_ints(a)正常工作并将所有值分配给数组a。如果我使用数组的静态分配，int a[N],b[N],c[N]代码可以完美运行。该代码与 NVIDIA CUDA 基础知识教程中提供的代码完全相同，只是实现了random_ints()函数。我已经尝试了其他替代方案，通过替换main()中的函数调用或分配另一个数组并从random_ints()返回到main()或删除 CUDA 片段。如果没有 CUDA 部分，函数调用可以完美运行。

我正在寻找这个问题的解释。提前谢谢。

#include<stdio.h>
#include<stdlib.h>
#define N 512
__global__ void add(int *a,int *b,int *c)
{
    c[blockIdx.x]= a[blockIdx.x] + b[blockIdx.x];
}
void random_ints(int *x,int n)
{
    int i;
    for(i=0;i<n;i++)
    {
            printf("n%d",i);
            x[i]=i;
    }
}
int main()
{
    int i;
    int *a,*b,*c;   //host copies of a,b,c
    int *d_a, *d_b, *d_c;   //device copies of a,b,c
    int size =sizeof(int)*N;
    //Allocate memories for device copies of a,b,c
    cudaMalloc((void **)&d_a,size);
    cudaMalloc((void **)&d_b,size); 
    cudaMalloc((void **)&d_c,size);
    //Alloc space for host copies of a,b,c and setup input values
    a=(int*)malloc(size);
    random_ints(a,N);
    b=(int*)malloc(size);
    random_ints(b,N);   
    c=(int*)malloc(size);
    //copy inputs to the device 
    cudaMemcpy(d_a, &a, size, cudaMemcpyHostToDevice);
    cudaMemcpy(d_b, &b, size, cudaMemcpyHostToDevice);
    //Launch add() kernel on GPU 
    add<<<N,1>>>(d_a,d_b,d_c);
    //copy results back to host
    cudaMemcpy(&c, d_c, size, cudaMemcpyDeviceToHost);
    for(i=0;i<N;i++)
    {
            printf("nc= %dn",c[i]);
    }
    //cleanup
    free(a);free(b);free(c);
    cudaFree(d_a); cudaFree(d_b); cudaFree(d_c);
    return 0;
}

Answer 1

您正在将不正确的参数传递给cudaMemcpy。这：

cudaMemcpy(d_a, &a, size, cudaMemcpyHostToDevice);
cudaMemcpy(d_b, &b, size, cudaMemcpyHostToDevice);

应该是

cudaMemcpy(d_a, a, size, cudaMemcpyHostToDevice);
cudaMemcpy(d_b, b, size, cudaMemcpyHostToDevice);

或

cudaMemcpy(d_a, &a[0], size, cudaMemcpyHostToDevice);
cudaMemcpy(d_b, &b[0], size, cudaMemcpyHostToDevice);

同样，这个：

cudaMemcpy(&c, d_c, size, cudaMemcpyDeviceToHost);

应该是

cudaMemcpy(c, d_c, size, cudaMemcpyDeviceToHost);

或

cudaMemcpy(&c[0], d_c, size, cudaMemcpyDeviceToHost);

它将是托管副本的设备，这将是您问题的根源 - 它将覆盖堆栈并导致您看到的段错误。