CUDA内核函数的连续执行

最新推荐文章于 2025-04-01 17:48:20 发布

原创

最新推荐文章于 2025-04-01 17:48:20 发布 · 1.7k 阅读

2 ·

CC 4.0 BY-SA版权

文章标签：

#c++ #cuda #内核

以下是两个连续的CUDA核函数衔接的一种思路：

要完成的功能：1. 向量的计算computer（暂时以两向量求和为例）；2. 对结果向量求和SUM。

思路1：写1个计算的内核函数，中间结果保留，求和函数调用cublas。

#include "cuda_runtime.h"
#include "device_launch_parameters.h"
#include "cublas_v2.h"

#include <stdio.h>
#include <stdlib.h>

#pragma comment(lib, "cublas.lib")

cudaError_t addWithCuda(const int *a, const int *b, unsigned int size, float& sum);

__global__ void addKernel(const int *a, const int *b, float *c)
{
    int i = threadIdx.x;
    c[i] = a[i] + b[i];
}

int main()
{
    const int arraySize = 5;
    const int a[arraySize] = { 1, 2, 3, 4, 5 };
    const int b[arraySize] = { 10, 20, 30, 40, 50 };

    // Add vectors in parallel.
	float sum = 0.0;
    cudaError_t cudaStatus = addWithCuda(a, b, arraySize, sum);
    if (cudaStatus != cudaSuccess) {