mirror of
https://github.com/marian-nmt/marian.git
synced 2024-09-11 06:15:56 +03:00
* Compile marian on mac and clang. Two linker errors left * MacOS defines has a different definition for unsigned long * Find OpenBLAS on mac * Fix a typo in the BLAS detection * Simplify and add comments * Refactor cpu allocation code. Do not fallback to malloc * Fix compilation warning on gcc * Refactor memory allocation * Make things compile with clang-8 with fewer warnings. * Eliminate clang warnings when compiling examples and when compiling without MKL * added USE_MKL option to compile without MKL for debugging even when MKL is installed * fixed issues with compiling examples with clang * Fix compile errors with clang in src/tests. * Fix missing whitespace in error message in src/tests/sqlite.cpp. * Responding to Frank Seide's code review. * Eliminate clang warnings when compiling with -DUSE_FBGEMM=on. * Fix compilation on gcc 8 * Get Marian to compile with Clang-10. * Fix Clang-8 warnings when compiling with marian-server * Add more comments and explicit unsigned long long for windows * Pull in fbgemm that supports mac * Fix warning flags order in CMakeLists.txt Co-authored-by: Kenneth Heafield <kpu@users.noreply.github.com> Co-authored-by: Ulrich Germann <ulrich.germann@gmail.com> Co-authored-by: Roman Grundkiewicz <romang@amu.edu.pl>
187 lines
5.8 KiB
CMake
187 lines
5.8 KiB
CMake
# - Find CBLAS library
|
|
#
|
|
# This module finds an installed fortran library that implements the CBLAS
|
|
# linear-algebra interface (see http://www.netlib.org/blas/), with CBLAS
|
|
# interface.
|
|
#
|
|
# This module sets the following variables:
|
|
# CBLAS_FOUND - set to true if a library implementing the CBLAS interface
|
|
# is found
|
|
# CBLAS_LINKER_FLAGS - uncached list of required linker flags (excluding -l
|
|
# and -L).
|
|
# CBLAS_LIBRARIES - uncached list of libraries (using full path name) to
|
|
# link against to use CBLAS
|
|
# CBLAS_INCLUDE_DIR - path to includes
|
|
# CBLAS_INCLUDE_FILE - the file to be included to use CBLAS
|
|
#
|
|
|
|
## Based on https://github.com/Eyescale/CMake/blob/master/FindCBLAS.cmake
|
|
|
|
INCLUDE(CheckFunctionExists)
|
|
INCLUDE(CheckIncludeFile)
|
|
|
|
MACRO(CHECK_ALL_LIBRARIES LIBRARIES INCLUDE _prefix _name _flags _list _include _search_include)
|
|
# This macro checks for the existence of the combination of fortran libraries
|
|
# given by _list. If the combination is found, this macro checks (using the
|
|
# Check_Fortran_Function_Exists macro) whether can link against that library
|
|
# combination using the name of a routine given by _name using the linker
|
|
# flags given by _flags. If the combination of libraries is found and passes
|
|
# the link test, LIBRARIES is set to the list of complete library paths that
|
|
# have been found. Otherwise, LIBRARIES is set to FALSE.
|
|
|
|
# N.B. _prefix is the prefix applied to the names of all cached variables that
|
|
# are generated internally and marked advanced by this macro.
|
|
|
|
SET(__list)
|
|
FOREACH(_elem ${_list})
|
|
IF(__list)
|
|
SET(__list "${__list} - ${_elem}")
|
|
ELSE(__list)
|
|
SET(__list "${_elem}")
|
|
ENDIF(__list)
|
|
ENDFOREACH(_elem)
|
|
MESSAGE(STATUS "Checking for [${__list}]")
|
|
SET(_libraries_work TRUE)
|
|
SET(${LIBRARIES})
|
|
SET(_combined_name)
|
|
SET(_paths)
|
|
FOREACH(_library ${_list})
|
|
SET(_combined_name ${_combined_name}_${_library})
|
|
|
|
# did we find all the libraries in the _list until now?
|
|
# (we stop at the first unfound one)
|
|
IF(_libraries_work)
|
|
IF(APPLE)
|
|
FIND_LIBRARY(${_prefix}_${_library}_LIBRARY
|
|
NAMES ${_library}
|
|
PATHS /usr/local/lib /usr/lib /usr/local/lib64 /usr/lib64 /usr/local/opt/openblas/lib ENV
|
|
DYLD_LIBRARY_PATH
|
|
)
|
|
ELSE(APPLE)
|
|
FIND_LIBRARY(${_prefix}_${_library}_LIBRARY
|
|
NAMES ${_library}
|
|
PATHS /usr/local/lib /usr/lib /usr/local/lib64 /usr/lib64 ENV
|
|
LD_LIBRARY_PATH
|
|
)
|
|
ENDIF(APPLE)
|
|
MARK_AS_ADVANCED(${_prefix}_${_library}_LIBRARY)
|
|
IF(${_prefix}_${_library}_LIBRARY)
|
|
GET_FILENAME_COMPONENT(_path ${${_prefix}_${_library}_LIBRARY} PATH)
|
|
LIST(APPEND _paths ${_path}/../include ${_path}/../../include)
|
|
ENDIF(${_prefix}_${_library}_LIBRARY)
|
|
SET(${LIBRARIES} ${${LIBRARIES}} ${${_prefix}_${_library}_LIBRARY})
|
|
SET(_libraries_work ${${_prefix}_${_library}_LIBRARY})
|
|
ENDIF(_libraries_work)
|
|
ENDFOREACH(_library ${_list})
|
|
|
|
# Test include
|
|
SET(_bug_search_include ${_search_include}) #CMAKE BUG!!! SHOULD NOT BE THAT
|
|
IF(_bug_search_include)
|
|
FIND_PATH(${_prefix}${_combined_name}_INCLUDE ${_include} ${_paths})
|
|
MARK_AS_ADVANCED(${_prefix}${_combined_name}_INCLUDE)
|
|
IF(${_prefix}${_combined_name}_INCLUDE)
|
|
MESSAGE(STATUS "Checking for [${__list}] -- includes found")
|
|
SET(${_prefix}_INCLUDE_DIR ${${_prefix}${_combined_name}_INCLUDE})
|
|
SET(${_prefix}_INCLUDE_FILE ${_include})
|
|
SET(${INCLUDE} ${${_prefix}_INCLUDE_DIR})
|
|
ELSE(${_prefix}${_combined_name}_INCLUDE)
|
|
MESSAGE(STATUS "Checking for [${__list}] -- includes not found")
|
|
SET(_libraries_work FALSE)
|
|
ENDIF(${_prefix}${_combined_name}_INCLUDE)
|
|
ELSE(_bug_search_include)
|
|
SET(${_prefix}_INCLUDE_DIR)
|
|
SET(${_prefix}_INCLUDE_FILE ${_include})
|
|
ENDIF(_bug_search_include)
|
|
|
|
IF(_libraries_work)
|
|
# Test this combination of libraries.
|
|
SET(CMAKE_REQUIRED_LIBRARIES ${_flags} ${${LIBRARIES}})
|
|
CHECK_FUNCTION_EXISTS(${_name} ${_prefix}${_combined_name}_WORKS)
|
|
SET(CMAKE_REQUIRED_LIBRARIES)
|
|
MARK_AS_ADVANCED(${_prefix}${_combined_name}_WORKS)
|
|
SET(_libraries_work ${${_prefix}${_combined_name}_WORKS})
|
|
|
|
IF(_libraries_work)
|
|
MESSAGE(STATUS "Checking for [${__list}] -- libraries found")
|
|
ENDIF(_libraries_work)
|
|
|
|
ENDIF(_libraries_work)
|
|
|
|
|
|
IF(NOT _libraries_work)
|
|
SET(${LIBRARIES} FALSE)
|
|
ENDIF(NOT _libraries_work)
|
|
|
|
ENDMACRO(CHECK_ALL_LIBRARIES)
|
|
|
|
SET(CBLAS_LINKER_FLAGS)
|
|
SET(CBLAS_LIBRARIES)
|
|
SET(CBLAS_INCLUDE_DIR)
|
|
|
|
# CBLAS in openBLAS
|
|
IF(NOT CBLAS_LIBRARIES)
|
|
CHECK_ALL_LIBRARIES(
|
|
CBLAS_LIBRARIES
|
|
CBLAS_INCLUDE_DIR
|
|
cblas
|
|
cblas_sgemm
|
|
""
|
|
"openblas"
|
|
"cblas.h"
|
|
TRUE
|
|
)
|
|
ENDIF(NOT CBLAS_LIBRARIES)
|
|
|
|
#MESSAGE(STATUS ${openblas_INCLUDE_DIR})
|
|
|
|
# CBLAS in CBLAS
|
|
IF(NOT CBLAS_LIBRARIES)
|
|
CHECK_ALL_LIBRARIES(
|
|
CBLAS_LIBRARIES
|
|
CBLAS_INCLUDE_DIR
|
|
cblas
|
|
cblas_sgemm
|
|
""
|
|
"cblas"
|
|
"cblas.h"
|
|
TRUE
|
|
)
|
|
ENDIF(NOT CBLAS_LIBRARIES)
|
|
|
|
#MESSAGE(STATUS ${cblas_INCLUDE_DIR})
|
|
|
|
# CBLAS in lapacke
|
|
IF(NOT CBLAS_LIBRARIES)
|
|
CHECK_ALL_LIBRARIES(
|
|
CBLAS_LIBRARIES
|
|
CBLAS_INCLUDE_DIR
|
|
cblas
|
|
cblas_sgemm
|
|
""
|
|
"lapacke"
|
|
"cblas.h"
|
|
TRUE
|
|
)
|
|
ENDIF(NOT CBLAS_LIBRARIES)
|
|
|
|
#MESSAGE(STATUS ${lapacke_INCLUDE_DIR})
|
|
|
|
IF(CBLAS_LIBRARIES)
|
|
SET(CBLAS_FOUND TRUE)
|
|
ELSE(CBLAS_LIBRARIES)
|
|
SET(CBLAS_FOUND FALSE)
|
|
ENDIF(CBLAS_LIBRARIES)
|
|
|
|
IF(NOT CBLAS_FOUND AND CBLAS_FIND_REQUIRED)
|
|
MESSAGE(FATAL_ERROR "CBLAS library not found. Please specify library location")
|
|
ENDIF(NOT CBLAS_FOUND AND CBLAS_FIND_REQUIRED)
|
|
|
|
IF(NOT CBLAS_FIND_QUIETLY)
|
|
IF(CBLAS_FOUND)
|
|
MESSAGE(STATUS "CBLAS library found: " ${CBLAS_LIBRARIES})
|
|
MESSAGE(STATUS "cblas.h include directory: " ${CBLAS_INCLUDE_DIR})
|
|
ELSE(CBLAS_FOUND)
|
|
MESSAGE(STATUS "CBLAS library not found. Please specify library location")
|
|
ENDIF(CBLAS_FOUND)
|
|
ENDIF(NOT CBLAS_FIND_QUIETLY)
|