trunk/2053ec330a7e9bc250fb9dd22d7c203ddd109410: [MPS] Add complex64 support to linalg.lu_solve (#195369)
- PyTorch 近 90 天出现 294 次
- PyTorch 近 90 天第 291 次发版
- 上一次：同一天稍早 · trunk/7bdc7ec2a3cd822a2cb22f4af0032c66496387f6: [triton hash update] update the pinned triton hash (#195236)
发生了什么
Extends the native Metal lu_solve to complex64, unblocking complex64 backward for det , slogdet , logdet , solve and solve_ex on MPS The lu_solve schedule is unchanged; only the per-step kernels are made dtype-generic: trsmDiagSolveLU and luApplyPivotsRHS are templated on the scalar type and instantiated for float and float2 . The float path is kept byte-identical (dcol register staging + fma , identity-padded diago…
摘要按规则整理自下方来源原文