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

(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编译生成库文件,供主函数调用。

你可能感兴趣的:(CUDA学习,c++,cuda,边缘计算,人工智能)