Question

我正在尝试测试CUDA网站http://docs.nvidia.com/cuda/cuda-c-programming-guide/index.html#kernels中的示例代码。

我只想添加两个大小为4的数组A和B，并将其存储在数组C中。这是我到目前为止所拥有的：

#include <stdio.h>
#include "util.h"
void print_array(int* array, int size) {
int i;
for (i = 0; i < size; i++) {
    printf("%d ", array[i]);
}
printf("\n");
}

__global__ void VecAdd(int* A, int* B, int* C) {
int i = threadIdx.x;
C[i] = A[i] + B[i];
}

int main(int argc , char **argv) {
int N = 4;
    int i;
int *A = (int *) malloc(N * sizeof(int));
int *B = (int *) malloc(N * sizeof(int));
int *C = (int *) malloc(N * sizeof(int));

for (i = 0; i < N; i++) {
    A[i] = i + 1;
    B[i] = i + 1;
}

print_array(A, N);
print_array(B, N);


VecAdd<<<1, N>>>(A, B, C);
print_array(C, N);
    return 0;
}

我期待C数组（输出的最后一行）为2,4,6,8，但似乎没有添加：

1 2 3 4
1 2 3 4
0 0 0 0

我错过了什么？

Answer 1

首先，您必须定义将保存将复制到GPU的数据的指针：

在您的示例中，我们要将数组'a'，'b'和'c'从CPU复制到GPU's全局内存。

int a[array_size], b[array_size],c[array_size]; // your original arrays
int *a_cuda,*b_cuda,*c_cuda;                    // defining the "cuda" pointers

定义每个数组将占用的大小。

int size = array_size * sizeof(int); // Is the same for the 3 arrays

然后，您将空间分配给将在cuda中使用的数据：

Cuda内存分配：

msg_erro[0] = cudaMalloc((void **)&a_cuda,size);
msg_erro[1] = cudaMalloc((void **)&b_cuda,size);
msg_erro[2] = cudaMalloc((void **)&c_cuda,size);

现在我们需要将这些数据从CPU复制到GPU：

从CPU复制到GPU ：

msg_erro[3] = cudaMemcpy(a_cuda, a,size,cudaMemcpyHostToDevice);
msg_erro[4] = cudaMemcpy(b_cuda, b,size,cudaMemcpyHostToDevice);
msg_erro[5] = cudaMemcpy(c_cuda, c,size,cudaMemcpyHostToDevice);

执行内核

int blocks = //;
int threads_per_block = //;
VecAdd<<<blocks, threads_per_block>>>(a_cuda, b_cuda, c_cuda);

将结果从GPU复制到CPU （在我们的示例数组C中）：

msg_erro[6] = cudaMemcpy(c,c_cuda,size,cudaMemcpyDeviceToHost);

免费记忆：

cudaFree(a_cuda);
cudaFree(b_cuda);
cudaFree(c_cuda);

出于调试目的，我通常会将函数的状态保存在数组中，如下所示：

cudaError_t msg_erro[var];

但是，这不是绝对必要的，但如果在分配或内存转移期间发生错误，它将节省您的时间。如果您愿意，可以从上面的代码中取出所有'msg_erro [x] ='。

如果您保留'msg_erro [x] ='，并且如果确实发生了错误，您可以使用类似下面的函数来打印这些错误：

void printErros(cudaError_t *erros,int size)
{
 for(int i = 0; i < size; i++)
      printf("{%d} => %s\n",i ,cudaGetErrorString(erros[i]));
}

Answer 2

你需要来回传输内存到GPU，比如

    int *a_GPU, *b_GPU, *c_GPU;

    cudaMalloc(&a_GPU, N*sizeof(int));
    cudaMalloc(&b_GPU, N*sizeof(int));
    cudaMalloc(&c_GPU, N*sizeof(int));

    cudaMemcpy(a_GPU, A, N*sizeof(int), cudaMemcpyHostToDevice);
    cudaMemcpy(b_GPU, B, N*sizeof(int), cudaMemcpyHostToDevice);

    VecAdd<<<1, N>>>(a_GPU, b_GPU, c_GPU);

    cudaMemcpy(C, c_GPU, N*sizeof(int), cudaMemcpyDeviceToHost);

    print_array(C, N);

    cudaFree(a_GPU);
    cudaFree(b_GPU);
    cudaFree(c_GPU);

从示例代码实现CUDA VecAdd

2 个答案: