大佬,咋样了,可以解决吗
抱歉,这个模型用了较多的CUDA自定义算子和一些扩展算子,MLU上没办法解决。
你从哪里扒的,可否贴个链接?
加上print这行报错就是上面说的那个,删掉最后一行的print(res),报错就变成Segmentation fault (core dumped)了,很神奇,我就是在GitHub上扒下来的
pred如果是在MLU上推理得到的结果,默认的存储位置还在MLU上,需要先pred.cpu(),然后才能print,否则会出错。但是你这里描述的是去掉print会出错,就不知道啥问题了。你这个模型,在github上有参考实现吗?
这是遇到了MLU不支持的算子,导致shape推断错误导致的,除非自定义实现这些算子或者替换掉这些算子
大佬,在帮我看看呗
请登录后评论