CUDA 是一个用于并行计算的开源平台和编程模型,它允许开发者利用 NVIDIA GPU 进行高性能计算。以下是一些基础的 CUDA 示例。
核心概念
- 线程:CUDA 程序由许多线程组成,这些线程可以在 GPU 上并行执行。
- 网格:线程被组织成网格,网格中的线程可以相互协作。
- 内存:CUDA 有三种内存类型:全局内存、共享内存和常量内存。
示例代码
以下是一个简单的 CUDA 示例,它计算两个一维数组中元素的和。
__global__ void add(int *a, int *b, int *c) {
int index = threadIdx.x;
c[index] = a[index] + b[index];
}
学习资源
更多关于 CUDA 的学习资源,可以访问 CUDA 官方文档。
图片示例
以下是 CUDA 核心概念之一的示例图片: