学习cuda编程结构以及案例

#include "cuda_runtime.h"
#include "device_launch_parameters.h"

#include <stdio.h>
#include<math.h>
#include<time.h>
#include<stdlib.h>
    /*     CUDA编程结构     */
/*
    1.分配GPU内存
    2.从CPU内存中拷贝数据到GPU内存
    3.调用CUDA内核函数来完成程序指定运算
    4.将数据从GPU拷贝回CPU
    5.释放CPU内存空间
*/
//定义核函数
__global__ void Square(int *d_out, int *d_in)
{
    int idx = threadIdx.x;
    //printf("%d\n", idx);
    d_out[idx] = d_in[idx] * d_in[idx];
}


int main()
{
    //定义变量
    const int ARRAY_SIZE = 64;
    const int ARRAY_BYTES = ARRAY_SIZE * sizeof(int);
    //h_开头代表在主机上 
    int h_in[ARRAY_SIZE];
    //赋值随机数
    for (int i = 0;i < ARRAY_SIZE;i++)
    {
        //h_in[i] = rand() % 10;
        h_in[i] = i;
    }
    int h_out[ARRAY_SIZE];
    //d_代表在设备上
    int *d_in, *d_out;
    //1.在GPU上为变量分配内存
    cudaMalloc((void**)&d_in, ARRAY_BYTES);
    cudaMalloc((void**)&d_out, ARRAY_BYTES);

    //2.把CPU数据拷贝到GPU
    cudaMemcpy(d_in, h_in, ARRAY_BYTES, cudaMemcpyHostToDevice);

    //3.调用内核函数  三个尖括号代表着从主线程到设备端代码的调用
    Square << <1, ARRAY_SIZE >> > (d_out, d_in);
    //4. 把数据从GPU拷贝回内存
    cudaMemcpy(h_out, d_out, ARRAY_BYTES, cudaMemcpyDeviceToHost);

    //打印输出
    for (int i = 0;i < ARRAY_SIZE;i++)
    {
        printf("%d ", h_out[i]);
        i % 4 != 3 ? printf("\t") : printf("\n");
    }

    //5.释放GPU内存空间
    cudaFree(d_in);
    cudaFree(d_out);

    return 0;
}