开源pytorch Releases·原文 2026年10月1日

PyTorch 2.14.1修复MPS线性代数静默错误与CUDA 13.2二进制问题

该补丁版本修复torch.linalg.lstsq、torch.linalg.svd在MPS上对复杂批量欠定系统和秩亏输入的静默正确性问题,并将CUDA 13.2 Linux二进制升级到CUDA 13.2.2。

AI解读:PyTorch 2.14.1是一个修复性补丁版本,重点处理MPS后端的线性代数静默正确性问题和CUDA 13.2二进制的上游缺陷。

最实际的影响是:在苹果芯片上跑lstsq、svd这类线性代数运算的用户,之前可能拿到错误结果却不报错;这次更新后需要重新验证数值结果。

CUDA 13.2 Linux二进制被更新到CUDA 13.2.2,解决cuBLAS NVFP4矩阵乘法忽略张量级缩放、以及编译器线程重收敛失败导致寄存器值损坏这两个NVIDIA上游问题。

另有输入超过 8192 个元素时MPS报Metal pipeline-state错误、以及MPS复数输入触发内部断言两个回归被修复。

PyTorch发布 2.14.1 版本,按发布说明,这是一个用于修复回归和静默正确性问题的补丁版本。

发布说明列出了MPS后端多项线性代数计算的静默正确性修复:torch.linalg.lstsq在MPS上对复数批量欠定系统给出错误解(issue #196113,由PR #196128 修复);torch.linalg.svd在MPS上对秩亏和病态输入返回非正交的U以及不准确的小奇异值(issue #196112,由PR #196139 和 #199063 修复)。

CUDA 13.2 Linux二进制更新到CUDA 13.2.2(PR #196351)。发布说明引述NVIDIA的CUDA 13.2.2 release notes称,这次更新解决两个可能导致错误结果的关键问题:cuBLAS的cublasLtMatmul() 在NVFP4矩阵乘法中可能忽略张量级缩放(该问题自CUDA 13.2 Update 1引入);编译器线程重收敛失败可能在存在嵌套线程分歧的kernel中留下过期或损坏的寄存器值(该问题自CUDA 12.8起存在)。

回归修复方面:torch.linalg.svd、torch.linalg.svdvals和torch.linalg.lstsq在MPS上处理超过 8192 个元素的输入时会因Metal pipeline-state错误而失败(issue #195937,由PR #195949 和 #195950 修复);torch.svd(out=) 在MPS上处理复数输入时会触发内部断言(issue #195822,由PR #195872 修复)。

信息来源

pytorch Releases原始来源