面试官面试问：你能写出十种矩阵加法的cuda程序吗？(二)

前言

在CUDA编程中，线程、线程块和网格是三个基本的并行计算单位：

1. 线程 (Thread)：线程是执行基本计算任务的最小单元，每个线程执行相同的代码但处理不同的数据。

2. 线程块 (Thread Block)：线程块是一组线程的集合，这些线程可以共享数据并通过同步机制协调工作，线程块的大小在执行时固定。

3. 网格 (Grid)：网格是多个线程块的集合，所有线程块并行执行任务，网格的大小同样在执行时固定。

01 一般编写

__global__ void addKernel(float *pA, float *pB, float *pC, int size)
{
    int index = blockIdx.x * blockDim.x + threadIdx.x; // 计算当前数组中的索引
    if (index >= size)
        return;

    pC[index] = pA[index] + pB[index];
}

addKernel<<<arr_len + 512 - 1, 512>>>(dev_a, dev_b, dev_c, arr_len);

02 使用线程

__global__ void addKernel_thread(float *pA, float *pB, float *pC, int size)
{
    int index = threadIdx.x; // 计算当前数组中的索引
    if (index >= size)
        return