打开微信,使用扫一扫进入页面后,点击右上角菜单,
点击“发送给朋友”或“分享到朋友圈”完成分享
在使用CNplugin添加pytorch的算子时,编译好cnplugin.so文件后使用cpp文件调用函数没有问题,但是将函数调用注册到pytorch框架后调用产生了问题,可以看到printf的标志进到bangc算子中,并且数据类型和数据大小的尺寸传参也是正确的。
Kernel函数的实现如下,
如果我想单步调试查看进入到该cpp文件的话要怎么调试,因为我单步调试.py文件并不会进入到该cpp文件内部
热门帖子
精华帖子