몽상가

#include <stdio.h>

#define N 1000

__global__ void add(int *a,int *b,int *c){

       /* (1) */

        int tid=blockIdx.x;

        if(tid<N) c[tid]=a[tid]+b[tid];

}

int main(void){

        int a[N],b[N],c[N];

        int *dev_a,*dev_b,*dev_c;

        /* (2) */

        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;

        }

        /* (3) */

        cudaMemcpy(dev_a,a,N*sizeof(int),cudaMemcpyHostToDevice);

        cudaMemcpy(dev_b,b,N*sizeof(int),cudaMemcpyHostToDevice);

        /* (4) */

        add<<<N,1>>>(dev_a,dev_b,dev_c);

        /* (5) */

        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]);

        /* (6) */

        cudaFree(dev_a);

        cudaFree(dev_b);

        cudaFree(dev_c);

        return 0;

}