FindBLAS.cmake 23 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802
  1. # Distributed under the OSI-approved BSD 3-Clause License. See accompanying
  2. # file Copyright.txt or https://cmake.org/licensing for details.
  3. #[=======================================================================[.rst:
  4. FindBLAS
  5. --------
  6. Find Basic Linear Algebra Subprograms (BLAS) library
  7. This module finds an installed Fortran library that implements the
  8. BLAS linear-algebra interface (see http://www.netlib.org/blas/). The
  9. list of libraries searched for is taken from the ``autoconf`` macro file,
  10. ``acx_blas.m4`` (distributed at
  11. http://ac-archive.sourceforge.net/ac-archive/acx_blas.html).
  12. Input Variables
  13. ^^^^^^^^^^^^^^^
  14. The following variables may be set to influence this module's behavior:
  15. ``BLA_STATIC``
  16. if ``ON`` use static linkage
  17. ``BLA_VENDOR``
  18. If set, checks only the specified vendor, if not set checks all the
  19. possibilities. List of vendors valid in this module:
  20. * Goto
  21. * OpenBLAS
  22. * FLAME
  23. * ATLAS PhiPACK
  24. * CXML
  25. * DXML
  26. * SunPerf
  27. * SCSL
  28. * SGIMATH
  29. * IBMESSL
  30. * Intel10_32 (intel mkl v10 32 bit)
  31. * Intel10_64lp (intel mkl v10+ 64 bit, threaded code, lp64 model)
  32. * Intel10_64lp_seq (intel mkl v10+ 64 bit, sequential code, lp64 model)
  33. * Intel10_64ilp (intel mkl v10+ 64 bit, threaded code, ilp64 model)
  34. * Intel10_64ilp_seq (intel mkl v10+ 64 bit, sequential code, ilp64 model)
  35. * Intel (obsolete versions of mkl 32 and 64 bit)
  36. * ACML
  37. * ACML_MP
  38. * ACML_GPU
  39. * Apple
  40. * NAS
  41. * Generic
  42. ``BLA_F95``
  43. if ``ON`` tries to find the BLAS95 interfaces
  44. ``BLA_PREFER_PKGCONFIG``
  45. if set ``pkg-config`` will be used to search for a BLAS library first
  46. and if one is found that is preferred
  47. Result Variables
  48. ^^^^^^^^^^^^^^^^
  49. This module defines the following variables:
  50. ``BLAS_FOUND``
  51. library implementing the BLAS interface is found
  52. ``BLAS_LINKER_FLAGS``
  53. uncached list of required linker flags (excluding ``-l`` and ``-L``).
  54. ``BLAS_LIBRARIES``
  55. uncached list of libraries (using full path name) to link against
  56. to use BLAS (may be empty if compiler implicitly links BLAS)
  57. ``BLAS95_LIBRARIES``
  58. uncached list of libraries (using full path name) to link against
  59. to use BLAS95 interface
  60. ``BLAS95_FOUND``
  61. library implementing the BLAS95 interface is found
  62. .. note::
  63. C or CXX must be enabled to use Intel Math Kernel Library (MKL)
  64. For example, to use Intel MKL libraries and/or Intel compiler:
  65. .. code-block:: cmake
  66. set(BLA_VENDOR Intel10_64lp)
  67. find_package(BLAS)
  68. Hints
  69. ^^^^^
  70. Set ``MKLROOT`` environment variable to a directory that contains an MKL
  71. installation.
  72. #]=======================================================================]
  73. include(${CMAKE_CURRENT_LIST_DIR}/CheckFunctionExists.cmake)
  74. include(${CMAKE_CURRENT_LIST_DIR}/CheckFortranFunctionExists.cmake)
  75. include(${CMAKE_CURRENT_LIST_DIR}/CMakePushCheckState.cmake)
  76. include(${CMAKE_CURRENT_LIST_DIR}/FindPackageHandleStandardArgs.cmake)
  77. cmake_push_check_state()
  78. set(CMAKE_REQUIRED_QUIET ${BLAS_FIND_QUIETLY})
  79. set(_blas_ORIG_CMAKE_FIND_LIBRARY_SUFFIXES ${CMAKE_FIND_LIBRARY_SUFFIXES})
  80. # Check the language being used
  81. if( NOT (CMAKE_C_COMPILER_LOADED OR CMAKE_CXX_COMPILER_LOADED OR CMAKE_Fortran_COMPILER_LOADED) )
  82. if(BLAS_FIND_REQUIRED)
  83. message(FATAL_ERROR "FindBLAS requires Fortran, C, or C++ to be enabled.")
  84. else()
  85. message(STATUS "Looking for BLAS... - NOT found (Unsupported languages)")
  86. return()
  87. endif()
  88. endif()
  89. if(BLA_PREFER_PKGCONFIG)
  90. find_package(PkgConfig)
  91. pkg_check_modules(PKGC_BLAS blas)
  92. if(PKGC_BLAS_FOUND)
  93. set(BLAS_FOUND ${PKGC_BLAS_FOUND})
  94. set(BLAS_LIBRARIES "${PKGC_BLAS_LINK_LIBRARIES}")
  95. return()
  96. endif()
  97. endif()
  98. macro(Check_Fortran_Libraries LIBRARIES _prefix _name _flags _list _thread)
  99. # This macro checks for the existence of the combination of fortran libraries
  100. # given by _list. If the combination is found, this macro checks (using the
  101. # Check_Fortran_Function_Exists macro) whether can link against that library
  102. # combination using the name of a routine given by _name using the linker
  103. # flags given by _flags. If the combination of libraries is found and passes
  104. # the link test, LIBRARIES is set to the list of complete library paths that
  105. # have been found. Otherwise, LIBRARIES is set to FALSE.
  106. # N.B. _prefix is the prefix applied to the names of all cached variables that
  107. # are generated internally and marked advanced by this macro.
  108. set(_libdir ${ARGN})
  109. set(_libraries_work TRUE)
  110. set(${LIBRARIES})
  111. set(_combined_name)
  112. if (NOT _libdir)
  113. if (WIN32)
  114. set(_libdir ENV LIB)
  115. elseif (APPLE)
  116. set(_libdir ENV DYLD_LIBRARY_PATH)
  117. else ()
  118. set(_libdir ENV LD_LIBRARY_PATH)
  119. endif ()
  120. endif ()
  121. list(APPEND _libdir "${CMAKE_C_IMPLICIT_LINK_DIRECTORIES}")
  122. foreach(_library ${_list})
  123. set(_combined_name ${_combined_name}_${_library})
  124. if(_libraries_work)
  125. if (BLA_STATIC)
  126. if (WIN32)
  127. set(CMAKE_FIND_LIBRARY_SUFFIXES .lib ${CMAKE_FIND_LIBRARY_SUFFIXES})
  128. endif ()
  129. if (APPLE)
  130. set(CMAKE_FIND_LIBRARY_SUFFIXES .lib ${CMAKE_FIND_LIBRARY_SUFFIXES})
  131. else ()
  132. set(CMAKE_FIND_LIBRARY_SUFFIXES .a ${CMAKE_FIND_LIBRARY_SUFFIXES})
  133. endif ()
  134. else ()
  135. if (CMAKE_SYSTEM_NAME STREQUAL "Linux")
  136. # for ubuntu's libblas3gf and liblapack3gf packages
  137. set(CMAKE_FIND_LIBRARY_SUFFIXES ${CMAKE_FIND_LIBRARY_SUFFIXES} .so.3gf)
  138. endif ()
  139. endif ()
  140. find_library(${_prefix}_${_library}_LIBRARY
  141. NAMES ${_library}
  142. PATHS ${_libdir}
  143. )
  144. mark_as_advanced(${_prefix}_${_library}_LIBRARY)
  145. set(${LIBRARIES} ${${LIBRARIES}} ${${_prefix}_${_library}_LIBRARY})
  146. set(_libraries_work ${${_prefix}_${_library}_LIBRARY})
  147. endif()
  148. endforeach()
  149. if(_libraries_work)
  150. # Test this combination of libraries.
  151. set(CMAKE_REQUIRED_LIBRARIES ${_flags} ${${LIBRARIES}} ${_thread})
  152. # message("DEBUG: CMAKE_REQUIRED_LIBRARIES = ${CMAKE_REQUIRED_LIBRARIES}")
  153. if (CMAKE_Fortran_COMPILER_LOADED)
  154. check_fortran_function_exists("${_name}" ${_prefix}${_combined_name}_WORKS)
  155. else()
  156. check_function_exists("${_name}_" ${_prefix}${_combined_name}_WORKS)
  157. endif()
  158. set(CMAKE_REQUIRED_LIBRARIES)
  159. set(_libraries_work ${${_prefix}${_combined_name}_WORKS})
  160. endif()
  161. if(_libraries_work)
  162. if("${_list}" STREQUAL "")
  163. set(${LIBRARIES} "${LIBRARIES}-PLACEHOLDER-FOR-EMPTY-LIBRARIES")
  164. else()
  165. set(${LIBRARIES} ${${LIBRARIES}} ${_thread}) # for static link
  166. endif()
  167. else()
  168. set(${LIBRARIES} FALSE)
  169. endif()
  170. #message("DEBUG: ${LIBRARIES} = ${${LIBRARIES}}")
  171. endmacro()
  172. set(BLAS_LINKER_FLAGS)
  173. set(BLAS_LIBRARIES)
  174. set(BLAS95_LIBRARIES)
  175. if (NOT $ENV{BLA_VENDOR} STREQUAL "")
  176. set(BLA_VENDOR $ENV{BLA_VENDOR})
  177. else ()
  178. if(NOT BLA_VENDOR)
  179. set(BLA_VENDOR "All")
  180. endif()
  181. endif ()
  182. if (BLA_VENDOR STREQUAL "All")
  183. if(NOT BLAS_LIBRARIES)
  184. # Implicitly linked BLAS libraries
  185. check_fortran_libraries(
  186. BLAS_LIBRARIES
  187. BLAS
  188. sgemm
  189. ""
  190. ""
  191. ""
  192. )
  193. endif()
  194. endif ()
  195. #BLAS in intel mkl 10+ library? (em64t 64bit)
  196. if (BLA_VENDOR MATCHES "Intel" OR BLA_VENDOR STREQUAL "All")
  197. if (NOT BLAS_LIBRARIES)
  198. # System-specific settings
  199. if (WIN32)
  200. if (BLA_STATIC)
  201. set(BLAS_mkl_DLL_SUFFIX "")
  202. else()
  203. set(BLAS_mkl_DLL_SUFFIX "_dll")
  204. endif()
  205. else()
  206. # Switch to GNU Fortran support layer if needed (but not on Apple, where MKL does not provide it)
  207. if(CMAKE_Fortran_COMPILER_LOADED AND CMAKE_Fortran_COMPILER_ID STREQUAL "GNU" AND NOT APPLE)
  208. set(BLAS_mkl_INTFACE "gf")
  209. set(BLAS_mkl_THREADING "gnu")
  210. set(BLAS_mkl_OMP "gomp")
  211. else()
  212. set(BLAS_mkl_INTFACE "intel")
  213. set(BLAS_mkl_THREADING "intel")
  214. set(BLAS_mkl_OMP "iomp5")
  215. endif()
  216. set(BLAS_mkl_LM "-lm")
  217. set(BLAS_mkl_LDL "-ldl")
  218. endif()
  219. if (BLA_VENDOR MATCHES "_64ilp")
  220. set(BLAS_mkl_ILP_MODE "ilp64")
  221. else ()
  222. set(BLAS_mkl_ILP_MODE "lp64")
  223. endif ()
  224. if (CMAKE_C_COMPILER_LOADED OR CMAKE_CXX_COMPILER_LOADED)
  225. if(BLAS_FIND_QUIETLY OR NOT BLAS_FIND_REQUIRED)
  226. find_package(Threads)
  227. else()
  228. find_package(Threads REQUIRED)
  229. endif()
  230. set(BLAS_SEARCH_LIBS "")
  231. if(BLA_F95)
  232. set(BLAS_mkl_SEARCH_SYMBOL sgemm_f95)
  233. set(_LIBRARIES BLAS95_LIBRARIES)
  234. if (WIN32)
  235. # Find the main file (32-bit or 64-bit)
  236. set(BLAS_SEARCH_LIBS_WIN_MAIN "")
  237. if (BLA_VENDOR STREQUAL "Intel10_32" OR BLA_VENDOR STREQUAL "All")
  238. list(APPEND BLAS_SEARCH_LIBS_WIN_MAIN
  239. "mkl_blas95${BLAS_mkl_DLL_SUFFIX} mkl_intel_c${BLAS_mkl_DLL_SUFFIX}")
  240. endif()
  241. if (BLA_VENDOR MATCHES "^Intel10_64i?lp" OR BLA_VENDOR STREQUAL "All")
  242. list(APPEND BLAS_SEARCH_LIBS_WIN_MAIN
  243. "mkl_blas95_${BLAS_mkl_ILP_MODE}${BLAS_mkl_DLL_SUFFIX} mkl_intel_${BLAS_mkl_ILP_MODE}${BLAS_mkl_DLL_SUFFIX}")
  244. endif ()
  245. # Add threading/sequential libs
  246. set(BLAS_SEARCH_LIBS_WIN_THREAD "")
  247. if (BLA_VENDOR MATCHES "_seq$" OR BLA_VENDOR STREQUAL "All")
  248. list(APPEND BLAS_SEARCH_LIBS_WIN_THREAD
  249. "mkl_sequential${BLAS_mkl_DLL_SUFFIX}")
  250. endif()
  251. if (NOT BLA_VENDOR MATCHES "_seq$" OR BLA_VENDOR STREQUAL "All")
  252. # old version
  253. list(APPEND BLAS_SEARCH_LIBS_WIN_THREAD
  254. "libguide40 mkl_intel_thread${BLAS_mkl_DLL_SUFFIX}")
  255. # mkl >= 10.3
  256. list(APPEND BLAS_SEARCH_LIBS_WIN_THREAD
  257. "libiomp5md mkl_intel_thread${BLAS_mkl_DLL_SUFFIX}")
  258. endif()
  259. # Cartesian product of the above
  260. foreach (MAIN ${BLAS_SEARCH_LIBS_WIN_MAIN})
  261. foreach (THREAD ${BLAS_SEARCH_LIBS_WIN_THREAD})
  262. list(APPEND BLAS_SEARCH_LIBS
  263. "${MAIN} ${THREAD} mkl_core${BLAS_mkl_DLL_SUFFIX}")
  264. endforeach()
  265. endforeach()
  266. else ()
  267. if (BLA_VENDOR STREQUAL "Intel10_32" OR BLA_VENDOR STREQUAL "All")
  268. # old version
  269. list(APPEND BLAS_SEARCH_LIBS
  270. "mkl_blas95 mkl_${BLAS_mkl_INTFACE} mkl_${BLAS_mkl_THREADING}_thread mkl_core guide")
  271. # mkl >= 10.3
  272. list(APPEND BLAS_SEARCH_LIBS
  273. "mkl_blas95 mkl_${BLAS_mkl_INTFACE} mkl_${BLAS_mkl_THREADING}_thread mkl_core ${BLAS_mkl_OMP}")
  274. endif ()
  275. if (BLA_VENDOR MATCHES "^Intel10_64i?lp$" OR BLA_VENDOR STREQUAL "All")
  276. # old version
  277. list(APPEND BLAS_SEARCH_LIBS
  278. "mkl_blas95 mkl_${BLAS_mkl_INTFACE}_${BLAS_mkl_ILP_MODE} mkl_${BLAS_mkl_THREADING}_thread mkl_core guide")
  279. # mkl >= 10.3
  280. list(APPEND BLAS_SEARCH_LIBS
  281. "mkl_blas95_${BLAS_mkl_ILP_MODE} mkl_${BLAS_mkl_INTFACE}_${BLAS_mkl_ILP_MODE} mkl_${BLAS_mkl_THREADING}_thread mkl_core ${BLAS_mkl_OMP}")
  282. endif ()
  283. if (BLA_VENDOR MATCHES "^Intel10_64i?lp_seq$" OR BLA_VENDOR STREQUAL "All")
  284. list(APPEND BLAS_SEARCH_LIBS
  285. "mkl_blas95_${BLAS_mkl_ILP_MODE} mkl_${BLAS_mkl_INTFACE}_${BLAS_mkl_ILP_MODE} mkl_sequential mkl_core")
  286. endif ()
  287. endif ()
  288. else ()
  289. set(BLAS_mkl_SEARCH_SYMBOL sgemm)
  290. set(_LIBRARIES BLAS_LIBRARIES)
  291. if (WIN32)
  292. # Find the main file (32-bit or 64-bit)
  293. set(BLAS_SEARCH_LIBS_WIN_MAIN "")
  294. if (BLA_VENDOR STREQUAL "Intel10_32" OR BLA_VENDOR STREQUAL "All")
  295. list(APPEND BLAS_SEARCH_LIBS_WIN_MAIN
  296. "mkl_intel_c${BLAS_mkl_DLL_SUFFIX}")
  297. endif()
  298. if (BLA_VENDOR MATCHES "^Intel10_64i?lp" OR BLA_VENDOR STREQUAL "All")
  299. list(APPEND BLAS_SEARCH_LIBS_WIN_MAIN
  300. "mkl_intel_${BLAS_mkl_ILP_MODE}${BLAS_mkl_DLL_SUFFIX}")
  301. endif ()
  302. # Add threading/sequential libs
  303. set(BLAS_SEARCH_LIBS_WIN_THREAD "")
  304. if (NOT BLA_VENDOR MATCHES "_seq$" OR BLA_VENDOR STREQUAL "All")
  305. # old version
  306. list(APPEND BLAS_SEARCH_LIBS_WIN_THREAD
  307. "libguide40 mkl_intel_thread${BLAS_mkl_DLL_SUFFIX}")
  308. # mkl >= 10.3
  309. list(APPEND BLAS_SEARCH_LIBS_WIN_THREAD
  310. "libiomp5md mkl_intel_thread${BLAS_mkl_DLL_SUFFIX}")
  311. endif()
  312. if (BLA_VENDOR MATCHES "_seq$" OR BLA_VENDOR STREQUAL "All")
  313. list(APPEND BLAS_SEARCH_LIBS_WIN_THREAD
  314. "mkl_sequential${BLAS_mkl_DLL_SUFFIX}")
  315. endif()
  316. # Cartesian product of the above
  317. foreach (MAIN ${BLAS_SEARCH_LIBS_WIN_MAIN})
  318. foreach (THREAD ${BLAS_SEARCH_LIBS_WIN_THREAD})
  319. list(APPEND BLAS_SEARCH_LIBS
  320. "${MAIN} ${THREAD} mkl_core${BLAS_mkl_DLL_SUFFIX}")
  321. endforeach()
  322. endforeach()
  323. else ()
  324. if (BLA_VENDOR STREQUAL "Intel10_32" OR BLA_VENDOR STREQUAL "All")
  325. # old version
  326. list(APPEND BLAS_SEARCH_LIBS
  327. "mkl_${BLAS_mkl_INTFACE} mkl_${BLAS_mkl_THREADING}_thread mkl_core guide")
  328. # mkl >= 10.3
  329. list(APPEND BLAS_SEARCH_LIBS
  330. "mkl_${BLAS_mkl_INTFACE} mkl_${BLAS_mkl_THREADING}_thread mkl_core ${BLAS_mkl_OMP}")
  331. endif ()
  332. if (BLA_VENDOR MATCHES "^Intel10_64i?lp$" OR BLA_VENDOR STREQUAL "All")
  333. # old version
  334. list(APPEND BLAS_SEARCH_LIBS
  335. "mkl_${BLAS_mkl_INTFACE}_${BLAS_mkl_ILP_MODE} mkl_${BLAS_mkl_THREADING}_thread mkl_core guide")
  336. # mkl >= 10.3
  337. list(APPEND BLAS_SEARCH_LIBS
  338. "mkl_${BLAS_mkl_INTFACE}_${BLAS_mkl_ILP_MODE} mkl_${BLAS_mkl_THREADING}_thread mkl_core ${BLAS_mkl_OMP}")
  339. endif ()
  340. if (BLA_VENDOR MATCHES "^Intel10_64i?lp_seq$" OR BLA_VENDOR STREQUAL "All")
  341. list(APPEND BLAS_SEARCH_LIBS
  342. "mkl_${BLAS_mkl_INTFACE}_${BLAS_mkl_ILP_MODE} mkl_sequential mkl_core")
  343. endif ()
  344. #older vesions of intel mkl libs
  345. if (BLA_VENDOR STREQUAL "Intel" OR BLA_VENDOR STREQUAL "All")
  346. list(APPEND BLAS_SEARCH_LIBS
  347. "mkl")
  348. list(APPEND BLAS_SEARCH_LIBS
  349. "mkl_ia32")
  350. list(APPEND BLAS_SEARCH_LIBS
  351. "mkl_em64t")
  352. endif ()
  353. endif ()
  354. endif ()
  355. if (DEFINED ENV{MKLROOT})
  356. if (BLA_VENDOR STREQUAL "Intel10_32")
  357. set(_BLAS_MKLROOT_LIB_DIR "$ENV{MKLROOT}/lib/ia32")
  358. elseif (BLA_VENDOR MATCHES "^Intel10_64i?lp$" OR BLA_VENDOR MATCHES "^Intel10_64i?lp_seq$")
  359. set(_BLAS_MKLROOT_LIB_DIR "$ENV{MKLROOT}/lib/intel64")
  360. endif ()
  361. endif ()
  362. if (_BLAS_MKLROOT_LIB_DIR)
  363. if (WIN32)
  364. string(APPEND _BLAS_MKLROOT_LIB_DIR "_win")
  365. elseif (APPLE)
  366. string(APPEND _BLAS_MKLROOT_LIB_DIR "_mac")
  367. else ()
  368. string(APPEND _BLAS_MKLROOT_LIB_DIR "_lin")
  369. endif ()
  370. endif ()
  371. foreach (IT ${BLAS_SEARCH_LIBS})
  372. string(REPLACE " " ";" SEARCH_LIBS ${IT})
  373. if (NOT ${_LIBRARIES})
  374. check_fortran_libraries(
  375. ${_LIBRARIES}
  376. BLAS
  377. ${BLAS_mkl_SEARCH_SYMBOL}
  378. ""
  379. "${SEARCH_LIBS}"
  380. "${CMAKE_THREAD_LIBS_INIT};${BLAS_mkl_LM};${BLAS_mkl_LDL}"
  381. "${_BLAS_MKLROOT_LIB_DIR}"
  382. )
  383. endif ()
  384. endforeach ()
  385. endif ()
  386. unset(BLAS_mkl_ILP_MODE)
  387. unset(BLAS_mkl_INTFACE)
  388. unset(BLAS_mkl_THREADING)
  389. unset(BLAS_mkl_OMP)
  390. unset(BLAS_mkl_DLL_SUFFIX)
  391. unset(BLAS_mkl_LM)
  392. unset(BLAS_mkl_LDL)
  393. endif ()
  394. endif ()
  395. if(BLA_F95)
  396. find_package_handle_standard_args(BLAS REQUIRED_VARS BLAS95_LIBRARIES)
  397. set(BLAS95_FOUND ${BLAS_FOUND})
  398. if(BLAS_FOUND)
  399. set(BLAS_LIBRARIES "${BLAS95_LIBRARIES}")
  400. endif()
  401. endif()
  402. if (BLA_VENDOR STREQUAL "Goto" OR BLA_VENDOR STREQUAL "All")
  403. if(NOT BLAS_LIBRARIES)
  404. # gotoblas (http://www.tacc.utexas.edu/tacc-projects/gotoblas2)
  405. check_fortran_libraries(
  406. BLAS_LIBRARIES
  407. BLAS
  408. sgemm
  409. ""
  410. "goto2"
  411. ""
  412. )
  413. endif()
  414. endif ()
  415. if (BLA_VENDOR STREQUAL "OpenBLAS" OR BLA_VENDOR STREQUAL "All")
  416. if(NOT BLAS_LIBRARIES)
  417. # OpenBLAS (http://www.openblas.net)
  418. check_fortran_libraries(
  419. BLAS_LIBRARIES
  420. BLAS
  421. sgemm
  422. ""
  423. "openblas"
  424. ""
  425. )
  426. endif()
  427. endif ()
  428. if (BLA_VENDOR STREQUAL "FLAME" OR BLA_VENDOR STREQUAL "All")
  429. if(NOT BLAS_LIBRARIES)
  430. # FLAME's blis library (https://github.com/flame/blis)
  431. check_fortran_libraries(
  432. BLAS_LIBRARIES
  433. BLAS
  434. sgemm
  435. ""
  436. "blis"
  437. ""
  438. )
  439. endif()
  440. endif ()
  441. if (BLA_VENDOR STREQUAL "ATLAS" OR BLA_VENDOR STREQUAL "All")
  442. if(NOT BLAS_LIBRARIES)
  443. # BLAS in ATLAS library? (http://math-atlas.sourceforge.net/)
  444. check_fortran_libraries(
  445. BLAS_LIBRARIES
  446. BLAS
  447. dgemm
  448. ""
  449. "f77blas;atlas"
  450. ""
  451. )
  452. endif()
  453. endif ()
  454. # BLAS in PhiPACK libraries? (requires generic BLAS lib, too)
  455. if (BLA_VENDOR STREQUAL "PhiPACK" OR BLA_VENDOR STREQUAL "All")
  456. if(NOT BLAS_LIBRARIES)
  457. check_fortran_libraries(
  458. BLAS_LIBRARIES
  459. BLAS
  460. sgemm
  461. ""
  462. "sgemm;dgemm;blas"
  463. ""
  464. )
  465. endif()
  466. endif ()
  467. # BLAS in Alpha CXML library?
  468. if (BLA_VENDOR STREQUAL "CXML" OR BLA_VENDOR STREQUAL "All")
  469. if(NOT BLAS_LIBRARIES)
  470. check_fortran_libraries(
  471. BLAS_LIBRARIES
  472. BLAS
  473. sgemm
  474. ""
  475. "cxml"
  476. ""
  477. )
  478. endif()
  479. endif ()
  480. # BLAS in Alpha DXML library? (now called CXML, see above)
  481. if (BLA_VENDOR STREQUAL "DXML" OR BLA_VENDOR STREQUAL "All")
  482. if(NOT BLAS_LIBRARIES)
  483. check_fortran_libraries(
  484. BLAS_LIBRARIES
  485. BLAS
  486. sgemm
  487. ""
  488. "dxml"
  489. ""
  490. )
  491. endif()
  492. endif ()
  493. # BLAS in Sun Performance library?
  494. if (BLA_VENDOR STREQUAL "SunPerf" OR BLA_VENDOR STREQUAL "All")
  495. if(NOT BLAS_LIBRARIES)
  496. check_fortran_libraries(
  497. BLAS_LIBRARIES
  498. BLAS
  499. sgemm
  500. "-xlic_lib=sunperf"
  501. "sunperf;sunmath"
  502. ""
  503. )
  504. if(BLAS_LIBRARIES)
  505. set(BLAS_LINKER_FLAGS "-xlic_lib=sunperf")
  506. endif()
  507. endif()
  508. endif ()
  509. # BLAS in SCSL library? (SGI/Cray Scientific Library)
  510. if (BLA_VENDOR STREQUAL "SCSL" OR BLA_VENDOR STREQUAL "All")
  511. if(NOT BLAS_LIBRARIES)
  512. check_fortran_libraries(
  513. BLAS_LIBRARIES
  514. BLAS
  515. sgemm
  516. ""
  517. "scsl"
  518. ""
  519. )
  520. endif()
  521. endif ()
  522. # BLAS in SGIMATH library?
  523. if (BLA_VENDOR STREQUAL "SGIMATH" OR BLA_VENDOR STREQUAL "All")
  524. if(NOT BLAS_LIBRARIES)
  525. check_fortran_libraries(
  526. BLAS_LIBRARIES
  527. BLAS
  528. sgemm
  529. ""
  530. "complib.sgimath"
  531. ""
  532. )
  533. endif()
  534. endif ()
  535. # BLAS in IBM ESSL library? (requires generic BLAS lib, too)
  536. if (BLA_VENDOR STREQUAL "IBMESSL" OR BLA_VENDOR STREQUAL "All")
  537. if(NOT BLAS_LIBRARIES)
  538. check_fortran_libraries(
  539. BLAS_LIBRARIES
  540. BLAS
  541. sgemm
  542. ""
  543. "essl;blas"
  544. ""
  545. )
  546. endif()
  547. endif ()
  548. #BLAS in acml library?
  549. if (BLA_VENDOR MATCHES "ACML" OR BLA_VENDOR STREQUAL "All")
  550. if( ((BLA_VENDOR STREQUAL "ACML") AND (NOT BLAS_ACML_LIB_DIRS)) OR
  551. ((BLA_VENDOR STREQUAL "ACML_MP") AND (NOT BLAS_ACML_MP_LIB_DIRS)) OR
  552. ((BLA_VENDOR STREQUAL "ACML_GPU") AND (NOT BLAS_ACML_GPU_LIB_DIRS))
  553. )
  554. # try to find acml in "standard" paths
  555. if( WIN32 )
  556. file( GLOB _ACML_ROOT "C:/AMD/acml*/ACML-EULA.txt" )
  557. else()
  558. file( GLOB _ACML_ROOT "/opt/acml*/ACML-EULA.txt" )
  559. endif()
  560. if( WIN32 )
  561. file( GLOB _ACML_GPU_ROOT "C:/AMD/acml*/GPGPUexamples" )
  562. else()
  563. file( GLOB _ACML_GPU_ROOT "/opt/acml*/GPGPUexamples" )
  564. endif()
  565. list(GET _ACML_ROOT 0 _ACML_ROOT)
  566. list(GET _ACML_GPU_ROOT 0 _ACML_GPU_ROOT)
  567. if( _ACML_ROOT )
  568. get_filename_component( _ACML_ROOT ${_ACML_ROOT} PATH )
  569. if( SIZEOF_INTEGER EQUAL 8 )
  570. set( _ACML_PATH_SUFFIX "_int64" )
  571. else()
  572. set( _ACML_PATH_SUFFIX "" )
  573. endif()
  574. if( CMAKE_Fortran_COMPILER_ID STREQUAL "Intel" )
  575. set( _ACML_COMPILER32 "ifort32" )
  576. set( _ACML_COMPILER64 "ifort64" )
  577. elseif( CMAKE_Fortran_COMPILER_ID STREQUAL "SunPro" )
  578. set( _ACML_COMPILER32 "sun32" )
  579. set( _ACML_COMPILER64 "sun64" )
  580. elseif( CMAKE_Fortran_COMPILER_ID STREQUAL "PGI" )
  581. set( _ACML_COMPILER32 "pgi32" )
  582. if( WIN32 )
  583. set( _ACML_COMPILER64 "win64" )
  584. else()
  585. set( _ACML_COMPILER64 "pgi64" )
  586. endif()
  587. elseif( CMAKE_Fortran_COMPILER_ID STREQUAL "Open64" )
  588. # 32 bit builds not supported on Open64 but for code simplicity
  589. # We'll just use the same directory twice
  590. set( _ACML_COMPILER32 "open64_64" )
  591. set( _ACML_COMPILER64 "open64_64" )
  592. elseif( CMAKE_Fortran_COMPILER_ID STREQUAL "NAG" )
  593. set( _ACML_COMPILER32 "nag32" )
  594. set( _ACML_COMPILER64 "nag64" )
  595. else()
  596. set( _ACML_COMPILER32 "gfortran32" )
  597. set( _ACML_COMPILER64 "gfortran64" )
  598. endif()
  599. if( BLA_VENDOR STREQUAL "ACML_MP" )
  600. set(_ACML_MP_LIB_DIRS
  601. "${_ACML_ROOT}/${_ACML_COMPILER32}_mp${_ACML_PATH_SUFFIX}/lib"
  602. "${_ACML_ROOT}/${_ACML_COMPILER64}_mp${_ACML_PATH_SUFFIX}/lib" )
  603. else()
  604. set(_ACML_LIB_DIRS
  605. "${_ACML_ROOT}/${_ACML_COMPILER32}${_ACML_PATH_SUFFIX}/lib"
  606. "${_ACML_ROOT}/${_ACML_COMPILER64}${_ACML_PATH_SUFFIX}/lib" )
  607. endif()
  608. endif()
  609. elseif(BLAS_${BLA_VENDOR}_LIB_DIRS)
  610. set(_${BLA_VENDOR}_LIB_DIRS ${BLAS_${BLA_VENDOR}_LIB_DIRS})
  611. endif()
  612. if( BLA_VENDOR STREQUAL "ACML_MP" )
  613. foreach( BLAS_ACML_MP_LIB_DIRS ${_ACML_MP_LIB_DIRS})
  614. check_fortran_libraries (
  615. BLAS_LIBRARIES
  616. BLAS
  617. sgemm
  618. "" "acml_mp;acml_mv" "" ${BLAS_ACML_MP_LIB_DIRS}
  619. )
  620. if( BLAS_LIBRARIES )
  621. break()
  622. endif()
  623. endforeach()
  624. elseif( BLA_VENDOR STREQUAL "ACML_GPU" )
  625. foreach( BLAS_ACML_GPU_LIB_DIRS ${_ACML_GPU_LIB_DIRS})
  626. check_fortran_libraries (
  627. BLAS_LIBRARIES
  628. BLAS
  629. sgemm
  630. "" "acml;acml_mv;CALBLAS" "" ${BLAS_ACML_GPU_LIB_DIRS}
  631. )
  632. if( BLAS_LIBRARIES )
  633. break()
  634. endif()
  635. endforeach()
  636. else()
  637. foreach( BLAS_ACML_LIB_DIRS ${_ACML_LIB_DIRS} )
  638. check_fortran_libraries (
  639. BLAS_LIBRARIES
  640. BLAS
  641. sgemm
  642. "" "acml;acml_mv" "" ${BLAS_ACML_LIB_DIRS}
  643. )
  644. if( BLAS_LIBRARIES )
  645. break()
  646. endif()
  647. endforeach()
  648. endif()
  649. # Either acml or acml_mp should be in LD_LIBRARY_PATH but not both
  650. if(NOT BLAS_LIBRARIES)
  651. check_fortran_libraries(
  652. BLAS_LIBRARIES
  653. BLAS
  654. sgemm
  655. ""
  656. "acml;acml_mv"
  657. ""
  658. )
  659. endif()
  660. if(NOT BLAS_LIBRARIES)
  661. check_fortran_libraries(
  662. BLAS_LIBRARIES
  663. BLAS
  664. sgemm
  665. ""
  666. "acml_mp;acml_mv"
  667. ""
  668. )
  669. endif()
  670. if(NOT BLAS_LIBRARIES)
  671. check_fortran_libraries(
  672. BLAS_LIBRARIES
  673. BLAS
  674. sgemm
  675. ""
  676. "acml;acml_mv;CALBLAS"
  677. ""
  678. )
  679. endif()
  680. endif () # ACML
  681. # Apple BLAS library?
  682. if (BLA_VENDOR STREQUAL "Apple" OR BLA_VENDOR STREQUAL "All")
  683. if(NOT BLAS_LIBRARIES)
  684. check_fortran_libraries(
  685. BLAS_LIBRARIES
  686. BLAS
  687. dgemm
  688. ""
  689. "Accelerate"
  690. ""
  691. )
  692. endif()
  693. endif ()
  694. if (BLA_VENDOR STREQUAL "NAS" OR BLA_VENDOR STREQUAL "All")
  695. if ( NOT BLAS_LIBRARIES )
  696. check_fortran_libraries(
  697. BLAS_LIBRARIES
  698. BLAS
  699. dgemm
  700. ""
  701. "vecLib"
  702. ""
  703. )
  704. endif ()
  705. endif ()
  706. # Generic BLAS library?
  707. if (BLA_VENDOR STREQUAL "Generic" OR BLA_VENDOR STREQUAL "All")
  708. if(NOT BLAS_LIBRARIES)
  709. check_fortran_libraries(
  710. BLAS_LIBRARIES
  711. BLAS
  712. sgemm
  713. ""
  714. "blas"
  715. ""
  716. )
  717. endif()
  718. endif ()
  719. if(NOT BLA_F95)
  720. find_package_handle_standard_args(BLAS REQUIRED_VARS BLAS_LIBRARIES)
  721. endif()
  722. # On compilers that implicitly link BLAS (such as ftn, cc, and CC on Cray HPC machines)
  723. # we used a placeholder for empty BLAS_LIBRARIES to get through our logic above.
  724. if (BLAS_LIBRARIES STREQUAL "BLAS_LIBRARIES-PLACEHOLDER-FOR-EMPTY-LIBRARIES")
  725. set(BLAS_LIBRARIES "")
  726. endif()
  727. cmake_pop_check_state()
  728. set(CMAKE_FIND_LIBRARY_SUFFIXES ${_blas_ORIG_CMAKE_FIND_LIBRARY_SUFFIXES})