pytorch
04a25060 - Fixed the size of the workspace array in functions calling MAGMA (#54009)

Commit
3 years ago
Fixed the size of the workspace array in functions calling MAGMA (#54009) Summary: The size of the workspace arrays should not be less than 1. This PR fixes lstsq calls to LAPACK and MAGMA. Also `max(1, ...)` guards were added to a few other functions (symeig, svd). ROCm testing is enabled for lstsq, pinv, pinverse. Fixes https://github.com/pytorch/pytorch/issues/53976 Pull Request resolved: https://github.com/pytorch/pytorch/pull/54009 Reviewed By: ejguan Differential Revision: D27155845 Pulled By: mruberry fbshipit-source-id: 04439bfa82a5bdbe2297a6d62b6e68ba1c30e4a2
Author
Parents
Loading