trunk/666b41fe6f33957e2ade4bd9b82f35d67857867f: [MPS] Fix SVD normalization for small singular values (#200536)
- PyTorch 近 90 天出现 2206 次
- PyTorch 近 90 天第 2177 次发版
- 上一次:同一天稍早 · trunk/605260d09526b2fad33658892a8eab2750665041
发生了什么
Fix MPS SVD dropping valid singular vectors below an absolute epsilon, causing incorrect pseudoinverses and least-squares solutions. Reproducer: import torch for device in ("cpu", "mps"): eye = torch.eye(4, dtype=torch.float32, device=device) a = 1e-8 * eye print(device, "pinv:", torch.linalg.pinv(a).cpu()) print(device, "lstsq:", torch.linalg.lstsq(a, eye).solution.cpu()) output on main: cpu pinv: tensor([[10000000…
摘要按规则整理自下方来源原文