×

签到

分享到微信

打开微信,使用扫一扫进入页面后,点击右上角菜单,

点击“发送给朋友”或“分享到朋友圈”完成分享

使用CNplugin扩展pytorch算子时调用的问题 已解决 sunxiaofeng2021-09-02 16:28:19 回复 13 查看 使用求助
使用CNplugin扩展pytorch算子时调用的问题
分享到:

在使用CNplugin添加pytorch的算子时,编译好cnplugin.so文件后使用cpp文件调用函数没有问题,但是将函数调用注册到pytorch框架后调用产生了问题,可以看到printf的标志进到bangc算子中,并且数据类型和数据大小的尺寸传参也是正确的。

image.png

Kernel函数的实现如下,image.png

如果我想单步调试查看进入到该cpp文件的话要怎么调试,因为我单步调试.py文件并不会进入到该cpp文件内部

版权所有 © 2024 寒武纪 Cambricon.com 备案/许可证号:京ICP备17003415号-1
关闭