viable/strict/1788773762: [MPS] Add naive triangular_solve for complex numbers (#195861)
- PyTorch 近 90 天出现 622 次
- PyTorch 近 90 天第 616 次发版
- 上一次：同一天稍早 · viable/strict/1788771294: [dynamo] Preserve is_inference in after-AOT repros (#193565)
发生了什么
MPSMatrixSolveTriangular is float-only, so add custom Metal kernel for complex64. The impl gains a conjugate flag: triangular_solve/solve_triangular use plain transpose (transpose=True means A^T, verified against CPU), while _cholesky_solve_helper_mps now uses the adjoint (A^H) on its transposed solve, which is required for correct complex cholesky_solve and a no-op for real. Fix bug for non-contiguous out handling…
摘要按规则整理自下方来源原文