CUDA系列二：向量的相加

最新推荐文章于 2024-05-28 22:11:33 发布

原创

最新推荐文章于 2024-05-28 22:11:33 发布 · 1.9k 阅读

4 ·

CC 4.0 BY-SA版权

本文主要介绍下cuda的向量加法的实现，该代码只是为了熟悉概念和初步上手cuda，过程也比较简单。cuda c上手其实还是挺简单的，但是要精通还是需要通过大量实践才能达到的，有兴趣的同学需要多加练习。本文的向量相加，由简逐渐深入。

初步实现只使用了一个block：

#include <iostream>


#define N 10

static void HandleError( cudaError_t err,
                         const char *file,
                         int line ) {
    if (err != cudaSuccess) {
        printf( "%s in %s at line %d\n", cudaGetErrorString( err ),
                file, line );
        exit( EXIT_FAILURE );
    }
}
#define HANDLE_ERROR( err ) (HandleError( err, __FILE__, __LINE__ ))

//核函数具体实现
__global__ void add(int *a,int *b,int *c)
{
	int tid = threadIdx.x; //获取数据索引位置，每个线程对应一个位置
	if(tid < N)
		c[tid] = a[tid] + b[tid];
}

int main(void)
{
	int a[N],b[N],c[N];
	int *dev_a,*dev_b,*dev_c;

    //设备端内存分配
	HANDLE_ERROR(cudaMalloc((void**)&dev_a,N * sizeof(int)));
	HANDLE_ERROR(cudaMalloc((void**)&dev_b,N * sizeof(int)));
	HANDLE_ERROR(cudaMalloc((void**)&dev_c,N * sizeo