Permalink
Cannot retrieve contributors at this time
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
71 lines (65 sloc)
2.33 KB
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
# (c) https://github.com/dev-cafe/autocmake/blob/master/AUTHORS.md | |
# licensed under BSD-3: https://github.com/dev-cafe/autocmake/blob/master/LICENSE | |
#.rst: | |
# | |
# Enables OpenMP support. | |
# | |
# Variables used:: | |
# | |
# ENABLE_OPENMP | |
# OPENMP_FOUND | |
# | |
# Variables modified (provided the corresponding language is enabled):: | |
# | |
# CMAKE_Fortran_FLAGS | |
# CMAKE_C_FLAGS | |
# CMAKE_CXX_FLAGS | |
# | |
# autocmake.yml configuration:: | |
# | |
# docopt: "--omp Enable OpenMP parallelization [default: False]." | |
# define: "'-DENABLE_OPENMP={0}'.format(arguments['--omp'])" | |
option(ENABLE_OPENMP "Enable OpenMP parallelization" OFF) | |
if(ENABLE_OPENMP) | |
if(NOT OPENMP_FOUND) | |
find_package(OpenMP) | |
endif() | |
if(OPENMP_FOUND) | |
add_definitions(-DHAVE_OPENMP) | |
if(DEFINED CMAKE_C_COMPILER_ID) | |
set(CMAKE_C_FLAGS "${CMAKE_C_FLAGS} ${OpenMP_C_FLAGS}") | |
endif() | |
if(DEFINED CMAKE_CXX_COMPILER_ID) | |
set(CMAKE_CXX_FLAGS "${CMAKE_CXX_FLAGS} ${OpenMP_CXX_FLAGS}") | |
endif() | |
if(DEFINED CMAKE_Fortran_COMPILER_ID) | |
set(CMAKE_Fortran_FLAGS "${CMAKE_Fortran_FLAGS} ${OpenMP_Fortran_FLAGS}") | |
endif() | |
endif() | |
# this is only needed for CMake below 3.5 | |
if(DEFINED CMAKE_Fortran_COMPILER_ID AND NOT DEFINED OpenMP_Fortran_FLAGS) | |
# we do this in a pedestrian way because the Fortran support is relatively recent | |
if(CMAKE_Fortran_COMPILER_ID MATCHES GNU) | |
set(CMAKE_Fortran_FLAGS "${CMAKE_Fortran_FLAGS} -fopenmp") | |
endif() | |
if(CMAKE_Fortran_COMPILER_ID MATCHES Intel) | |
if(WIN32) | |
set(CMAKE_Fortran_FLAGS "${CMAKE_Fortran_FLAGS} -Qopenmp") | |
elseif("${CMAKE_Fortran_COMPILER_VERSION}" VERSION_LESS "15.0.0.20140528") | |
set(CMAKE_Fortran_FLAGS "${CMAKE_Fortran_FLAGS} -openmp") | |
else() | |
set(CMAKE_Fortran_FLAGS "${CMAKE_Fortran_FLAGS} -qopenmp") | |
endif() | |
endif() | |
if(CMAKE_Fortran_COMPILER_ID MATCHES PGI) | |
set(CMAKE_Fortran_FLAGS "${CMAKE_Fortran_FLAGS} -mp") | |
endif() | |
if(CMAKE_Fortran_COMPILER_ID MATCHES XL) | |
set(CMAKE_Fortran_FLAGS "${CMAKE_Fortran_FLAGS} -qsmp=omp") | |
endif() | |
if(CMAKE_Fortran_COMPILER_ID MATCHES Cray) | |
# do nothing in this case | |
endif() | |
set(OPENMP_FOUND TRUE) | |
endif() | |
endif() |