cuda中全局变量的使用方法

最新推荐文章于 2024-06-09 20:03:08 发布

CCC_bi

最新推荐文章于 2024-06-09 20:03:08 发布

阅读量1.2k

点赞数 1

分类专栏： C++学习文章标签： c++ c语言开发语言 cuda

本文链接：https://blog.csdn.net/Dianhui_Bi/article/details/124610430

版权

C++学习专栏收录该内容

12 篇文章 0 订阅

订阅专栏

本文介绍了如何使用CUDA在C++中进行GPU编程，展示了如何在CUDA中创建全局变量，使用__global__函数进行数据操作，并通过Memcpy进行设备间数据传输。重点讲解了data_add和arr_add两个GPU kernel的实现与同步机制。

摘要由CSDN通过智能技术生成


#include <stdio.h>
#include <iostream>
#include <stdlib.h>
#include <string>
#include <math.h>
#include <time.h>
#include <windows.h>
#include <fstream>

//CUDA RunTime API
#include <cuda_runtime.h>
#include <device_launch_parameters.h>
using namespace std;
__device__ float devdata;
__device__ float a[10];

__global__ void data_add();
__global__ void arr_add();

int main()
{
    float value = 1.0;
    //将变量数据传输给GPU的全局变量
    cudaMemcpyToSymbol(devdata, &value, sizeof(float));
    data_add << <1, 1 >> > ();
    //将GPU的全局变量传输给变量
    cudaMemcpyFromSymbol(&value, devdata, sizeof(float));
    printf("%f\n", value);


    float ah[10];
    arr_add << <1, 10 >> > ();

    //将GPU中的全局数组变量数据传输给CPU
    cudaMemcpyFromSymbol(ah, a, sizeof(ah));

    for (int i = 0; i < 10; i++)
    {
        printf("%f ", ah[i]);
    }
    printf("\n");
    return 0;
}

__global__ void data_add()
{
    const int tid = threadIdx.x;
    devdata++;
}

__global__ void arr_add()
{
    const int tid = threadIdx.x;
    if(tid<10)
    a[tid] = tid;

    __syncthreads();
}