outthinker

摘要： 以下面这个例子作为教程，实现功能是element-wise add； （pytorch中想调用cuda模块，还是用另外使用C编写接口脚本） 第一步：cuda编程的源文件和头文件 第二步：C编程的源文件和头文件（接口函数） 第三步：编译，先编译cuda模块，再编译接口函数模块（不能放在一起同时编译） 阅读全文