trunk/a7a0e15894ee2c285d5065c3672291a19b0a8091: [MPS] Add naive triangular_solve for complex numbers (#195861)
- PyTorch 近 90 天出现 616 次
- PyTorch 近 90 天第 610 次发版
- 上一次：同一天稍早 · trunk/4a053096707d3eb77d814636c1f1f24fe5df2a33: [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…
摘要按规则整理自下方来源原文