Mindspore调用ascendc算子,算子为原地计算,但是mindspore无法获取原地计算的值
收藏回复举报
Mindspore调用ascendc算子,算子为原地计算,但是mindspore无法获取原地计算的值
t('forum.solved') 已解决
发表于2024-08-27 22:22:20
0 查看

使用mindspore的2.3.1版本调用ascendc算子时,该算子是原地计算的算子,算子定义如下:

cke_1427.png

算子内部更新向量x,在用c++进行aclnn调用时,可以在计算完毕后将device侧的x内存拷贝到host侧输出

但是使用mindspore进行aclnn调用时,CustomRegOP注册该算子时,无论添不添加output注册,都无法获取原地输出的值

cke_14436.png

请问该如何将原地计算的值从device侧拷贝到host侧

我要发帖子