FindBLAS.cmake 29 KB

1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162636465666768697071727374757677787980818283848586878889909192939495969798991001011021031041051061071081091101111121131141151161171181191201211221231241251261271281291301311321331341351361371381391401411421431441451461471481491501511521531541551561571581591601611621631641651661671681691701711721731741751761771781791801811821831841851861871881891901911921931941951961971981992002012022032042052062072082092102112122132142152162172182192202212222232242252262272282292302312322332342352362372382392402412422432442452462472482492502512522532542552562572582592602612622632642652662672682692702712722732742752762772782792802812822832842852862872882892902912922932942952962972982993003013023033043053063073083093103113123133143153163173183193203213223233243253263273283293303313323333343353363373383393403413423433443453463473483493503513523533543553563573583593603613623633643653663673683693703713723733743753763773783793803813823833843853863873883893903913923933943953963973983994004014024034044054064074084094104114124134144154164174184194204214224234244254264274284294304314324334344354364374384394404414424434444454464474484494504514524534544554564574584594604614624634644654664674684694704714724734744754764774784794804814824834844854864874884894904914924934944954964974984995005015025035045055065075085095105115125135145155165175185195205215225235245255265275285295305315325335345355365375385395405415425435445455465475485495505515525535545555565575585595605615625635645655665675685695705715725735745755765775785795805815825835845855865875885895905915925935945955965975985996006016026036046056066076086096106116126136146156166176186196206216226236246256266276286296306316326336346356366376386396406416426436446456466476486496506516526536546556566576586596606616626636646656666676686696706716726736746756766776786796806816826836846856866876886896906916926936946956966976986997007017027037047057067077087097107117127137147157167177187197207217227237247257267277287297307317327337347357367377387397407417427437447457467477487497507517527537547557567577587597607617627637647657667677687697707717727737747757767777787797807817827837847857867877887897907917927937947957967977987998008018028038048058068078088098108118128138148158168178188198208218228238248258268278288298308318328338348358368378388398408418428438448458468478488498508518528538548558568578588598608618628638648658668678688698708718728738748758768778788798808818828838848858868878888898908918928938948958968978988999009019029039049059069079089099109119129139149159169179189199209219229239249259269279289299309319329339349359369379389399409419429439449459469479489499509519529539549559569579589599609619629639649659669679689699709719729739749759769779789799809819829839849859869879889899909919929939949959969979989991000100110021003100410051006100710081009101010111012101310141015
  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/).
  9. The approach follows that taken for 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. * ``FlexiBLAS``
  22. * ``OpenBLAS``
  23. * ``FLAME``
  24. * ``ATLAS PhiPACK``
  25. * ``CXML``
  26. * ``DXML``
  27. * ``SunPerf``
  28. * ``SCSL``
  29. * ``SGIMATH``
  30. * ``IBMESSL``
  31. * ``Intel10_32`` (intel mkl v10 32 bit, threaded code)
  32. * ``Intel10_64lp`` (intel mkl v10+ 64 bit, threaded code, lp64 model)
  33. * ``Intel10_64lp_seq`` (intel mkl v10+ 64 bit, sequential code, lp64 model)
  34. * ``Intel10_64ilp`` (intel mkl v10+ 64 bit, threaded code, ilp64 model)
  35. * ``Intel10_64ilp_seq`` (intel mkl v10+ 64 bit, sequential code, ilp64 model)
  36. * ``Intel10_64_dyn`` (intel mkl v10+ 64 bit, single dynamic library)
  37. * ``Intel`` (obsolete versions of mkl 32 and 64 bit)
  38. * ``ACML``
  39. * ``ACML_MP``
  40. * ``ACML_GPU``
  41. * ``Apple``
  42. * ``NAS``
  43. * ``Arm``
  44. * ``Arm_mp``
  45. * ``Arm_ilp64``
  46. * ``Arm_ilp64_mp``
  47. * ``Generic``
  48. .. versionadded:: 3.6
  49. ``OpenBLAS`` support.
  50. .. versionadded:: 3.11
  51. ``FLAME`` support.
  52. .. versionadded:: 3.13
  53. Added ILP64 MKL variants (``Intel10_64ilp``, ``Intel10_64ilp_seq``).
  54. .. versionadded:: 3.17
  55. Added single dynamic library MKL variant (``Intel10_64_dyn``).
  56. .. versionadded:: 3.18
  57. Arm Performance Libraries support (``Arm``, ``Arm_mp``, ``Arm_ilp64``,
  58. ``Arm_ilp64_mp``).
  59. .. versionadded:: 3.19
  60. ``FlexiBLAS`` support.
  61. ``BLA_F95``
  62. if ``ON`` tries to find the BLAS95 interfaces
  63. ``BLA_PREFER_PKGCONFIG``
  64. .. versionadded:: 3.11
  65. if set ``pkg-config`` will be used to search for a BLAS library first
  66. and if one is found that is preferred
  67. Imported targets
  68. ^^^^^^^^^^^^^^^^
  69. .. versionadded:: 3.18
  70. This module defines the following :prop_tgt:`IMPORTED` target:
  71. ``BLAS::BLAS``
  72. The libraries to use for BLAS, if found.
  73. Result Variables
  74. ^^^^^^^^^^^^^^^^
  75. This module defines the following variables:
  76. ``BLAS_FOUND``
  77. library implementing the BLAS interface is found
  78. ``BLAS_LINKER_FLAGS``
  79. uncached list of required linker flags (excluding ``-l`` and ``-L``).
  80. ``BLAS_LIBRARIES``
  81. uncached list of libraries (using full path name) to link against
  82. to use BLAS (may be empty if compiler implicitly links BLAS)
  83. ``BLAS95_LIBRARIES``
  84. uncached list of libraries (using full path name) to link against
  85. to use BLAS95 interface
  86. ``BLAS95_FOUND``
  87. library implementing the BLAS95 interface is found
  88. .. note::
  89. C, CXX or Fortran must be enabled to detect a BLAS library.
  90. C or CXX must be enabled to use Intel Math Kernel Library (MKL).
  91. For example, to use Intel MKL libraries and/or Intel compiler:
  92. .. code-block:: cmake
  93. set(BLA_VENDOR Intel10_64lp)
  94. find_package(BLAS)
  95. Hints
  96. ^^^^^
  97. ``MKLROOT``
  98. .. versionadded:: 3.15
  99. Set this environment variable to a directory that contains an MKL
  100. installation, or add the directory to the dynamic library loader environment
  101. variable for your platform (``LIB``, ``DYLD_LIBRARY_PATH`` or
  102. ``LD_LIBRARY_PATH``).
  103. #]=======================================================================]
  104. # Check the language being used
  105. if(NOT (CMAKE_C_COMPILER_LOADED OR CMAKE_CXX_COMPILER_LOADED OR CMAKE_Fortran_COMPILER_LOADED))
  106. if(BLAS_FIND_REQUIRED)
  107. message(FATAL_ERROR "FindBLAS requires Fortran, C, or C++ to be enabled.")
  108. else()
  109. message(STATUS "Looking for BLAS... - NOT found (Unsupported languages)")
  110. return()
  111. endif()
  112. endif()
  113. function(_add_blas_target)
  114. if(NOT TARGET BLAS::BLAS)
  115. add_library(BLAS::BLAS INTERFACE IMPORTED)
  116. if(BLAS_LIBRARIES)
  117. set_target_properties(BLAS::BLAS PROPERTIES
  118. INTERFACE_LINK_LIBRARIES "${BLAS_LIBRARIES}"
  119. )
  120. endif()
  121. endif()
  122. endfunction()
  123. if(CMAKE_Fortran_COMPILER_LOADED)
  124. include(${CMAKE_CURRENT_LIST_DIR}/CheckFortranFunctionExists.cmake)
  125. else()
  126. include(${CMAKE_CURRENT_LIST_DIR}/CheckFunctionExists.cmake)
  127. endif()
  128. include(${CMAKE_CURRENT_LIST_DIR}/CMakePushCheckState.cmake)
  129. include(${CMAKE_CURRENT_LIST_DIR}/FindPackageHandleStandardArgs.cmake)
  130. cmake_push_check_state()
  131. set(CMAKE_REQUIRED_QUIET ${BLAS_FIND_QUIETLY})
  132. if(BLA_PREFER_PKGCONFIG)
  133. find_package(PkgConfig)
  134. pkg_check_modules(PKGC_BLAS blas)
  135. if(PKGC_BLAS_FOUND)
  136. set(BLAS_FOUND ${PKGC_BLAS_FOUND})
  137. set(BLAS_LIBRARIES "${PKGC_BLAS_LINK_LIBRARIES}")
  138. _add_blas_target()
  139. return()
  140. endif()
  141. endif()
  142. set(_blas_ORIG_CMAKE_FIND_LIBRARY_SUFFIXES ${CMAKE_FIND_LIBRARY_SUFFIXES})
  143. if(BLA_STATIC)
  144. if(WIN32)
  145. set(CMAKE_FIND_LIBRARY_SUFFIXES .lib ${CMAKE_FIND_LIBRARY_SUFFIXES})
  146. else()
  147. set(CMAKE_FIND_LIBRARY_SUFFIXES .a ${CMAKE_FIND_LIBRARY_SUFFIXES})
  148. endif()
  149. else()
  150. if(CMAKE_SYSTEM_NAME STREQUAL "Linux")
  151. # for ubuntu's libblas3gf and liblapack3gf packages
  152. set(CMAKE_FIND_LIBRARY_SUFFIXES ${CMAKE_FIND_LIBRARY_SUFFIXES} .so.3gf)
  153. endif()
  154. endif()
  155. # TODO: move this stuff to a separate module
  156. macro(CHECK_BLAS_LIBRARIES LIBRARIES _prefix _name _flags _list _threadlibs _addlibdir _subdirs)
  157. # This macro checks for the existence of the combination of fortran libraries
  158. # given by _list. If the combination is found, this macro checks (using the
  159. # Check_Fortran_Function_Exists macro) whether can link against that library
  160. # combination using the name of a routine given by _name using the linker
  161. # flags given by _flags. If the combination of libraries is found and passes
  162. # the link test, LIBRARIES is set to the list of complete library paths that
  163. # have been found. Otherwise, LIBRARIES is set to FALSE.
  164. # N.B. _prefix is the prefix applied to the names of all cached variables that
  165. # are generated internally and marked advanced by this macro.
  166. # _addlibdir is a list of additional search paths. _subdirs is a list of path
  167. # suffixes to be used by find_library().
  168. set(_libraries_work TRUE)
  169. set(${LIBRARIES})
  170. set(_combined_name)
  171. set(_extaddlibdir "${_addlibdir}")
  172. if(WIN32)
  173. list(APPEND _extaddlibdir ENV LIB)
  174. elseif(APPLE)
  175. list(APPEND _extaddlibdir ENV DYLD_LIBRARY_PATH)
  176. else()
  177. list(APPEND _extaddlibdir ENV LD_LIBRARY_PATH)
  178. endif()
  179. list(APPEND _extaddlibdir "${CMAKE_C_IMPLICIT_LINK_DIRECTORIES}")
  180. foreach(_library ${_list})
  181. if(_library MATCHES "^-Wl,--(start|end)-group$")
  182. # Respect linker flags like --start/end-group (required by MKL)
  183. set(${LIBRARIES} ${${LIBRARIES}} "${_library}")
  184. else()
  185. set(_combined_name ${_combined_name}_${_library})
  186. if(NOT "${_threadlibs}" STREQUAL "")
  187. set(_combined_name ${_combined_name}_threadlibs)
  188. endif()
  189. if(_libraries_work)
  190. find_library(${_prefix}_${_library}_LIBRARY
  191. NAMES ${_library}
  192. NAMES_PER_DIR
  193. PATHS ${_extaddlibdir}
  194. PATH_SUFFIXES ${_subdirs}
  195. )
  196. #message("DEBUG: find_library(${_library}) got ${${_prefix}_${_library}_LIBRARY}")
  197. mark_as_advanced(${_prefix}_${_library}_LIBRARY)
  198. set(${LIBRARIES} ${${LIBRARIES}} ${${_prefix}_${_library}_LIBRARY})
  199. set(_libraries_work ${${_prefix}_${_library}_LIBRARY})
  200. endif()
  201. endif()
  202. endforeach()
  203. if(_libraries_work)
  204. # Test this combination of libraries.
  205. set(CMAKE_REQUIRED_LIBRARIES ${_flags} ${${LIBRARIES}} ${_threadlibs})
  206. #message("DEBUG: CMAKE_REQUIRED_LIBRARIES = ${CMAKE_REQUIRED_LIBRARIES}")
  207. if(CMAKE_Fortran_COMPILER_LOADED)
  208. check_fortran_function_exists("${_name}" ${_prefix}${_combined_name}_WORKS)
  209. else()
  210. check_function_exists("${_name}_" ${_prefix}${_combined_name}_WORKS)
  211. endif()
  212. set(CMAKE_REQUIRED_LIBRARIES)
  213. set(_libraries_work ${${_prefix}${_combined_name}_WORKS})
  214. endif()
  215. if(_libraries_work)
  216. if("${_list}" STREQUAL "")
  217. set(${LIBRARIES} "${LIBRARIES}-PLACEHOLDER-FOR-EMPTY-LIBRARIES")
  218. else()
  219. set(${LIBRARIES} ${${LIBRARIES}} ${_threadlibs})
  220. endif()
  221. else()
  222. set(${LIBRARIES} FALSE)
  223. endif()
  224. #message("DEBUG: ${LIBRARIES} = ${${LIBRARIES}}")
  225. endmacro()
  226. set(BLAS_LINKER_FLAGS)
  227. set(BLAS_LIBRARIES)
  228. set(BLAS95_LIBRARIES)
  229. if(NOT $ENV{BLA_VENDOR} STREQUAL "")
  230. set(BLA_VENDOR $ENV{BLA_VENDOR})
  231. else()
  232. if(NOT BLA_VENDOR)
  233. set(BLA_VENDOR "All")
  234. endif()
  235. endif()
  236. # Implicitly linked BLAS libraries?
  237. if(BLA_VENDOR STREQUAL "All")
  238. if(NOT BLAS_LIBRARIES)
  239. check_blas_libraries(
  240. BLAS_LIBRARIES
  241. BLAS
  242. sgemm
  243. ""
  244. ""
  245. ""
  246. ""
  247. ""
  248. )
  249. endif()
  250. endif()
  251. # BLAS in the Intel MKL 10+ library?
  252. if(BLA_VENDOR MATCHES "Intel" OR BLA_VENDOR STREQUAL "All")
  253. if(NOT BLAS_LIBRARIES)
  254. if(CMAKE_C_COMPILER_LOADED OR CMAKE_CXX_COMPILER_LOADED)
  255. # System-specific settings
  256. if(WIN32)
  257. if(BLA_STATIC)
  258. set(BLAS_mkl_DLL_SUFFIX "")
  259. else()
  260. set(BLAS_mkl_DLL_SUFFIX "_dll")
  261. endif()
  262. else()
  263. if(BLA_STATIC)
  264. set(BLAS_mkl_START_GROUP "-Wl,--start-group")
  265. set(BLAS_mkl_END_GROUP "-Wl,--end-group")
  266. else()
  267. set(BLAS_mkl_START_GROUP "")
  268. set(BLAS_mkl_END_GROUP "")
  269. endif()
  270. # Switch to GNU Fortran support layer if needed (but not on Apple, where MKL does not provide it)
  271. if(CMAKE_Fortran_COMPILER_LOADED AND CMAKE_Fortran_COMPILER_ID STREQUAL "GNU" AND NOT APPLE)
  272. set(BLAS_mkl_INTFACE "gf")
  273. set(BLAS_mkl_THREADING "gnu")
  274. set(BLAS_mkl_OMP "gomp")
  275. else()
  276. set(BLAS_mkl_INTFACE "intel")
  277. set(BLAS_mkl_THREADING "intel")
  278. set(BLAS_mkl_OMP "iomp5")
  279. endif()
  280. set(BLAS_mkl_LM "-lm")
  281. set(BLAS_mkl_LDL "-ldl")
  282. endif()
  283. if(BLAS_FIND_QUIETLY OR NOT BLAS_FIND_REQUIRED)
  284. find_package(Threads)
  285. else()
  286. find_package(Threads REQUIRED)
  287. endif()
  288. if(BLA_VENDOR MATCHES "_64ilp")
  289. set(BLAS_mkl_ILP_MODE "ilp64")
  290. else()
  291. set(BLAS_mkl_ILP_MODE "lp64")
  292. endif()
  293. set(BLAS_SEARCH_LIBS "")
  294. if(BLA_F95)
  295. set(BLAS_mkl_SEARCH_SYMBOL "sgemm_f95")
  296. set(_LIBRARIES BLAS95_LIBRARIES)
  297. if(WIN32)
  298. # Find the main file (32-bit or 64-bit)
  299. set(BLAS_SEARCH_LIBS_WIN_MAIN "")
  300. if(BLA_VENDOR STREQUAL "Intel10_32" OR BLA_VENDOR STREQUAL "All")
  301. list(APPEND BLAS_SEARCH_LIBS_WIN_MAIN
  302. "mkl_blas95${BLAS_mkl_DLL_SUFFIX} mkl_intel_c${BLAS_mkl_DLL_SUFFIX}")
  303. endif()
  304. if(BLA_VENDOR MATCHES "^Intel10_64i?lp" OR BLA_VENDOR STREQUAL "All")
  305. list(APPEND BLAS_SEARCH_LIBS_WIN_MAIN
  306. "mkl_blas95_${BLAS_mkl_ILP_MODE}${BLAS_mkl_DLL_SUFFIX} mkl_intel_${BLAS_mkl_ILP_MODE}${BLAS_mkl_DLL_SUFFIX}")
  307. endif()
  308. # Add threading/sequential libs
  309. set(BLAS_SEARCH_LIBS_WIN_THREAD "")
  310. if(BLA_VENDOR MATCHES "^Intel10_64i?lp$" OR BLA_VENDOR STREQUAL "All")
  311. # old version
  312. list(APPEND BLAS_SEARCH_LIBS_WIN_THREAD
  313. "libguide40 mkl_intel_thread${BLAS_mkl_DLL_SUFFIX}")
  314. # mkl >= 10.3
  315. list(APPEND BLAS_SEARCH_LIBS_WIN_THREAD
  316. "libiomp5md mkl_intel_thread${BLAS_mkl_DLL_SUFFIX}")
  317. endif()
  318. if(BLA_VENDOR MATCHES "^Intel10_64i?lp_seq$" OR BLA_VENDOR STREQUAL "All")
  319. list(APPEND BLAS_SEARCH_LIBS_WIN_THREAD
  320. "mkl_sequential${BLAS_mkl_DLL_SUFFIX}")
  321. endif()
  322. # Cartesian product of the above
  323. foreach(MAIN ${BLAS_SEARCH_LIBS_WIN_MAIN})
  324. foreach(THREAD ${BLAS_SEARCH_LIBS_WIN_THREAD})
  325. list(APPEND BLAS_SEARCH_LIBS
  326. "${MAIN} ${THREAD} mkl_core${BLAS_mkl_DLL_SUFFIX}")
  327. endforeach()
  328. endforeach()
  329. else()
  330. if(BLA_VENDOR STREQUAL "Intel10_32" OR BLA_VENDOR STREQUAL "All")
  331. # old version
  332. list(APPEND BLAS_SEARCH_LIBS
  333. "mkl_blas95 mkl_${BLAS_mkl_INTFACE} mkl_${BLAS_mkl_THREADING}_thread mkl_core guide")
  334. # mkl >= 10.3
  335. list(APPEND BLAS_SEARCH_LIBS
  336. "${BLAS_mkl_START_GROUP} mkl_blas95 mkl_${BLAS_mkl_INTFACE} mkl_${BLAS_mkl_THREADING}_thread mkl_core ${BLAS_mkl_END_GROUP} ${BLAS_mkl_OMP}")
  337. endif()
  338. if(BLA_VENDOR MATCHES "^Intel10_64i?lp$" OR BLA_VENDOR STREQUAL "All")
  339. # old version
  340. list(APPEND BLAS_SEARCH_LIBS
  341. "mkl_blas95 mkl_${BLAS_mkl_INTFACE}_${BLAS_mkl_ILP_MODE} mkl_${BLAS_mkl_THREADING}_thread mkl_core guide")
  342. # mkl >= 10.3
  343. list(APPEND BLAS_SEARCH_LIBS
  344. "${BLAS_mkl_START_GROUP} mkl_blas95_${BLAS_mkl_ILP_MODE} mkl_${BLAS_mkl_INTFACE}_${BLAS_mkl_ILP_MODE} mkl_${BLAS_mkl_THREADING}_thread mkl_core ${BLAS_mkl_END_GROUP} ${BLAS_mkl_OMP}")
  345. endif()
  346. if(BLA_VENDOR MATCHES "^Intel10_64i?lp_seq$" OR BLA_VENDOR STREQUAL "All")
  347. list(APPEND BLAS_SEARCH_LIBS
  348. "${BLAS_mkl_START_GROUP} mkl_blas95_${BLAS_mkl_ILP_MODE} mkl_${BLAS_mkl_INTFACE}_${BLAS_mkl_ILP_MODE} mkl_sequential mkl_core ${BLAS_mkl_END_GROUP}")
  349. endif()
  350. endif()
  351. else()
  352. set(BLAS_mkl_SEARCH_SYMBOL sgemm)
  353. set(_LIBRARIES BLAS_LIBRARIES)
  354. if(WIN32)
  355. # Find the main file (32-bit or 64-bit)
  356. set(BLAS_SEARCH_LIBS_WIN_MAIN "")
  357. if(BLA_VENDOR STREQUAL "Intel10_32" OR BLA_VENDOR STREQUAL "All")
  358. list(APPEND BLAS_SEARCH_LIBS_WIN_MAIN
  359. "mkl_intel_c${BLAS_mkl_DLL_SUFFIX}")
  360. endif()
  361. if(BLA_VENDOR MATCHES "^Intel10_64i?lp" OR BLA_VENDOR STREQUAL "All")
  362. list(APPEND BLAS_SEARCH_LIBS_WIN_MAIN
  363. "mkl_intel_${BLAS_mkl_ILP_MODE}${BLAS_mkl_DLL_SUFFIX}")
  364. endif()
  365. # Add threading/sequential libs
  366. set(BLAS_SEARCH_LIBS_WIN_THREAD "")
  367. if(BLA_VENDOR STREQUAL "Intel10_32" OR BLA_VENDOR STREQUAL "All")
  368. list(APPEND BLAS_SEARCH_LIBS_WIN_THREAD
  369. "libiomp5md mkl_intel_thread${BLAS_mkl_DLL_SUFFIX}")
  370. endif()
  371. if(BLA_VENDOR MATCHES "^Intel10_64i?lp$" OR BLA_VENDOR STREQUAL "All")
  372. # old version
  373. list(APPEND BLAS_SEARCH_LIBS_WIN_THREAD
  374. "libguide40 mkl_intel_thread${BLAS_mkl_DLL_SUFFIX}")
  375. # mkl >= 10.3
  376. list(APPEND BLAS_SEARCH_LIBS_WIN_THREAD
  377. "libiomp5md mkl_intel_thread${BLAS_mkl_DLL_SUFFIX}")
  378. endif()
  379. if(BLA_VENDOR MATCHES "^Intel10_64i?lp_seq$" OR BLA_VENDOR STREQUAL "All")
  380. list(APPEND BLAS_SEARCH_LIBS_WIN_THREAD
  381. "mkl_sequential${BLAS_mkl_DLL_SUFFIX}")
  382. endif()
  383. # Cartesian product of the above
  384. foreach(MAIN ${BLAS_SEARCH_LIBS_WIN_MAIN})
  385. foreach(THREAD ${BLAS_SEARCH_LIBS_WIN_THREAD})
  386. list(APPEND BLAS_SEARCH_LIBS
  387. "${MAIN} ${THREAD} mkl_core${BLAS_mkl_DLL_SUFFIX}")
  388. endforeach()
  389. endforeach()
  390. else()
  391. if(BLA_VENDOR STREQUAL "Intel10_32" OR BLA_VENDOR STREQUAL "All")
  392. # old version
  393. list(APPEND BLAS_SEARCH_LIBS
  394. "mkl_${BLAS_mkl_INTFACE} mkl_${BLAS_mkl_THREADING}_thread mkl_core guide")
  395. # mkl >= 10.3
  396. list(APPEND BLAS_SEARCH_LIBS
  397. "${BLAS_mkl_START_GROUP} mkl_${BLAS_mkl_INTFACE} mkl_${BLAS_mkl_THREADING}_thread mkl_core ${BLAS_mkl_END_GROUP} ${BLAS_mkl_OMP}")
  398. endif()
  399. if(BLA_VENDOR MATCHES "^Intel10_64i?lp$" OR BLA_VENDOR STREQUAL "All")
  400. # old version
  401. list(APPEND BLAS_SEARCH_LIBS
  402. "mkl_${BLAS_mkl_INTFACE}_${BLAS_mkl_ILP_MODE} mkl_${BLAS_mkl_THREADING}_thread mkl_core guide")
  403. # mkl >= 10.3
  404. list(APPEND BLAS_SEARCH_LIBS
  405. "${BLAS_mkl_START_GROUP} mkl_${BLAS_mkl_INTFACE}_${BLAS_mkl_ILP_MODE} mkl_${BLAS_mkl_THREADING}_thread mkl_core ${BLAS_mkl_END_GROUP} ${BLAS_mkl_OMP}")
  406. endif()
  407. if(BLA_VENDOR MATCHES "^Intel10_64i?lp_seq$" OR BLA_VENDOR STREQUAL "All")
  408. list(APPEND BLAS_SEARCH_LIBS
  409. "${BLAS_mkl_START_GROUP} mkl_${BLAS_mkl_INTFACE}_${BLAS_mkl_ILP_MODE} mkl_sequential mkl_core ${BLAS_mkl_END_GROUP}")
  410. endif()
  411. #older versions of intel mkl libs
  412. if(BLA_VENDOR STREQUAL "Intel" OR BLA_VENDOR STREQUAL "All")
  413. list(APPEND BLAS_SEARCH_LIBS
  414. "mkl")
  415. list(APPEND BLAS_SEARCH_LIBS
  416. "mkl_ia32")
  417. list(APPEND BLAS_SEARCH_LIBS
  418. "mkl_em64t")
  419. endif()
  420. endif()
  421. endif()
  422. if(BLA_VENDOR MATCHES "^Intel10_64_dyn$" OR BLA_VENDOR STREQUAL "All")
  423. # mkl >= 10.3 with single dynamic library
  424. list(APPEND BLAS_SEARCH_LIBS
  425. "mkl_rt")
  426. endif()
  427. # MKL uses a multitude of partially platform-specific subdirectories:
  428. if(BLA_VENDOR STREQUAL "Intel10_32")
  429. set(BLAS_mkl_ARCH_NAME "ia32")
  430. else()
  431. set(BLAS_mkl_ARCH_NAME "intel64")
  432. endif()
  433. if(WIN32)
  434. set(BLAS_mkl_OS_NAME "win")
  435. elseif(APPLE)
  436. set(BLAS_mkl_OS_NAME "mac")
  437. else()
  438. set(BLAS_mkl_OS_NAME "lin")
  439. endif()
  440. if(DEFINED ENV{MKLROOT})
  441. file(TO_CMAKE_PATH "$ENV{MKLROOT}" BLAS_mkl_MKLROOT)
  442. # If MKLROOT points to the subdirectory 'mkl', use the parent directory instead
  443. # so we can better detect other relevant libraries in 'compiler' or 'tbb':
  444. get_filename_component(BLAS_mkl_MKLROOT_LAST_DIR "${BLAS_mkl_MKLROOT}" NAME)
  445. if(BLAS_mkl_MKLROOT_LAST_DIR STREQUAL "mkl")
  446. get_filename_component(BLAS_mkl_MKLROOT "${BLAS_mkl_MKLROOT}" DIRECTORY)
  447. endif()
  448. endif()
  449. set(BLAS_mkl_LIB_PATH_SUFFIXES
  450. "compiler/lib" "compiler/lib/${BLAS_mkl_ARCH_NAME}_${BLAS_mkl_OS_NAME}"
  451. "compiler/lib/${BLAS_mkl_ARCH_NAME}"
  452. "mkl/lib" "mkl/lib/${BLAS_mkl_ARCH_NAME}_${BLAS_mkl_OS_NAME}"
  453. "mkl/lib/${BLAS_mkl_ARCH_NAME}"
  454. "lib/${BLAS_mkl_ARCH_NAME}_${BLAS_mkl_OS_NAME}")
  455. foreach(IT ${BLAS_SEARCH_LIBS})
  456. string(REPLACE " " ";" SEARCH_LIBS ${IT})
  457. if(NOT ${_LIBRARIES})
  458. check_blas_libraries(
  459. ${_LIBRARIES}
  460. BLAS
  461. ${BLAS_mkl_SEARCH_SYMBOL}
  462. ""
  463. "${SEARCH_LIBS}"
  464. "${CMAKE_THREAD_LIBS_INIT};${BLAS_mkl_LM};${BLAS_mkl_LDL}"
  465. "${BLAS_mkl_MKLROOT}"
  466. "${BLAS_mkl_LIB_PATH_SUFFIXES}"
  467. )
  468. endif()
  469. endforeach()
  470. unset(BLAS_mkl_ILP_MODE)
  471. unset(BLAS_mkl_INTFACE)
  472. unset(BLAS_mkl_THREADING)
  473. unset(BLAS_mkl_OMP)
  474. unset(BLAS_mkl_DLL_SUFFIX)
  475. unset(BLAS_mkl_LM)
  476. unset(BLAS_mkl_LDL)
  477. unset(BLAS_mkl_MKLROOT)
  478. unset(BLAS_mkl_MKLROOT_LAST_DIR)
  479. unset(BLAS_mkl_ARCH_NAME)
  480. unset(BLAS_mkl_OS_NAME)
  481. unset(BLAS_mkl_LIB_PATH_SUFFIXES)
  482. endif()
  483. endif()
  484. endif()
  485. if(BLA_F95)
  486. find_package_handle_standard_args(BLAS REQUIRED_VARS BLAS95_LIBRARIES)
  487. set(BLAS95_FOUND ${BLAS_FOUND})
  488. if(BLAS_FOUND)
  489. set(BLAS_LIBRARIES "${BLAS95_LIBRARIES}")
  490. endif()
  491. endif()
  492. # gotoblas? (http://www.tacc.utexas.edu/tacc-projects/gotoblas2)
  493. if(BLA_VENDOR STREQUAL "Goto" OR BLA_VENDOR STREQUAL "All")
  494. if(NOT BLAS_LIBRARIES)
  495. check_blas_libraries(
  496. BLAS_LIBRARIES
  497. BLAS
  498. sgemm
  499. ""
  500. "goto2"
  501. ""
  502. ""
  503. ""
  504. )
  505. endif()
  506. endif()
  507. # FlexiBLAS? (http://www.mpi-magdeburg.mpg.de/mpcsc/software/FlexiBLAS/)
  508. if(BLA_VENDOR STREQUAL "FlexiBLAS" OR BLA_VENDOR STREQUAL "All")
  509. if(NOT BLAS_LIBRARIES)
  510. check_blas_libraries(
  511. BLAS_LIBRARIES
  512. BLAS
  513. sgemm
  514. ""
  515. "flexiblas"
  516. ""
  517. ""
  518. ""
  519. )
  520. endif()
  521. endif()
  522. # OpenBLAS? (http://www.openblas.net)
  523. if(BLA_VENDOR STREQUAL "OpenBLAS" OR BLA_VENDOR STREQUAL "All")
  524. if(NOT BLAS_LIBRARIES)
  525. check_blas_libraries(
  526. BLAS_LIBRARIES
  527. BLAS
  528. sgemm
  529. ""
  530. "openblas"
  531. ""
  532. ""
  533. ""
  534. )
  535. endif()
  536. if(NOT BLAS_LIBRARIES AND (CMAKE_C_COMPILER_LOADED OR CMAKE_CXX_COMPILER_LOADED))
  537. if(BLAS_FIND_QUIETLY OR NOT BLAS_FIND_REQUIRED)
  538. find_package(Threads)
  539. else()
  540. find_package(Threads REQUIRED)
  541. endif()
  542. set(_threadlibs "${CMAKE_THREAD_LIBS_INIT}")
  543. if(BLA_STATIC)
  544. find_package(OpenMP COMPONENTS C)
  545. list(PREPEND _threadlibs "${OpenMP_C_LIBRARIES}")
  546. endif()
  547. check_blas_libraries(
  548. BLAS_LIBRARIES
  549. BLAS
  550. sgemm
  551. ""
  552. "openblas"
  553. "${_threadlibs}"
  554. ""
  555. ""
  556. )
  557. unset(_threadlibs)
  558. endif()
  559. endif()
  560. # ArmPL blas library? (https://developer.arm.com/tools-and-software/server-and-hpc/compile/arm-compiler-for-linux/arm-performance-libraries)
  561. if(BLA_VENDOR MATCHES "Arm" OR BLA_VENDOR STREQUAL "All")
  562. # Check for 64bit Integer support
  563. if(BLA_VENDOR MATCHES "_ilp64")
  564. set(BLAS_armpl_LIB "armpl_ilp64")
  565. else()
  566. set(BLAS_armpl_LIB "armpl_lp64")
  567. endif()
  568. # Check for OpenMP support, VIA BLA_VENDOR of Arm_mp or Arm_ipl64_mp
  569. if(BLA_VENDOR MATCHES "_mp")
  570. set(BLAS_armpl_LIB "${BLAS_armpl_LIB}_mp")
  571. endif()
  572. if(NOT BLAS_LIBRARIES)
  573. check_blas_libraries(
  574. BLAS_LIBRARIES
  575. BLAS
  576. sgemm
  577. ""
  578. "${BLAS_armpl_LIB}"
  579. ""
  580. ""
  581. ""
  582. )
  583. endif()
  584. endif()
  585. # FLAME's blis library? (https://github.com/flame/blis)
  586. if(BLA_VENDOR STREQUAL "FLAME" OR BLA_VENDOR STREQUAL "All")
  587. if(NOT BLAS_LIBRARIES)
  588. check_blas_libraries(
  589. BLAS_LIBRARIES
  590. BLAS
  591. sgemm
  592. ""
  593. "blis"
  594. ""
  595. ""
  596. ""
  597. )
  598. endif()
  599. endif()
  600. # BLAS in the ATLAS library? (http://math-atlas.sourceforge.net/)
  601. if(BLA_VENDOR STREQUAL "ATLAS" OR BLA_VENDOR STREQUAL "All")
  602. if(NOT BLAS_LIBRARIES)
  603. check_blas_libraries(
  604. BLAS_LIBRARIES
  605. BLAS
  606. dgemm
  607. ""
  608. "blas;f77blas;atlas"
  609. ""
  610. ""
  611. ""
  612. )
  613. endif()
  614. endif()
  615. # BLAS in PhiPACK libraries? (requires generic BLAS lib, too)
  616. if(BLA_VENDOR STREQUAL "PhiPACK" OR BLA_VENDOR STREQUAL "All")
  617. if(NOT BLAS_LIBRARIES)
  618. check_blas_libraries(
  619. BLAS_LIBRARIES
  620. BLAS
  621. sgemm
  622. ""
  623. "sgemm;dgemm;blas"
  624. ""
  625. ""
  626. ""
  627. )
  628. endif()
  629. endif()
  630. # BLAS in Alpha CXML library?
  631. if(BLA_VENDOR STREQUAL "CXML" OR BLA_VENDOR STREQUAL "All")
  632. if(NOT BLAS_LIBRARIES)
  633. check_blas_libraries(
  634. BLAS_LIBRARIES
  635. BLAS
  636. sgemm
  637. ""
  638. "cxml"
  639. ""
  640. ""
  641. ""
  642. )
  643. endif()
  644. endif()
  645. # BLAS in Alpha DXML library? (now called CXML, see above)
  646. if(BLA_VENDOR STREQUAL "DXML" OR BLA_VENDOR STREQUAL "All")
  647. if(NOT BLAS_LIBRARIES)
  648. check_blas_libraries(
  649. BLAS_LIBRARIES
  650. BLAS
  651. sgemm
  652. ""
  653. "dxml"
  654. ""
  655. ""
  656. ""
  657. )
  658. endif()
  659. endif()
  660. # BLAS in Sun Performance library?
  661. if(BLA_VENDOR STREQUAL "SunPerf" OR BLA_VENDOR STREQUAL "All")
  662. if(NOT BLAS_LIBRARIES)
  663. check_blas_libraries(
  664. BLAS_LIBRARIES
  665. BLAS
  666. sgemm
  667. "-xlic_lib=sunperf"
  668. "sunperf;sunmath"
  669. ""
  670. ""
  671. ""
  672. )
  673. if(BLAS_LIBRARIES)
  674. set(BLAS_LINKER_FLAGS "-xlic_lib=sunperf")
  675. endif()
  676. endif()
  677. endif()
  678. # BLAS in SCSL library? (SGI/Cray Scientific Library)
  679. if(BLA_VENDOR STREQUAL "SCSL" OR BLA_VENDOR STREQUAL "All")
  680. if(NOT BLAS_LIBRARIES)
  681. check_blas_libraries(
  682. BLAS_LIBRARIES
  683. BLAS
  684. sgemm
  685. ""
  686. "scsl"
  687. ""
  688. ""
  689. ""
  690. )
  691. endif()
  692. endif()
  693. # BLAS in SGIMATH library?
  694. if(BLA_VENDOR STREQUAL "SGIMATH" OR BLA_VENDOR STREQUAL "All")
  695. if(NOT BLAS_LIBRARIES)
  696. check_blas_libraries(
  697. BLAS_LIBRARIES
  698. BLAS
  699. sgemm
  700. ""
  701. "complib.sgimath"
  702. ""
  703. ""
  704. ""
  705. )
  706. endif()
  707. endif()
  708. # BLAS in IBM ESSL library? (requires generic BLAS lib, too)
  709. if(BLA_VENDOR STREQUAL "IBMESSL" OR BLA_VENDOR STREQUAL "All")
  710. if(NOT BLAS_LIBRARIES)
  711. check_blas_libraries(
  712. BLAS_LIBRARIES
  713. BLAS
  714. sgemm
  715. ""
  716. "essl;blas"
  717. ""
  718. ""
  719. ""
  720. )
  721. endif()
  722. endif()
  723. # BLAS in acml library?
  724. if(BLA_VENDOR MATCHES "ACML" OR BLA_VENDOR STREQUAL "All")
  725. if(((BLA_VENDOR STREQUAL "ACML") AND (NOT BLAS_ACML_LIB_DIRS)) OR
  726. ((BLA_VENDOR STREQUAL "ACML_MP") AND (NOT BLAS_ACML_MP_LIB_DIRS)) OR
  727. ((BLA_VENDOR STREQUAL "ACML_GPU") AND (NOT BLAS_ACML_GPU_LIB_DIRS))
  728. )
  729. # try to find acml in "standard" paths
  730. if(WIN32)
  731. file(GLOB _ACML_ROOT "C:/AMD/acml*/ACML-EULA.txt")
  732. else()
  733. file(GLOB _ACML_ROOT "/opt/acml*/ACML-EULA.txt")
  734. endif()
  735. if(WIN32)
  736. file(GLOB _ACML_GPU_ROOT "C:/AMD/acml*/GPGPUexamples")
  737. else()
  738. file(GLOB _ACML_GPU_ROOT "/opt/acml*/GPGPUexamples")
  739. endif()
  740. list(GET _ACML_ROOT 0 _ACML_ROOT)
  741. list(GET _ACML_GPU_ROOT 0 _ACML_GPU_ROOT)
  742. if(_ACML_ROOT)
  743. get_filename_component(_ACML_ROOT ${_ACML_ROOT} PATH)
  744. if(SIZEOF_INTEGER EQUAL 8)
  745. set(_ACML_PATH_SUFFIX "_int64")
  746. else()
  747. set(_ACML_PATH_SUFFIX "")
  748. endif()
  749. if(CMAKE_Fortran_COMPILER_ID STREQUAL "Intel")
  750. set(_ACML_COMPILER32 "ifort32")
  751. set(_ACML_COMPILER64 "ifort64")
  752. elseif(CMAKE_Fortran_COMPILER_ID STREQUAL "SunPro")
  753. set(_ACML_COMPILER32 "sun32")
  754. set(_ACML_COMPILER64 "sun64")
  755. elseif(CMAKE_Fortran_COMPILER_ID STREQUAL "PGI")
  756. set(_ACML_COMPILER32 "pgi32")
  757. if(WIN32)
  758. set(_ACML_COMPILER64 "win64")
  759. else()
  760. set(_ACML_COMPILER64 "pgi64")
  761. endif()
  762. elseif(CMAKE_Fortran_COMPILER_ID STREQUAL "Open64")
  763. # 32 bit builds not supported on Open64 but for code simplicity
  764. # We'll just use the same directory twice
  765. set(_ACML_COMPILER32 "open64_64")
  766. set(_ACML_COMPILER64 "open64_64")
  767. elseif(CMAKE_Fortran_COMPILER_ID STREQUAL "NAG")
  768. set(_ACML_COMPILER32 "nag32")
  769. set(_ACML_COMPILER64 "nag64")
  770. else()
  771. set(_ACML_COMPILER32 "gfortran32")
  772. set(_ACML_COMPILER64 "gfortran64")
  773. endif()
  774. if(BLA_VENDOR STREQUAL "ACML_MP")
  775. set(_ACML_MP_LIB_DIRS
  776. "${_ACML_ROOT}/${_ACML_COMPILER32}_mp${_ACML_PATH_SUFFIX}/lib"
  777. "${_ACML_ROOT}/${_ACML_COMPILER64}_mp${_ACML_PATH_SUFFIX}/lib")
  778. else()
  779. set(_ACML_LIB_DIRS
  780. "${_ACML_ROOT}/${_ACML_COMPILER32}${_ACML_PATH_SUFFIX}/lib"
  781. "${_ACML_ROOT}/${_ACML_COMPILER64}${_ACML_PATH_SUFFIX}/lib")
  782. endif()
  783. endif()
  784. elseif(BLAS_${BLA_VENDOR}_LIB_DIRS)
  785. set(_${BLA_VENDOR}_LIB_DIRS ${BLAS_${BLA_VENDOR}_LIB_DIRS})
  786. endif()
  787. if(BLA_VENDOR STREQUAL "ACML_MP")
  788. foreach(BLAS_ACML_MP_LIB_DIRS ${_ACML_MP_LIB_DIRS})
  789. check_blas_libraries(
  790. BLAS_LIBRARIES
  791. BLAS
  792. sgemm
  793. "" "acml_mp;acml_mv" "" ${BLAS_ACML_MP_LIB_DIRS} ""
  794. )
  795. if(BLAS_LIBRARIES)
  796. break()
  797. endif()
  798. endforeach()
  799. elseif(BLA_VENDOR STREQUAL "ACML_GPU")
  800. foreach(BLAS_ACML_GPU_LIB_DIRS ${_ACML_GPU_LIB_DIRS})
  801. check_blas_libraries(
  802. BLAS_LIBRARIES
  803. BLAS
  804. sgemm
  805. "" "acml;acml_mv;CALBLAS" "" ${BLAS_ACML_GPU_LIB_DIRS} ""
  806. )
  807. if(BLAS_LIBRARIES)
  808. break()
  809. endif()
  810. endforeach()
  811. else()
  812. foreach(BLAS_ACML_LIB_DIRS ${_ACML_LIB_DIRS})
  813. check_blas_libraries(
  814. BLAS_LIBRARIES
  815. BLAS
  816. sgemm
  817. "" "acml;acml_mv" "" ${BLAS_ACML_LIB_DIRS} ""
  818. )
  819. if(BLAS_LIBRARIES)
  820. break()
  821. endif()
  822. endforeach()
  823. endif()
  824. # Either acml or acml_mp should be in LD_LIBRARY_PATH but not both
  825. if(NOT BLAS_LIBRARIES)
  826. check_blas_libraries(
  827. BLAS_LIBRARIES
  828. BLAS
  829. sgemm
  830. ""
  831. "acml;acml_mv"
  832. ""
  833. ""
  834. ""
  835. )
  836. endif()
  837. if(NOT BLAS_LIBRARIES)
  838. check_blas_libraries(
  839. BLAS_LIBRARIES
  840. BLAS
  841. sgemm
  842. ""
  843. "acml_mp;acml_mv"
  844. ""
  845. ""
  846. ""
  847. )
  848. endif()
  849. if(NOT BLAS_LIBRARIES)
  850. check_blas_libraries(
  851. BLAS_LIBRARIES
  852. BLAS
  853. sgemm
  854. ""
  855. "acml;acml_mv;CALBLAS"
  856. ""
  857. ""
  858. ""
  859. )
  860. endif()
  861. endif() # ACML
  862. # Apple BLAS library?
  863. if(BLA_VENDOR STREQUAL "Apple" OR BLA_VENDOR STREQUAL "All")
  864. if(NOT BLAS_LIBRARIES)
  865. check_blas_libraries(
  866. BLAS_LIBRARIES
  867. BLAS
  868. dgemm
  869. ""
  870. "Accelerate"
  871. ""
  872. ""
  873. ""
  874. )
  875. endif()
  876. endif()
  877. # Apple NAS (vecLib) library?
  878. if(BLA_VENDOR STREQUAL "NAS" OR BLA_VENDOR STREQUAL "All")
  879. if(NOT BLAS_LIBRARIES)
  880. check_blas_libraries(
  881. BLAS_LIBRARIES
  882. BLAS
  883. dgemm
  884. ""
  885. "vecLib"
  886. ""
  887. ""
  888. ""
  889. )
  890. endif()
  891. endif()
  892. # Generic BLAS library?
  893. if(BLA_VENDOR STREQUAL "Generic" OR BLA_VENDOR STREQUAL "All")
  894. if(NOT BLAS_LIBRARIES)
  895. check_blas_libraries(
  896. BLAS_LIBRARIES
  897. BLAS
  898. sgemm
  899. ""
  900. "blas"
  901. ""
  902. ""
  903. ""
  904. )
  905. endif()
  906. endif()
  907. if(NOT BLA_F95)
  908. find_package_handle_standard_args(BLAS REQUIRED_VARS BLAS_LIBRARIES)
  909. endif()
  910. # On compilers that implicitly link BLAS (such as ftn, cc, and CC on Cray HPC machines)
  911. # we used a placeholder for empty BLAS_LIBRARIES to get through our logic above.
  912. if(BLAS_LIBRARIES STREQUAL "BLAS_LIBRARIES-PLACEHOLDER-FOR-EMPTY-LIBRARIES")
  913. set(BLAS_LIBRARIES "")
  914. endif()
  915. _add_blas_target()
  916. cmake_pop_check_state()
  917. set(CMAKE_FIND_LIBRARY_SUFFIXES ${_blas_ORIG_CMAKE_FIND_LIBRARY_SUFFIXES})