SIGN IN SIGN UP

Use Eigen blas for custom sdpa (#6229)

Summary:
Pull Request resolved: https://github.com/pytorch/executorch/pull/6229

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: 248095356
exported-using-ghexport

Reviewed By: cccclai

Differential Revision: D64398816

fbshipit-source-id: 46f2d10855007e35e5d6fb925e494f12330978fa
K
Kimish Patel committed
97a19658f2fb2f5704aab1c86a9e3ec5ca3aac4b
Parent: 96ecf78
Committed by Facebook GitHub Bot <facebook-github-bot@users.noreply.github.com> on 10/15/2024, 10:32:07 PM