You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
331 lines
10 KiB
331 lines
10 KiB
# Distributed under the OSI-approved BSD 3-Clause License. See accompanying
|
|
# file Copyright.txt or https://cmake.org/licensing for details.
|
|
|
|
#.rst:
|
|
# FindOpenMP
|
|
# ----------
|
|
#
|
|
# Finds OpenMP support
|
|
#
|
|
# This module can be used to detect OpenMP support in a compiler. If
|
|
# the compiler supports OpenMP, the flags required to compile with
|
|
# OpenMP support are returned in variables for the different languages.
|
|
# The variables may be empty if the compiler does not need a special
|
|
# flag to support OpenMP.
|
|
#
|
|
# The following variables are set:
|
|
#
|
|
# ``OpenMP_C_FLAGS``
|
|
# Flags to add to the C compiler for OpenMP support.
|
|
# ``OpenMP_CXX_FLAGS``
|
|
# Flags to add to the CXX compiler for OpenMP support.
|
|
# ``OpenMP_Fortran_FLAGS``
|
|
# Flags to add to the Fortran compiler for OpenMP support.
|
|
# ``OPENMP_FOUND``
|
|
# True if openmp is detected.
|
|
#
|
|
# The following internal variables are set, if detected:
|
|
#
|
|
# ``OpenMP_C_SPEC_DATE``
|
|
# Specification date of OpenMP version of C compiler.
|
|
# ``OpenMP_CXX_SPEC_DATE``
|
|
# Specification date of OpenMP version of CXX compiler.
|
|
# ``OpenMP_Fortran_SPEC_DATE``
|
|
# Specification date of OpenMP version of Fortran compiler.
|
|
#
|
|
# The specification dates are formatted as integers of the form
|
|
# ``CCYYMM`` where these represent the decimal digits of the century,
|
|
# year, and month.
|
|
|
|
set(_OPENMP_REQUIRED_VARS)
|
|
set(CMAKE_REQUIRED_QUIET_SAVE ${CMAKE_REQUIRED_QUIET})
|
|
set(CMAKE_REQUIRED_QUIET ${OpenMP_FIND_QUIETLY})
|
|
|
|
function(_OPENMP_FLAG_CANDIDATES LANG)
|
|
set(OpenMP_FLAG_CANDIDATES
|
|
#Empty, if compiler automatically accepts openmp
|
|
" "
|
|
#GNU
|
|
"-fopenmp"
|
|
#Clang
|
|
"-fopenmp=libomp"
|
|
#Microsoft Visual Studio
|
|
"/openmp"
|
|
#Intel windows
|
|
"-Qopenmp"
|
|
#PathScale, Intel
|
|
"-openmp"
|
|
#Sun
|
|
"-xopenmp"
|
|
#HP
|
|
"+Oopenmp"
|
|
#IBM XL C/c++
|
|
"-qsmp"
|
|
#Portland Group, MIPSpro
|
|
"-mp"
|
|
)
|
|
|
|
set(OMP_FLAG_GNU "-fopenmp")
|
|
set(OMP_FLAG_Clang "-fopenmp=libomp")
|
|
set(OMP_FLAG_HP "+Oopenmp")
|
|
if(WIN32)
|
|
set(OMP_FLAG_Intel "-Qopenmp")
|
|
elseif(CMAKE_${LANG}_COMPILER_ID STREQUAL "Intel" AND
|
|
"${CMAKE_${LANG}_COMPILER_VERSION}" VERSION_LESS "15.0.0.20140528")
|
|
set(OMP_FLAG_Intel "-openmp")
|
|
else()
|
|
set(OMP_FLAG_Intel "-qopenmp")
|
|
endif()
|
|
set(OMP_FLAG_MIPSpro "-mp")
|
|
set(OMP_FLAG_MSVC "/openmp")
|
|
set(OMP_FLAG_PathScale "-openmp")
|
|
set(OMP_FLAG_PGI "-mp")
|
|
set(OMP_FLAG_SunPro "-xopenmp")
|
|
set(OMP_FLAG_XL "-qsmp")
|
|
set(OMP_FLAG_Cray " ")
|
|
|
|
# Move the flag that matches the compiler to the head of the list,
|
|
# this is faster and doesn't clutter the output that much. If that
|
|
# flag doesn't work we will still try all.
|
|
if(OMP_FLAG_${CMAKE_${LANG}_COMPILER_ID})
|
|
list(REMOVE_ITEM OpenMP_FLAG_CANDIDATES "${OMP_FLAG_${CMAKE_${LANG}_COMPILER_ID}}")
|
|
list(INSERT OpenMP_FLAG_CANDIDATES 0 "${OMP_FLAG_${CMAKE_${LANG}_COMPILER_ID}}")
|
|
endif()
|
|
|
|
set(OpenMP_${LANG}_FLAG_CANDIDATES "${OpenMP_FLAG_CANDIDATES}" PARENT_SCOPE)
|
|
endfunction()
|
|
|
|
# sample openmp source code to test
|
|
set(OpenMP_C_TEST_SOURCE
|
|
"
|
|
#include <omp.h>
|
|
int main() {
|
|
#ifdef _OPENMP
|
|
return 0;
|
|
#else
|
|
breaks_on_purpose
|
|
#endif
|
|
}
|
|
")
|
|
|
|
# same in Fortran
|
|
set(OpenMP_Fortran_TEST_SOURCE
|
|
"
|
|
program test
|
|
use omp_lib
|
|
integer :: n
|
|
n = omp_get_num_threads()
|
|
end program test
|
|
"
|
|
)
|
|
|
|
set(OpenMP_C_CXX_CHECK_VERSION_SOURCE
|
|
"
|
|
#include <stdio.h>
|
|
#include <omp.h>
|
|
const char ompver_str[] = { 'I', 'N', 'F', 'O', ':', 'O', 'p', 'e', 'n', 'M',
|
|
'P', '-', 'd', 'a', 't', 'e', '[',
|
|
('0' + ((_OPENMP/100000)%10)),
|
|
('0' + ((_OPENMP/10000)%10)),
|
|
('0' + ((_OPENMP/1000)%10)),
|
|
('0' + ((_OPENMP/100)%10)),
|
|
('0' + ((_OPENMP/10)%10)),
|
|
('0' + ((_OPENMP/1)%10)),
|
|
']', '\\0' };
|
|
int main(int argc, char *argv[])
|
|
{
|
|
printf(\"%s\\n\", ompver_str);
|
|
return 0;
|
|
}
|
|
")
|
|
|
|
set(OpenMP_Fortran_CHECK_VERSION_SOURCE
|
|
"
|
|
program omp_ver
|
|
use omp_lib
|
|
integer, parameter :: zero = ichar('0')
|
|
integer, parameter :: ompv = openmp_version
|
|
character, dimension(24), parameter :: ompver_str =&
|
|
(/ 'I', 'N', 'F', 'O', ':', 'O', 'p', 'e', 'n', 'M', 'P', '-',&
|
|
'd', 'a', 't', 'e', '[',&
|
|
char(zero + mod(ompv/100000, 10)),&
|
|
char(zero + mod(ompv/10000, 10)),&
|
|
char(zero + mod(ompv/1000, 10)),&
|
|
char(zero + mod(ompv/100, 10)),&
|
|
char(zero + mod(ompv/10, 10)),&
|
|
char(zero + mod(ompv/1, 10)), ']' /)
|
|
print *, ompver_str
|
|
end program omp_ver
|
|
")
|
|
|
|
function(_OPENMP_GET_SPEC_DATE LANG SPEC_DATE)
|
|
set(WORK_DIR ${CMAKE_BINARY_DIR}${CMAKE_FILES_DIRECTORY}/FindOpenMP)
|
|
if("${LANG}" STREQUAL "C")
|
|
set(SRC_FILE ${WORK_DIR}/ompver.c)
|
|
file(WRITE ${SRC_FILE} "${OpenMP_C_CXX_CHECK_VERSION_SOURCE}")
|
|
elseif("${LANG}" STREQUAL "CXX")
|
|
set(SRC_FILE ${WORK_DIR}/ompver.cpp)
|
|
file(WRITE ${SRC_FILE} "${OpenMP_C_CXX_CHECK_VERSION_SOURCE}")
|
|
else() # ("${LANG}" STREQUAL "Fortran")
|
|
set(SRC_FILE ${WORK_DIR}/ompver.f90)
|
|
file(WRITE ${SRC_FILE} "${OpenMP_Fortran_CHECK_VERSION_SOURCE}")
|
|
endif()
|
|
|
|
set(BIN_FILE ${WORK_DIR}/ompver_${LANG}.bin)
|
|
try_compile(OpenMP_TRY_COMPILE_RESULT ${CMAKE_BINARY_DIR} ${SRC_FILE}
|
|
CMAKE_FLAGS "-DCOMPILE_DEFINITIONS:STRING=${OpenMP_${LANG}_FLAGS}"
|
|
COPY_FILE ${BIN_FILE})
|
|
|
|
if(${OpenMP_TRY_COMPILE_RESULT})
|
|
file(STRINGS ${BIN_FILE} specstr LIMIT_COUNT 1 REGEX "INFO:OpenMP-date")
|
|
set(regex_spec_date ".*INFO:OpenMP-date\\[0*([^]]*)\\].*")
|
|
if("${specstr}" MATCHES "${regex_spec_date}")
|
|
set(${SPEC_DATE} "${CMAKE_MATCH_1}" PARENT_SCOPE)
|
|
endif()
|
|
endif()
|
|
|
|
unset(OpenMP_TRY_COMPILE_RESULT CACHE)
|
|
endfunction()
|
|
|
|
|
|
# check c compiler
|
|
if(CMAKE_C_COMPILER_LOADED)
|
|
# if these are set then do not try to find them again,
|
|
# by avoiding any try_compiles for the flags
|
|
if(OpenMP_C_FLAGS)
|
|
unset(OpenMP_C_FLAG_CANDIDATES)
|
|
else()
|
|
_OPENMP_FLAG_CANDIDATES("C")
|
|
include(${CMAKE_CURRENT_LIST_DIR}/CheckCSourceCompiles.cmake)
|
|
endif()
|
|
|
|
foreach(FLAG IN LISTS OpenMP_C_FLAG_CANDIDATES)
|
|
set(SAFE_CMAKE_REQUIRED_FLAGS "${CMAKE_REQUIRED_FLAGS}")
|
|
set(CMAKE_REQUIRED_FLAGS "${FLAG}")
|
|
unset(OpenMP_FLAG_DETECTED CACHE)
|
|
if(NOT CMAKE_REQUIRED_QUIET)
|
|
message(STATUS "Try OpenMP C flag = [${FLAG}]")
|
|
endif()
|
|
check_c_source_compiles("${OpenMP_C_TEST_SOURCE}" OpenMP_FLAG_DETECTED)
|
|
set(CMAKE_REQUIRED_FLAGS "${SAFE_CMAKE_REQUIRED_FLAGS}")
|
|
if(OpenMP_FLAG_DETECTED)
|
|
set(OpenMP_C_FLAGS_INTERNAL "${FLAG}")
|
|
break()
|
|
endif()
|
|
endforeach()
|
|
|
|
set(OpenMP_C_FLAGS "${OpenMP_C_FLAGS_INTERNAL}"
|
|
CACHE STRING "C compiler flags for OpenMP parallization")
|
|
|
|
list(APPEND _OPENMP_REQUIRED_VARS OpenMP_C_FLAGS)
|
|
unset(OpenMP_C_FLAG_CANDIDATES)
|
|
|
|
if (NOT OpenMP_C_SPEC_DATE)
|
|
_OPENMP_GET_SPEC_DATE("C" OpenMP_C_SPEC_DATE_INTERNAL)
|
|
set(OpenMP_C_SPEC_DATE "${OpenMP_C_SPEC_DATE_INTERNAL}" CACHE
|
|
INTERNAL "C compiler's OpenMP specification date")
|
|
endif()
|
|
endif()
|
|
|
|
# check cxx compiler
|
|
if(CMAKE_CXX_COMPILER_LOADED)
|
|
# if these are set then do not try to find them again,
|
|
# by avoiding any try_compiles for the flags
|
|
if(OpenMP_CXX_FLAGS)
|
|
unset(OpenMP_CXX_FLAG_CANDIDATES)
|
|
else()
|
|
_OPENMP_FLAG_CANDIDATES("CXX")
|
|
include(${CMAKE_CURRENT_LIST_DIR}/CheckCXXSourceCompiles.cmake)
|
|
|
|
# use the same source for CXX as C for now
|
|
set(OpenMP_CXX_TEST_SOURCE ${OpenMP_C_TEST_SOURCE})
|
|
endif()
|
|
|
|
foreach(FLAG IN LISTS OpenMP_CXX_FLAG_CANDIDATES)
|
|
set(SAFE_CMAKE_REQUIRED_FLAGS "${CMAKE_REQUIRED_FLAGS}")
|
|
set(CMAKE_REQUIRED_FLAGS "${FLAG}")
|
|
unset(OpenMP_FLAG_DETECTED CACHE)
|
|
if(NOT CMAKE_REQUIRED_QUIET)
|
|
message(STATUS "Try OpenMP CXX flag = [${FLAG}]")
|
|
endif()
|
|
check_cxx_source_compiles("${OpenMP_CXX_TEST_SOURCE}" OpenMP_FLAG_DETECTED)
|
|
set(CMAKE_REQUIRED_FLAGS "${SAFE_CMAKE_REQUIRED_FLAGS}")
|
|
if(OpenMP_FLAG_DETECTED)
|
|
set(OpenMP_CXX_FLAGS_INTERNAL "${FLAG}")
|
|
break()
|
|
endif()
|
|
endforeach()
|
|
|
|
set(OpenMP_CXX_FLAGS "${OpenMP_CXX_FLAGS_INTERNAL}"
|
|
CACHE STRING "C++ compiler flags for OpenMP parallization")
|
|
|
|
list(APPEND _OPENMP_REQUIRED_VARS OpenMP_CXX_FLAGS)
|
|
unset(OpenMP_CXX_FLAG_CANDIDATES)
|
|
|
|
if (NOT OpenMP_CXX_SPEC_DATE)
|
|
_OPENMP_GET_SPEC_DATE("CXX" OpenMP_CXX_SPEC_DATE_INTERNAL)
|
|
set(OpenMP_CXX_SPEC_DATE "${OpenMP_CXX_SPEC_DATE_INTERNAL}" CACHE
|
|
INTERNAL "C++ compiler's OpenMP specification date")
|
|
endif()
|
|
endif()
|
|
|
|
# check Fortran compiler
|
|
if(CMAKE_Fortran_COMPILER_LOADED)
|
|
# if these are set then do not try to find them again,
|
|
# by avoiding any try_compiles for the flags
|
|
if(OpenMP_Fortran_FLAGS)
|
|
unset(OpenMP_Fortran_FLAG_CANDIDATES)
|
|
else()
|
|
_OPENMP_FLAG_CANDIDATES("Fortran")
|
|
include(${CMAKE_CURRENT_LIST_DIR}/CheckFortranSourceCompiles.cmake)
|
|
endif()
|
|
|
|
foreach(FLAG IN LISTS OpenMP_Fortran_FLAG_CANDIDATES)
|
|
set(SAFE_CMAKE_REQUIRED_FLAGS "${CMAKE_REQUIRED_FLAGS}")
|
|
set(CMAKE_REQUIRED_FLAGS "${FLAG}")
|
|
unset(OpenMP_FLAG_DETECTED CACHE)
|
|
if(NOT CMAKE_REQUIRED_QUIET)
|
|
message(STATUS "Try OpenMP Fortran flag = [${FLAG}]")
|
|
endif()
|
|
check_fortran_source_compiles("${OpenMP_Fortran_TEST_SOURCE}" OpenMP_FLAG_DETECTED)
|
|
set(CMAKE_REQUIRED_FLAGS "${SAFE_CMAKE_REQUIRED_FLAGS}")
|
|
if(OpenMP_FLAG_DETECTED)
|
|
set(OpenMP_Fortran_FLAGS_INTERNAL "${FLAG}")
|
|
break()
|
|
endif()
|
|
endforeach()
|
|
|
|
set(OpenMP_Fortran_FLAGS "${OpenMP_Fortran_FLAGS_INTERNAL}"
|
|
CACHE STRING "Fortran compiler flags for OpenMP parallization")
|
|
|
|
list(APPEND _OPENMP_REQUIRED_VARS OpenMP_Fortran_FLAGS)
|
|
unset(OpenMP_Fortran_FLAG_CANDIDATES)
|
|
|
|
if (NOT OpenMP_Fortran_SPEC_DATE)
|
|
_OPENMP_GET_SPEC_DATE("Fortran" OpenMP_Fortran_SPEC_DATE_INTERNAL)
|
|
set(OpenMP_Fortran_SPEC_DATE "${OpenMP_Fortran_SPEC_DATE_INTERNAL}" CACHE
|
|
INTERNAL "Fortran compiler's OpenMP specification date")
|
|
endif()
|
|
endif()
|
|
|
|
set(CMAKE_REQUIRED_QUIET ${CMAKE_REQUIRED_QUIET_SAVE})
|
|
|
|
if(_OPENMP_REQUIRED_VARS)
|
|
include(${CMAKE_CURRENT_LIST_DIR}/FindPackageHandleStandardArgs.cmake)
|
|
|
|
find_package_handle_standard_args(OpenMP
|
|
REQUIRED_VARS ${_OPENMP_REQUIRED_VARS})
|
|
|
|
mark_as_advanced(${_OPENMP_REQUIRED_VARS})
|
|
|
|
unset(_OPENMP_REQUIRED_VARS)
|
|
else()
|
|
message(SEND_ERROR "FindOpenMP requires C or CXX language to be enabled")
|
|
endif()
|
|
|
|
unset(OpenMP_C_TEST_SOURCE)
|
|
unset(OpenMP_CXX_TEST_SOURCE)
|
|
unset(OpenMP_Fortran_TEST_SOURCE)
|
|
unset(OpenMP_C_CXX_CHECK_VERSION_SOURCE)
|
|
unset(OpenMP_Fortran_CHECK_VERSION_SOURCE)
|