CUDA----.cpp文件和.cu文件应用区别

Aster
新星创作者: 人工智能技术领域
2022-09-09 14:57:17


(1).cu文件可以定义cuda操作的基本流程,包括:

数据由host向device的传输、device进行的运算、数据由device向host的传输,函数定义包括核函数(__global__)、设备函数(__device__)和主机函数(__host__),内存分配和线程块儿分配

    定义了cuda编程模型的文件必须以.cu文件作为文件名后缀,用nvcc编译形成库或文件

(2)定义头文件和库链接

.cu文件在采用nvcc编译时不需要包含CUDA头文件,.cpp用nvcc编译时需要明确增加头文件,用gcc编译时需要明确链接一些CUDA库

(3)调用关系

深度学习中,.cu文件一般用于定义模型可并行执行的layer,通过nvcc编译生成库文件,供主函数调用
 

...全文
613 回复 打赏 收藏 转发到动态 举报
写回复
用AI写文章
回复
切换为时间正序
请发表友善的回复…
发表回复

66

社区成员

发帖
与我相关
我的任务
社区描述
基于NVIDIA的DeepStream SDK开发技术交流社区。通过DeepStream可从USB/CSI/RTSP相机中获取流数据,用于快速构建具有AI能力的应用程序。
自动驾驶pytorch边缘计算 个人社区 陕西省·西安市
社区管理员
  • 鼾声鼾语
加入社区
  • 近7日
  • 近30日
  • 至今
社区公告
暂无公告

试试用AI创作助手写篇文章吧