当你要写一个CUDA函数计算后返回多个参数时
2016-06-06 23:49
323 查看
一点建议
如果你打算利用CUDA进行一番计算后获得几个不同的返回值时也许你会这样:
cudaMalloc;
…….*n
cudaMemcpy(…, cudaMemcpyHostToDevice);
…….*n
kernelCal<<<>>>();
之后,调用函数并打算用数个
cudaMymcpy(…, cudaMemcpyDeviceToHost);
…….*k
来返回K个计算结果时,请注意:
当计算序列比较长的时候,这样来获得多个返回值时有时会发生某一类的返回莫名有误的情况(如开头一定长度全0)。
其实解决方案比较简单——保持一个函数计算完后只进行一次返回。体现在代码上的话,就是只进行一次cudaMymcpy(…, cudaMemcpyDeviceToHost);
那么如何是好呢?个人建议利用结构体,将所需的各项返回写到一个结构体myStruct中,然后无论是cudaMalloc, cudaMymcpy, kernelCal都以myStruct作为参数(或者需要多个返回时,用*myStruct就像普通数组一样)。
需要感谢CUDA 6.0(貌似是这一版本),使得结构体的应用变得更加简洁,不需要我等再去考虑什么结构体内类型对齐=_=
转载请标明出处:http://blog.csdn.net/u011185952/article/details/51599137
相关文章推荐
- CUDA搭建
- 一个程序来比较cuda/c在GPU/CPU的运行效率
- cuda之二维数组的高效内存管理(cudaMallocPitch/cudaMemcpy2D)
- Some Notes of Caffe Installation
- 用python做GPU计算(1)——安装以及配置
- GPU(CUDA)学习日记(三)------ CUDA基本架构介绍以及编程入门!
- vs2008--CUDA环境配置
- window 下 pycuda 安装运行
- CUDA技术帖: 3D texture linear data VS 3D cudaArray
- 一个你可能不知道的cuda细节
- py-faster-rcnn训练笔记(ubuntu14.04+cuda7.5+cuDNNv3+Python2.7)
- NVIDIA CUDA统一计算设备架构编程手册(1)
- CUDA中文教程01之心得体会
- CUDA中文教程03之心得体会
- 【GPU开发笔记】二:CUDA初探——查询设备
- cuda一些小问题(不断更新)
- 一种简单的GPU三维图像分割算法
- Linux系统下CUDA中的原子操作
- CUDA --1 基础知识
- ubuntu 12.04 13.10 安装 cuda 5.5