linalg-cmake-modules/util/BLASUtilities.cmake at 252f066c02c51f503d60ffc6654f0bd0ce119d94 · wavefunction91/linalg-cmake-modules · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
set( BLAS_UTILITY_CMAKE_FILE_DIR ${CMAKE_CURRENT_LIST_DIR} )

function( check_blas_int _libs _dgemm_name _libs_are_lp64 )

  check_lang_enabled( "C"   _c_enabled   )
  check_lang_enabled( "CXX" _cxx_enabled )
  check_lang_enabled( "Fortran" _fortran_enabled )

  set(_func_src_file)
  if( _c_enabled )
    set( _func_src_file ${BLAS_UTILITY_CMAKE_FILE_DIR}/ilp64_checker.c )
  elseif( _cxx_enabled )
    set( _func_src_file ${BLAS_UTILITY_CMAKE_FILE_DIR}/ilp64_checker.cxx )
  elseif( _fortran_enabled )
    set( _func_src_file ${BLAS_UTILITY_CMAKE_FILE_DIR}/ilp64_checker.f )
  endif()


if(_func_src_file)
  try_run( _run_result _compile_result ${CMAKE_CURRENT_BINARY_DIR}
    SOURCES       ${_func_src_file}
    LINK_LIBRARIES ${${_libs}}
    COMPILE_DEFINITIONS "-DDGEMM_NAME=${_dgemm_name}"
    COMPILE_OUTPUT_VARIABLE _compile_output
    RUN_OUTPUT_VARIABLE     _run_output
  )
else()
  set(_compile_result FALSE)
endif()

if (NOT _compile_result)
  message(FATAL_ERROR "check_blas_int: try_run failed: _compile_output=${_compile_output}")
endif()

if( _run_output MATCHES "BLAS IS LP64" )
  set( ${_libs_are_lp64} TRUE PARENT_SCOPE )
else()
  set( ${_libs_are_lp64} FALSE PARENT_SCOPE )
endif()

endfunction()