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

Commit
3 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: osalpekar Differential Revision: D32626563 Pulled By: ngimel fbshipit-source-id: 09042f07cdc9c24ce1fa5cd6f4483340c7b5b06c (cherry picked from commit aadf50731945ac626936956e229cf2056a291741)
Author
Committer
Parents
  • aten/src/ATen/native
    • File
      LinearAlgebraUtils.h
  • test
    • File
      test_linalg.py