如何解决cupy 中点积的意外结果
以下代码产生了意外大的结果：
import cupy as cp,numpy as np
N=500
aa1=cp.random.randn(N,N)
aa2= cp.random.randn(N,N)
aa3=cp.random.randn(N,N)
cc=aa1.dot(aa2).dot(aa3)
cp.dot(aa1,aa2,out=aa2)
cp.dot(aa2,aa3,out=aa2)
print(cp.max(cp.abs(cc-aa2)))
%output: 40070455.39606033
此外，当我多次运行它时，它会打印出不同的（但总是很大的）数字。我想问题出在 cp.dot 部分，因此 aa2 矩阵不等于三个原始矩阵的乘积。
有趣的是，当我将问题大小从 N=500 减少到 N=100 时，它总是按照我的预期打印 0。据我了解，并发执行和竞争条件没有问题，因为每个操作都是由 NULL 流按顺序执行的（默认流在开始下一个矩阵乘法之前等待所有指令完成）。
Ubuntu 18、GPU Titan RTX、cuda 工具包 11.0、cupy 9.2.0
版权声明：本文内容由互联网用户自发贡献，该文观点与技术仅代表作者本人。本站仅提供信息存储空间服务，不拥有所有权，不承担相关法律责任。如发现本站有涉嫌侵权/违法违规的内容， 请发送邮件至 dio@foxmail.com 举报，一经查实，本站将立刻删除。
