cudaHeader.cu 648 Bytes
#include"cudaHeader.h"
#include<iostream>

#include "cuda_runtime.h"
#include <cuda.h>

//核函数,计算a+b
__global__ void add(int a,int b,int *c)
{
    //保存a+b的计算结果
    *c=a+b;
}


//cuda测试函数的实现
void cudaTest()
{
    int c = 0;
    //在gpu上开辟一个相同的内存
    int *deviceC;
    cudaMalloc((void**)&deviceC,sizeof(int));
    //调用核函数
    add<<<1,1>>>(3,7,deviceC);
    //把计算结果复制到cpu上
    cudaMemcpy(&c,deviceC,sizeof(int),cudaMemcpyDeviceToHost);
    //展示计算结果
    std::cout<<c<<std::endl;
    //释放内存
    cudaFree(deviceC);
}