华为计算微信公众号
昇腾AI开发者公众号
华为计算微博
华为计算今日头条
使用mindspore的2.3.1版本调用ascendc算子时,该算子是原地计算的算子,算子定义如下:
算子内部更新向量x,在用c++进行aclnn调用时,可以在计算完毕后将device侧的x内存拷贝到host侧输出
但是使用mindspore进行aclnn调用时,CustomRegOP注册该算子时,无论添不添加output注册,都无法获取原地输出的值
请问该如何将原地计算的值从device侧拷贝到host侧
我要发帖子
使用mindspore的2.3.1版本调用ascendc算子时,该算子是原地计算的算子,算子定义如下:
算子内部更新向量x,在用c++进行aclnn调用时,可以在计算完毕后将device侧的x内存拷贝到host侧输出
但是使用mindspore进行aclnn调用时,CustomRegOP注册该算子时,无论添不添加output注册,都无法获取原地输出的值
请问该如何将原地计算的值从device侧拷贝到host侧