trunk/2053ec330a7e9bc250fb9dd22d7c203ddd109410: [MPS] Add complex64 support to linalg.lu_solve (#195369)
- PyTorch: 293 events in the last 90 days
- PyTorch: 290th Release in the last 90 days
- Previous: earlier the same day · trunk/7bdc7ec2a3cd822a2cb22f4af0032c66496387f6: [triton hash update] update the pinned triton hash (#195236)
What happened
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…
Summary assembled by rule from the sources below