系统环境
win10 64位
VS2017
CUDA 10
Qt 5.12.0
一,封装为dll文件
1.VS创建"动态编译库"工程
2.添加CUDA文件
3.输入需要被封装的测试代码
.h文件
#pragma once
extern "C" __declspec(dllexport) int vectorAdd(int c[], int a[], int b[], int size);
.cu文件
注意这里一定要包含上面的.h文件(#include “DLL1.h”)否则只能生成dll文件没有lib文件
#include "cuda_runtime.h"
#include "device_launch_parameters.h"
#include "DLL1.h"
//CUDA核函数
__global__ void addKernel(int *c, const int *a, const int *b)
{
int i = threadIdx.x;
c[i] = a[i] + b[i];
}
//向量相加
int vectorAdd(int c[], int a[], int b[], int size)
{
int result = -1;
int *dev_a = 0;
int *dev_b = 0;
int *dev_c = 0;
cudaError_t cudaStatus;
// 选择用于运行的GPU
cudaStatus = cudaSetDevice(0);
if (cudaStatus != cudaSuccess) {
result = 1;
goto Error;
}
// 在GPU中为变量dev_a、dev_b、dev_c分配内存空间.
cudaStatus = cudaMalloc((void**)&dev_c, size * sizeof(int));
if (cudaStatus != cudaSuccess) {
result = 2;
goto Error;
}
cudaStatus = cudaMalloc((void**)&dev_a, size * sizeof(int));
if (cudaStatus != cudaSuccess) {
result = 3;
goto Error;
}
cudaStatus = cudaMalloc((void**)&dev_b, size * sizeof(int));
if (cudaStatus != cudaSuccess) {
result = 4;
goto Error;
}
// 从主机内存复制数据到GPU内存中.
cudaStatus = cudaMemcpy(dev_a, a, size * sizeof(int), cudaMemcpyHostToDevice);
if (cudaStatus != cudaSuccess) {
result = 5;
goto Error;
}
cudaStatus = cudaMemcpy(dev_b, b, size * sizeof(int), cudaMemcpyHostToDevice);
if (cudaStatus != cudaSuccess) {
result = 6;
goto Error;
}
// 启动GPU内核函数
addKernel << <1, size >> > (dev_c, dev_a, dev_b);
// 采用cudaDeviceSynchronize等待GPU内核函数执行完成并且返回遇到的任何错误信息
cudaStatus = cudaDeviceSynchronize();
if (cudaStatus != cudaSuccess) {
result = 7;
goto Error;
}
// 从GPU内存中复制数据到主机内存中
cudaStatus = cudaMemcpy(c, dev_c, size * sizeof(int), cudaMemcpyDeviceToHost);
if (cudaStatus != cudaSuccess) {
result = 8;
goto Error;
}
result = 0;
// 重置CUDA设备,在退出之前必须调用cudaDeviceReset
cudaStatus = cudaDeviceReset();
if (cudaStatus != cudaSuccess) {
return 9;
}
Error:
//释放设备中变量所占内存
cudaFree(dev_c);
cudaFree(dev_a);
cudaFree(dev_b);
return result;
}
4.配置编译环境
cudart.lib是必须要添加的依赖库,如果在CUDA程序中用到了cufft.h(用于傅里叶变换计算)或者cusolverDn.h(用于计算常用的矩阵分解,包括特征值分解,SVD分解等)就需要在这里再添加cufft.lib和cusolver.lib,否则会出现 MSB3721错误 提示
5.输出结果显示
注意这里使用64位编译器,得到64位的dll文件,使用dll文件时需要64位编译器链接。如果使用32位编译器得到32位dll文件,需要在使用dll时也对应使用32位编译器进行链接
二、dll文件VS工程调用
上面章节一共产生三个我们用到的文件:
DLL1.h
DLL1.dll
DLL1.lib
将这三个文件复制到所需使用到的工程的文件目录中,下面以简单的控制台应用程序为例:
添加头文件DLL.h
添加依赖库DLL.lib
运行
三、Qt调用
如果是通过VS 64bit编译器生成的DLL,请务必保证使用Qt的编译环境也为64bit:
-
根据当前模式(Debug/Release)复制xxx.lib,xxx.dll和xxx.h文件到Debug或Release文件夹下:
-
引入库
会在Pro文件中自动添加以下内容:
-
引入头文件
会出现:
即可头文件包含的外部函数。
VS项目内容:
整个工程已经上传到Github,请下载运行:https://github.com/ZhangPHinNEU/Demo_DLL