您好,我是用了寒武纪提供的pytorch镜像,安装了torch_mlu-0.15.0,怎么测试CONV3D卷积算子,我导入torch.ops.torch_mlu.MLUConv3d 时会报错。
您好,CONV3D算子在pytorch框架上已实现MLU270加速,请问您使用的是什么框架?
您好,我是用了寒武纪提供的pytorch镜像,安装了torch_mlu-0.15.0,怎么测试CONV3D卷积算子,我导入
torch.ops.torch_mlu.MLUConv3d 时会报错。
您好,CONV3D算子MLU270加速的源码是否开源?在哪里可以获取?
感谢回复,我看了源码,这个只写了在CPU上卷积的实现,并没有调用MLU加速,这样不管是在270,还是220效率应该都很低的,请问后续什么时候会有更新?
您好,该算子我们并未在MLU220上做过验证。
[图片]
您好,这里的3D卷积为什么不支持MLU220?[图片]
您好,您可以尝试自定义开发该算子。
您好,这里的3D卷积为什么不支持MLU220?
感谢回复,如果用BANG C自定义开发也不可行吗?
您好,目前并没有3D稀疏卷积算子的自定义开发,但目前提供了3D卷积算子的开发,您可以参考CNPlugin Datatypes Reference — CNPlugin Developer Guide 1.12.1 documentation (cambricon.com)
请登录后评论