关于ubuntu cuda工程

参考:https://www.cnblogs.com/viviman/archive/2012/11/15/2775093.html
工程流程如下:
1 新建cuda项目
2 新建main.cpp作为程序入口,用的是c++编译器。
3 新建业务级别.h和.cpp,业务封装在.h文件的class中。
比如:
class A{
              public:
              fun();
}
4 新建计算级别的.h或.cuh和.cu,其实这里头文件可以不建立,而直接在.cu文件中定义,然后声明成extern,但是为了统一函数声明,都用了头文件。
比如:
calc.cuh
#ifndef _CU_H_
#define _CU_H_

calcfun();

#endif

calc.cu
#include "calc.cuh"
__device__ kernelsubcalc(){}
__global__ kernelcalc(){kernelsubcalc();}
calcfun()
{
 kernelcalc<<<,>>>();
}
注意一个原则:与设备有关的关键字比如global和<<<>>>符号必须只能出现在.cu文件中。
5 封装设备操作函数
cuda的设备操作很多,为了编程的方便,通常封装自己常用的操作。
比如我封装的一个设备内存操作函数:
#ifndef _GUPMEM_
#define _GUPMEM_

#include "common.h"

// 分配设备内存
template
int GPUMalloc(T **t, size_t n)
{
 HANDLE_ERROR( cudaMalloc( (void**)&*t, n * sizeof(T) ) );
 printf("gpu malloc right!\n");
 return 0;
}

// 释放设备内存
template
class GPUGuard
{
public:
            GPUGuard(T *t):m_t(t){}
            ~GPUGuard(){
            cudaFree(m_t);
            printf("cuda mem free\n");

 }
private:
 T *m_t;
};


// 内存在设备与主机之间的拷贝
enum CopyModel{
 HOST2DEV, // 主机->设备
 DEV2HOST // 设备->主机
};
template
int MemCopy(T **dev_t, T **host_t, size_t n, CopyModel model)
{
 if(model == HOST2DEV)
 {
  printf("host to dev\n");
  HANDLE_ERROR( cudaMemcpy( *dev_t, *host_t, n * sizeof(T), cudaMemcpyHostToDevice ) );
 }
 else if (model == DEV2HOST)
 {
  printf("dev to host\n");
  HANDLE_ERROR( cudaMemcpy( *host_t, *dev_t, n * sizeof(T), cudaMemcpyDeviceToHost ) );
 }
 else
 {
  printf("copy model error!\n");
  return -1;
 }
 return 0;
};


#endif

设备操作函数的使用就很简单:
int *dev_a;
GPUGuardg1(dev_a);

if(GPUMalloc(&dev_a, N) < 0)
 return -1;

if(MemCopy(&dev_a, &a, N, HOST2DEV) < 0)
 return -1;


总结:
我需要的cuda工程文件有.cpp .h .cu,并行计算放在cu中,其他业务放在.cpp中。

  • 1
    点赞
  • 1
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值