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

viable/strict/1788168892: [MPS] Add complex64 support to linalg.lu_solve (#195369)

发生了什么

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…

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

为什么在扩散

来源