adding a module for cblas

This commit is contained in:
Radovan Bast 2015-09-17 00:08:43 +02:00
parent c2d8abcca2
commit c0199ccd43
3 changed files with 84 additions and 9 deletions

78
modules/cblas.cmake Normal file
View File

@ -0,0 +1,78 @@
#.rst:
#
# Detect and link to CBLAS. Work in progress.
#
# Variables used::
#
# ENABLE_STATIC_LINKING
#
# Variables defined::
#
# CBLAS_FOUND - describe me, uncached
# CBLAS_LIBRARIES - describe me, uncached
# CBLAS_INCLUDE_DIR - describe me, uncached
#
# autocmake.cfg configuration::
#
# docopt: --cblas Detect and link to CBLAS [default: False].
# define: '-DENABLE_CBLAS=%s' % arguments['--cblas']
option(ENABLE_CBLAS "Enable CBLAS" OFF)
if(ENABLE_CBLAS)
if(ENABLE_STATIC_LINKING)
set(CMAKE_FIND_LIBRARY_SUFFIXES .a)
endif()
include(CheckIncludeFile)
function(_find_include_dir _names _hints _result)
find_path(_include_dir
NAMES ${_names}
HINTS ${_hints}
)
set(_all_include_files_work TRUE)
foreach(_name ${_names})
check_include_file(${_include_dir}/${_name} _include_file_works)
set(_all_include_files_work (${_all_include_files_work} AND ${_include_file_works}))
endforeach()
if(${_all_include_files_work})
set(${_result} ${_include_dir} PARENT_SCOPE)
endif()
endfunction()
include(CheckFunctionExists)
function(_find_library _names _check_function _result)
if(APPLE)
find_library(_lib
NAMES ${_names}
PATHS /usr/local/lib /usr/lib /usr/local/lib64 /usr/lib64
ENV DYLD_LIBRARY_PATH
)
else()
find_library(_lib
NAMES ${_names}
PATHS /usr/local/lib /usr/lib /usr/local/lib64 /usr/lib64
ENV LD_LIBRARY_PATH
)
endif()
set(CMAKE_REQUIRED_LIBRARIES ${_lib})
check_function_exists(${_check_function} _library_works)
if(${_library_works})
set(${_result} ${_lib} PARENT_SCOPE)
endif()
endfunction()
set(CBLAS_FOUND FALSE)
set(CBLAS_INCLUDE_DIR "undefined")
_find_include_dir(cblas.h /usr CBLAS_INCLUDE_DIR)
set(CBLAS_LIBRARIES "undefined")
_find_library(cblas cblas_dgemm CBLAS_LIBRARIES)
if(NOT ${CBLAS_INCLUDE_DIR} STREQUAL "undefined" AND NOT ${CBLAS_LIBRARIES} STREQUAL "undefined")
set(CBLAS_FOUND TRUE)
endif()
endif()

View File

@ -8,8 +8,8 @@ source: ../../../modules/cc.cmake
[static]
source: ../../../modules/static_linking.cmake
[math_libs]
source: ../../../modules/math_libs.cmake
[cblas]
source: ../../../modules/cblas.cmake
[default_build_paths]
source: ../../../modules/default_build_paths.cmake

View File

@ -1,10 +1,7 @@
if(BLAS_FOUND)
if(${CMAKE_SYSTEM_NAME} MATCHES "Windows")
#windows does not find cblas.h of openblas by default
include_directories($ENV{PATH})
endif()
if(CBLAS_FOUND)
include_directories(${CBLAS_INCLUDE_DIR})
add_executable(example example.c)
target_link_libraries(example ${MATH_LIBS})
target_link_libraries(example ${CBLAS_LIBRARIES})
else()
message(FATAL_ERROR "BLAS library not found for the test cc_cblas!")
message(FATAL_ERROR "CBLAS library not found")
endif()