CUDA的原子操作

CUDA的原子操作是用于在并行计算中保证多个线程对共享数据进行安全访问的一种机制。由于CUDA允许多个线程同时执行，这可能会导致数据竞争和不一致性，原子操作通过确保某个操作在执行过程中不会被其他线程打断，从而解决这些问题。

原子操作的特点

1. 原子性：

   • 原子操作是不可分割的操作。在执行原子操作时，其他线程无法对该数据进行读写，确保了数据的一致性。

2. 支持的数据类型：

   • CUDA提供了对基本数据类型（如整数、浮点数）的原子操作支持，包括 int, unsigned int, float 等。

3. 常用的原子操作：

   • atomicAdd(): 对变量进行加法操作。
   • atomicSub(): 对变量进行减法操作。
   • atomicExch(): 交换两个变量的值。
   • atomicMin(): 更新变量为最小值。
   • atomicMax(): 更新变量为最大值。
   • atomicCAS(): 原子比较并交换。

#include <stdio.h>
#include <cuda.h>

__global__ void vectorAdd(const float *A, const float *B, float *C, int N) {
    // 计算当前线程的全局索引
    int i = bl