Use Eigen blas for custom sdpa (#6197)
Summary: Pull Request resolved: https://github.com/pytorch/executorch/pull/6197 OpenBlas's implementation is not thread safe. Thus when used within parallel_for, it produces incorrect output. This has been documented in a few places like here https://github.com/OpenMathLib/OpenBLAS/issues/1441 and https://github.com/OpenMathLib/OpenBLAS/issues/2543. I tried few options to disable openblas's multithreading but none of them seemed to work. It is possible that upstream openblas has fixed this. Will validate this by pulling in latest, but in the meanwhile using eigen_blas to unblock. ghstack-source-id: 247950073 exported-using-ghexport Reviewed By: guangy10, nausicaasnow Differential Revision: D64334733 fbshipit-source-id: a7117233584860c2910f45bfaa8dd8fa41f9f28d
K
Kimish Patel committed
95e7aa3a6412c242758003b905638f4add01ad86
Parent: 4745070
Committed by Facebook GitHub Bot <facebook-github-bot@users.noreply.github.com>
on 10/15/2024, 4:46:11 AM