记录一下创建cuda静态库过程,使用创建cuda工程时自动生成的项目代码作为示例。

一、封装静态库

1. 新建项目

新建cuda项目,输入项目名称。
在这里插入图片描述

2. 修改自动生成的kernel.cu文件

删掉或注释main函数。

3. 生成头文件

选择当前项目新建头文件,文件名自己命名。我是为global函数和device函数创建各自的头文件(.h和.cuh都可以)以及对应的实现.cu文件。
在这里插入图片描述

4. 修改项目属性

修改项目的配置类型为静态库。
在这里插入图片描述

5. 运行生成静态库

如果出现编译报错:MSB3721,返回代码255。在工程属性的CUDA C/C+±>Generate Relocatable Device Code下选择***是 (-rdc=true)***

二、调用静态库

1. 新建cuda项目

操作同新建静态库时一样。

2. 修改文件

修改kernel.cu文件,只保留main函数,去掉其他函数。(Kernel.cu可以将后缀名改为cpp,一样可以运行。)

3. 配置项目属性

注意debug和release模式以及平台位数。
加入静态库所在路径:
在这里插入图片描述
加入静态库名:
在这里插入图片描述
加入静态库头文件:
在项目属性中配置或者将头文件复制到当前项目下。
在这里插入图片描述

完成

Logo

欢迎来到由智源人工智能研究院发起的Triton中文社区,这里是一个汇聚了AI开发者、数据科学家、机器学习爱好者以及业界专家的活力平台。我们致力于成为业内领先的Triton技术交流与应用分享的殿堂,为推动人工智能技术的普及与深化应用贡献力量。

更多推荐