关于ReduceSum结果出错的问题
收藏回复举报
关于ReduceSum结果出错的问题
t('forum.solved') 已解决
发表于2024-04-29 16:05:57
0 查看

对两个向量乘之后进行ReduceSum:

cke_742.png

发现结果出错,

cke_7196.png

将规约使用的worklocal替换为另一个不同的tensor(代码注释部分),会得到更离谱的结果:

cke_12939.png

详细代码见附件。

因为代码过长,可只看compute函数部分:代码413行-460行

首先会打印两个输入向量,各自打印的每一行是将要计算的每一行数据,可以只看打印数据的第一行

cke_32990.png

后只针对第一次循环进行打印,

第一次循环会将两个输入的第一行进行Mul,打印查看Mul数据正确。然后进行ReduceSum,打印查看发现数据错误。

bash run.sh ->1.txt -r cpu -v npu Ascend910ProB即可复现,建议输出重定向到文件,不然太长了🤦‍♂️

我要发帖子