我的第一个 CUDA 程序有问题。在其中我尝试添加两个数组,但是当我打印结果时完全错误。
这是我的代码:
#include <stdio.h>
const int N=10;
__global__ void add(int *a, int *b, int *c) {
int tid = threadIdx.x;
c[tid] = a[tid] + b[tid];
}
int main(){
int a[N], b[N], c[N];
int *dev_a, *dev_b, *dev_c;
cudaMalloc( (void**)&dev_a, N * sizeof(int) );
cudaMalloc( (void**)&dev_b, N * sizeof(int) );
cudaMalloc( (void**)&dev_c, N * sizeof(int) );
for (int i=0; i<N; i++) {
a[i] = -i; b[i] = i * i;
}
cudaMemcpy ( dev_a, a, N * sizeof(int), cudaMemcpyHostToDevice );
cudaMemcpy ( dev_b, b, N * sizeof(int), cudaMemcpyHostToDevice );
add<<<1,N>>>(dev_a, dev_b, dev_c);
cudaMemcpy(c, dev_c, N * sizeof(int),cudaMemcpyDeviceToHost );
for (int i=0; i<N; i++) {
printf("%d + %d = %d\n", a[i],b[i],c[i]);
}
cudaFree (dev_a); cudaFree (dev_b); cudaFree (dev_c);
return 0;
}
正如您将看到的,我尝试在 C 中添加 A 和 B,但是,我不知道为什么,它不起作用。
这是输出:
0 + 0 = 0
-1 + 1 = 0
-2 + 4 = 1139031032
-3 + 9 = 32737
-4 + 16 = 0
-5 + 25 = 0
-6 + 36 = 1203890432
-7 + 49 = 32767
-8 + 64 = 6299112
-9 + 81 = 0
最后,我为我的英语道歉……在学习了 cuda 之后,我接下来要做的就是改进它。