cuda核函数编程小结

文章目录

一、环境配置

几个注意事项先写在前面:

  1. 环境问题:一定要VS2019以上的!!!切记注意,目前VS2019以下已经不更新了,这里我就是用的2019,然后下面参考文档还是2015,反正过程没错,把VS版本改成2019即可;
  2. GPU及CUDA版本:我这边是NV RTX 3050Laptop 和 CUDA 11.8版本,建议CUDA版本别太低;
  3. 按照以下步骤在添加文件的时候没有NVIDIA的选项也没关系,创建后缀为".cu"的文件即可,其实是".c",核函数的格式是c语言的格式,而非C++,所以不能写在.cpp的文件中,不然会报错:error C2059: 语法错误:"<" 问题;

二、核函数代码书写格式

关于核函数的书写格式,看这篇比较易懂:https://www.cxyzjd.com/article/sinat_33718563/79156147

浅显的说,就是在cu文件中extern "C"声明C语言函数,然后在这个函数里面调用核函数,且核函数也放在cu文件中。第二步是在cpp文件中也用extern "C"声明cu中的C语言函数,然后用的时候直接用即可。

三、小结

cuda核函数编程环境配置和写法,其实不是很难,第一次接触可能觉得麻烦点,在核函数里面的多grid和多线程编程还是有点麻烦的,这个写的时候要特别注意,也可以啥都不用直接1,1在调用核函数的时候,然后可以参考下以下链接:

https://blog.csdn.net/zhanghaijun2013/article/details/107018029

https://blog.51cto.com/u_15316847/5949054

https://blog.csdn.net/weixin_41449637/article/details/120200121

https://cloud.tencent.com/developer/article/2065713

相关推荐
扫地的小何尚5 小时前
Isaac Lab 2.3深度解析:全身控制与增强遥操作如何重塑机器人学习
arm开发·人工智能·学习·自然语言处理·机器人·gpu·nvidia
ouliten7 小时前
cuda编程笔记(27)-- NVTX的使用
笔记·cuda
攻城狮7号8 天前
NVIDIA开源Audio2Face模型与SDK,数字人表情迎来“灵魂”时刻
人工智能·nvidia·开源模型·audio2face
ouliten8 天前
cuda编程笔记(24)-- Global Memory之间的拷贝
笔记·cuda
Jzzzzzzzzzzzzzz9 天前
NVIDIA-SMI has failed because it couldn‘t communicate with the NVIDIA driver.
ubuntu·nvidia·显卡驱动
小脑斧要动脑10 天前
【CUDA】【WIP】环境安装-wsl2下cuda安装
cuda
深度学习机器12 天前
解密vLLM:基于nano-vllm源码剖析其推理加速之道
pytorch·llm·nvidia
扫地的小何尚13 天前
NVIDIA Dynamo深度解析:如何优雅地解决LLM推理中的KV缓存瓶颈
开发语言·人工智能·深度学习·机器学习·缓存·llm·nvidia
扫地的小何尚13 天前
CUDA 13.0深度解析:统一ARM生态、UVM增强与GPU共享的革命
arm开发·人工智能·自然语言处理·gpu·nvidia·jetson·nvidia thro
扫地的小何尚15 天前
R²D²深度解析:NVIDIA三大神经网络突破如何变革机器人学习
神经网络·r语言·机器人·llm·gpu·nvidia