×

签到

分享到微信

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

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

cnml和Op错误 已解决 誓约2021-12-13 17:15:07 回复 17 查看 技术答疑 使用求助 经验交流
cnml和Op错误
分享到:


import torch
import torch_mlu.core.mlu_model as ct

input1_cpu = torch.ones(2,3)
input2_cpu = 2*torch.ones(2,3)
input1_mlu = input1_cpu.to(ct.mlu_device())
input2_mlu = input2_cpu.to(ct.mlu_device())
input3_mlu = input1_mlu + input2_mlu
input1_mlu = torch.cat((input2_mlu,input1_mlu),0)

print(input1_mlu.cpu())

为什么写这样一个简单的程序,调试的时候会出现多个算子无法加载到mlu上的问题,求解

image.png

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