Commit 95e7aa3
Use Eigen blas for custom sdpa (#6197)
Summary:
Pull Request resolved: #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 OpenMathLib/OpenBLAS#1441 and
OpenMathLib/OpenBLAS#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: a7117233584860c2910f45bfaa8dd8fa41f9f28d1 parent 4745070 commit 95e7aa3
1 file changed
+1
-1
lines changed| Original file line number | Diff line number | Diff line change | |
|---|---|---|---|
| |||
144 | 144 | | |
145 | 145 | | |
146 | 146 | | |
147 | | - | |
| 147 | + | |
148 | 148 | | |
149 | 149 | | |
150 | 150 | | |
| |||
0 commit comments