v2.14.1-rc2: [MPS] Fix Jacobi SVD convergence for small columns (#199086)
- PyTorch: 1673 events in the last 90 days
- PyTorch: 1654th Release in the last 90 days
- Previous: earlier the same day · trunk/0993a6c66933eecdcfddbdc79a80f702067bc367: [Pipeline] Defer FSDP gradient reduction waits (#196725)
What happened
[MPS] Fix Jacobi SVD convergence for small columns ( #199063 ) This PR addresses the remaining ill-conditioned case in #196112 . Codex prepared the following description for review: The MPS Jacobi SVD skips rotations when the product of two column norms falls below float32 epsilon, even when the columns are strongly correlated. For the ill-conditioned input in #196112 this produces non-orthogonal U and inflated smal…
Summary assembled by rule from the sources below