首页 > 代码库 > CUDA编程学习(二)

CUDA编程学习(二)

将数据加载到GPU后,如何在grid下的block进行并行计算(一个grid包含多个block)

/****How do we run code in parallel on the device****/    /****Use block****/    _global_ void add(int *a, int *b, int *c){    c[blockIdx.x] = a[blockIdx.x] + b[blockIdx.x];}#define N 512int main(){    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 = N * sizeof(int);        //Alloc space 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 the data into device    cudeMemcpy(d_a, a, size, cudaMemcpyHostToDevice);    cudaMemcpy(d_b, b, size, cudaMemcpyHostToDevice);        //Launch add() kernel on GPU with N blocks    add<<<N,1>>>(d_a, d_b, d_c);        //Copy result back to host    cudaMemcpy(c, d_c, size, cudaMemcpyDeviceToHost);        //Cleanup    free(a); free(b); free(c);    cudeFree(d_a); cudaFree(d_b); cudaFree(d_c);    return 0;}/**** What‘s the function of random_ints****/void random_ints(int* a, int N){ int i; for (i = 0; i < N; ++i) a[i] = rand();}

 

CUDA编程学习(二)