×

签到

分享到微信

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

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

自定义nms后处理算子嵌入遇到的问题 已解决 zjunk2023-02-20 17:42:30 回复 7 查看 技术答疑 使用求助 经验交流
自定义nms后处理算子嵌入遇到的问题
分享到:

各位老师你们好,之前一直在做PPyoloe后处理算子嵌入(mlunmsthresh)的工作,经过了cnplguin编译动态库pytorch与算子集成(catch编译)

现在终于catch重新编译成功了如图:

image.png


接下来就是在pytorch端进行调用,可惜的是并不能调用这个接口如图所示:

image.png

后面经过排查

发现之前catch编译出来在torch/src/catch/torch_mlu/csrc/lib生成的4个动态库分别是libaten_mlu.so  libcatch_python.so libdistributed_mlu.so libjit_mlu.so。由于原始pytorch环境下动态库未更新,我们将/torch/venv3/pytorch/lib/python3.6/site-packages/torch_mlu/csrc/lib下的.so替换成为torch/src/catch/torch_mlu/csrc/lib重新编译后的,我们尝试导入import torch import torch_mlu会出现以下错误:

image.png

在更新了/torch/venv3/pytorch/lib/python3.6/site-packages/torch_mlu/csrc/lib的动态库以后他却报错找不到ComputepluginPpyoloenmsOpForward(该函数为自定义算子函数)。现在不知道到底哪儿出了问题希望老师们帮忙看看


 目前cnplugin catch编译都能通过



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