← 返回事件
持续讨论AI发版

trunk/1782e8ec673e736ffda3414fd921f3c1b1b62002: [MPS] Fix linear backward SIGABRT with a 1-D weight (#187989)

图:PyTorch Releases

发生了什么

Fixes #187988 What linear() accepts a 1-D weight (shape [in_features] ): it contracts and drops the last input dimension. mps_linear (forward) already handles this by unsqueezing the weight to [1, in_features] . The backward pass did not mirror that, so it lowered grad_input and grad_weight to invalid rank-1 × rank-1 matmuls and aborted the process during MPSGraph verification: 'mps.matmul' op contracting dimensions…

摘要按规则整理自下方来源原文

为什么在扩散

来源