pytorch
2017b404 - Fix SVD error code handling for OpenBLAS 0.3.15+ and MKL 2022+ (#68812)

Commit
2 years ago
Fix SVD error code handling for OpenBLAS 0.3.15+ and MKL 2022+ (#68812) Summary: Fixes https://github.com/pytorch/pytorch/issues/67693. Reference LAPACK (used in OpenBLAS) changed info error code for svd when inputs contain non-finite numbers. In PyTorch, we raise an internal assert error for negative `info` error codes because usually, it would indicate wrong implementation. However, this is not the case with SVD now in newer versions of LAPACK. MKL (tried 2021.4.0) still gives a positive error code for this kind of input. This change aligns with the OpenBLAS and MKL behavior in our code. **UPDATE:** MKL 2022 has uses the latest reference LAPACK behavior and returns the same `info` as OpenBLAS 0.3.15+ This PR fixes https://github.com/pytorch/pytorch/issues/71645 that is due to the updated MKL version in CI. Pull Request resolved: https://github.com/pytorch/pytorch/pull/68812 Reviewed By: mrshenli Differential Revision: D33844257 Pulled By: ngimel fbshipit-source-id: fd1c86e37e405b330633d039f49dce466391b66e (cherry picked from commit c00a9bdeb0dc8d49317b93d19b7b938a4cfb7a38)
Author
Committer
Parents
Loading