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

发现结果出错,

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

详细代码见附件。
因为代码过长,可只看compute函数部分:代码413行-460行
首先会打印两个输入向量,各自打印的每一行是将要计算的每一行数据,可以只看打印数据的第一行

后只针对第一次循环进行打印,
第一次循环会将两个输入的第一行进行Mul,打印查看Mul数据正确。然后进行ReduceSum,打印查看发现数据错误。
bash run.sh ->1.txt -r cpu -v npu Ascend910ProB即可复现,建议输出重定向到文件,不然太长了🤦♂️
对两个向量乘之后进行ReduceSum:
发现结果出错,
将规约使用的worklocal替换为另一个不同的tensor(代码注释部分),会得到更离谱的结果:
详细代码见附件。
因为代码过长,可只看compute函数部分:代码413行-460行
首先会打印两个输入向量,各自打印的每一行是将要计算的每一行数据,可以只看打印数据的第一行
后只针对第一次循环进行打印,
第一次循环会将两个输入的第一行进行Mul,打印查看Mul数据正确。然后进行ReduceSum,打印查看发现数据错误。
bash run.sh ->1.txt -r cpu -v npu Ascend910ProB即可复现,建议输出重定向到文件,不然太长了🤦♂️