在使用c++进行扩展算子开发的时候,有一个方便的代码联想是非常舒服的一件事情,下面让我们开始搭建这个平台吧
1、先完成paddlepaddle的安装和vscode的安装(这部分不再赘述)。 2、完成vscode的c/c++扩展的安装。 3、在python里面用paddle.__path__找到paddle/extension.h的位置(在vscode中导入头文件需要用到这个地址)。 4、在cmd中输入set cuda找到cuda.h的位置(在vscode中导入头文件需要用到这个地址)。 5、在vscode中用“ctrl+shift+p”打开后,找到这个json文件。 6、在“includePath”中写入在步骤3中找到的这个地址(后面一般为paddle/include/**),保存即可。
paddle.__path__
paddle/extension.h
set cuda
cuda.h
小作者会将在AI Studio上的划桨记录分享到公众号上,而且公众号不定期更新深度学习相关内容,有关于深度学习方面好玩的应用,有论文解读复现,有精读深度学习著作等,小作者还会将在AI Studio公开的项目的背后故事和思考点在公众号同步更新,欢迎关注鸭~