frozenleaves/pytorch
1
0
Fork 0
mirror of https://github.com/pytorch/pytorch.git synced 2025-10-20 21:14:14 +08:00
Code Issues Packages Projects Releases Wiki Activity

added pthread detection for gotoblas2/openblas, which is necessary on some distributions

Browse Source
This commit is contained in:
Ronan Collobert
2012-03-06 11:19:10 +01:00
parent ed9de62be4
commit d60f220e18
1 changed files with 27 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

27
lib/TH/cmake/FindBLAS.cmake
View File

@ -118,6 +118,20 @@ if((NOT BLAS_LIBRARIES)
endif(BLAS_LIBRARIES)
endif()
if((NOT BLAS_LIBRARIES)
AND ((NOT WITH_BLAS) OR (WITH_BLAS STREQUAL "open")))
check_fortran_libraries(
BLAS_LIBRARIES
BLAS
sgemm
""
"openblas;gfortran;pthread")
if(BLAS_LIBRARIES)
set(BLAS_INFO "open")
endif(BLAS_LIBRARIES)
endif()
if((NOT BLAS_LIBRARIES)
AND ((NOT WITH_BLAS) OR (WITH_BLAS STREQUAL "goto")))
check_fortran_libraries(
@ -131,6 +145,19 @@ if((NOT BLAS_LIBRARIES)
endif (BLAS_LIBRARIES)
endif()
if((NOT BLAS_LIBRARIES)
AND ((NOT WITH_BLAS) OR (WITH_BLAS STREQUAL "goto")))
check_fortran_libraries(
BLAS_LIBRARIES
BLAS
sgemm
""
"goto2;gfortran;pthread")
if (BLAS_LIBRARIES)
set(BLAS_INFO "goto")
endif (BLAS_LIBRARIES)
endif()
if((NOT BLAS_LIBRARIES)
AND ((NOT WITH_BLAS) OR (WITH_BLAS STREQUAL "acml")))
check_fortran_libraries(