CUDA DLL 开发流程

结合了几篇文章的方法，前半部分主要参考 http://hi.baidu.com/yilinghl/item/a94b63959e89a3f028164739

需要补充的一点是，编写DLL工程时，需要添加链接器，方法如 http://www.cnblogs.com/chunshan/archive/2011/08/15/2139992.html 所示，在Linker | Input中添加 cudart.lib。

后半部分调用CUDA DLL主要参考：

http://blog.csdn.net/openhero/article/details/4059411

http://www.cnblogs.com/c1230v/articles/1401448.html

http://www.cppblog.com/amyvmiwei/archive/2008/01/02/40202.html

步骤总结起来就是：

1. 新建DLL工程，在工程上单击右键 -> 生成自定义，勾选CUDA；

2. 进入工程属性页，选择Linker | Input，添加 cudart.lib；

3. 新建CUDA文件，编写代码，Release 生成dll和lib；

4. 新建调用工程，编写代码，生成，将dll和lib复制到新工程exe同目录下，运行。

主要代码：

cuda.cu

#include "cuda_runtime.h"#include "device_launch_parameters.h"__global__ void vecAdd(int *a, int *b, int *c){int i = threadIdx.x;c[i] = a[i] + b[i];}bool cudaVecAdd(int *a, int *b, int *c, int n){int *dev_a;int *dev_b;int *dev_c;cudaError_t cudaStatus;cudaStatus = cudaSetDevice(0);if (cudaStatus != cudaSuccess) {return false;}cudaMalloc((void**)&dev_a, n * sizeof(int));cudaMemcpy(dev_a, a, n * sizeof(int), cudaMemcpyHostToDevice);cudaMalloc((void**)&dev_b, n * sizeof(int));cudaMemcpy(dev_b, b, n * sizeof(int), cudaMemcpyHostToDevice);cudaMalloc((void**)&dev_c, n * sizeof(int));vecAdd<<<1, n>>>(dev_a, dev_b, dev_c);cudaDeviceSynchronize();cudaMemcpy(c, dev_c, n * sizeof(int), cudaMemcpyDeviceToHost);cudaFree(dev_a);cudaFree(dev_b);cudaFree(dev_c);return true;}

CUDA_dll.cpp

bool cudaVecAdd(int *a, int *b, int *c, int n);extern "C" _declspec(dllexport) bool cuda_vec_add(int *a, int *b, int *c, int n){return cudaVecAdd(a, b, c, n);}

test.cpp

#include "stdafx.h"#include <Windows.h>typedef int (*HFUNC)(int *a, int *b, int *c, int n);int _tmain(int argc, _TCHAR* argv[]){int a[2] = {1, 2};int b[2] = {3, 4};int c[2] = {0, 0};HINSTANCE hDll = LoadLibrary(__T("CUDA_dll.dll"));if (hDll){HFUNC hFun = (HFUNC)GetProcAddress(hDll, "cuda_vec_add");if (hFun){hFun(a, b, c, 2);printf("%d %d\n", c[0], c[1]);}}return 0;}

上面是动态调用DLL，静态调用方法如下：

1. 将dll文件复制到DEBUG目录下，lib文件复制到工程文件下；

2. 将dll的头文件包含到工程目录下；

3. 直接在程序中调用即可。

CUDA_dll.h

#pragma comment(lib, "CUDA_dll.lib")extern "C" _declspec(dllimport) bool cuda_vec_add(int *a, int *b, int *c, int n);

test.cpp

#include "CUDA_dll.h"int _tmain(int argc, _TCHAR* argv[]){int a[2] = {1, 2};int b[2] = {3, 4};int c[2] = {0, 0};cuda_vec_add(a, b, c, 2);printf("%d %d\n", c[0], c[1]);return 0;}