C++ 使用 __global__
声明说明符定义内核,并使用新的 <<<...>>>
执行配置语法指定内核调用的 CUDA 线程数(请参阅 C++ 语言扩展)。 每个执行内核的线程都有一个唯一的线程 ID,可以通过内置变量在内核中访问。
示例代码使用内置变量 threadIdx
将两个大小为 N 的向量 A 和 B 相加,并将结果存储到向量 C 中:
#include <cuda_runtime_api.h>
#include <iostream>
#define RANDOM(x) (rand() % x)
#define MAX 10
// single block multiple threads
__global__ void vector_add_gpu_2(int *d_a, int *d_b, int *d_c, int n){
int tid = threadIdx.x;
const int t_n = blockDim.x;
while(tid < n){
d_c[tid] = d_a[tid] + d_b[tid];
tid+=t_n;
}
}
int main(){
/***向量相加的实现***/
int n = 5;
int *a = (int *)malloc(sizeof(int)*n);
int *b = (int *)malloc(sizeof(int)*n);
int *c = (int *)malloc(sizeof(int)*n)