trunk/620cac79649a8e9aedb624229e1dfe54e180ced5: [MPS] Prescale Jacobi SVD by the largest entry (#200776)

发生了什么

svd_jacobi forms the Gram terms as fp32 sums of squares and multiplies them, which underflows for entries below 1e-9 and overflows above 1e9. Since this was added by #195950 I'm attaching milestone for 2.15. This PR prescales it by a power of two, similar to what LAPACK does ( https://github.com/Reference-LAPACK/lapack/blob/v3.12.1/SRC/sgesdd.f#L597-L610 ). import torch A = torch . randn ( 8 , 8 ) for scale in ( 1e-…

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

来源