91 lines
3.8 KiB
CMake
91 lines
3.8 KiB
CMake
SET(Kokkos_DEVICES @KOKKOS_ENABLED_DEVICES@)
|
|
SET(Kokkos_OPTIONS @KOKKOS_ENABLED_OPTIONS@)
|
|
SET(Kokkos_TPLS @KOKKOS_ENABLED_TPLS@)
|
|
SET(Kokkos_ARCH @KOKKOS_ENABLED_ARCH_LIST@)
|
|
|
|
# These are needed by KokkosKernels
|
|
FOREACH(DEV ${Kokkos_DEVICES})
|
|
SET(Kokkos_ENABLE_${DEV} ON)
|
|
ENDFOREACH()
|
|
|
|
IF(NOT Kokkos_FIND_QUIETLY)
|
|
MESSAGE(STATUS "Enabled Kokkos devices: ${Kokkos_DEVICES}")
|
|
ENDIF()
|
|
|
|
IF (Kokkos_ENABLE_CUDA AND ${CMAKE_VERSION} VERSION_GREATER_EQUAL "3.14.0")
|
|
#If we are building CUDA, we have tricked CMake because we declare a CXX project
|
|
#If the default C++ standard for a given compiler matches the requested
|
|
#standard, then CMake just omits the -std flag in later versions of CMake
|
|
#This breaks CUDA compilation (CUDA compiler can have a different default
|
|
#-std then the underlying host compiler by itself). Setting this variable
|
|
#forces CMake to always add the -std flag even if it thinks it doesn't need it
|
|
SET(CMAKE_CXX_STANDARD_DEFAULT 98 CACHE INTERNAL "" FORCE)
|
|
ENDIF()
|
|
|
|
SET(KOKKOS_USE_CXX_EXTENSIONS @KOKKOS_USE_CXX_EXTENSIONS@)
|
|
IF (NOT DEFINED CMAKE_CXX_EXTENSIONS OR CMAKE_CXX_EXTENSIONS)
|
|
IF (NOT KOKKOS_USE_CXX_EXTENSIONS)
|
|
MESSAGE(WARNING "The installed Kokkos configuration does not support CXX extensions. Forcing -DCMAKE_CXX_EXTENSIONS=Off")
|
|
SET(CMAKE_CXX_EXTENSIONS OFF CACHE BOOL "" FORCE)
|
|
ENDIF()
|
|
ENDIF()
|
|
|
|
include(FindPackageHandleStandardArgs)
|
|
|
|
# This function makes sure that Kokkos was built with the requested backends
|
|
# and target architectures and generates a fatal error if it was not.
|
|
#
|
|
# kokkos_check(
|
|
# [DEVICES <devices>...] # Set of backends (e.g. "OpenMP" and/or "Cuda")
|
|
# [ARCH <archs>...] # Target architectures (e.g. "Power9" and/or "Volta70")
|
|
# [OPTIONS <options>...] # Optional settings (e.g. "PROFILING")
|
|
# [TPLS <tpls>...] # Third party libraries
|
|
# [RETURN_VALUE <result>] # Set a variable that indicates the result of the
|
|
# # check instead of a fatal error
|
|
# )
|
|
function(kokkos_check)
|
|
set(ALLOWED_ARGS DEVICES ARCH OPTIONS TPLS)
|
|
cmake_parse_arguments(KOKKOS_CHECK "" "RETURN_VALUE" "${ALLOWED_ARGS}" ${ARGN})
|
|
foreach(_arg ${KOKKOS_CHECK_UNPARSED_ARGUMENTS})
|
|
message(SEND_ERROR "Argument '${_arg}' passed to kokkos_check() was not recognized")
|
|
endforeach()
|
|
# Get the list of keywords that were actually passed to the function.
|
|
set(REQUESTED_ARGS)
|
|
foreach(arg ${ALLOWED_ARGS})
|
|
if(KOKKOS_CHECK_${arg})
|
|
list(APPEND REQUESTED_ARGS ${arg})
|
|
endif()
|
|
endforeach()
|
|
set(KOKKOS_CHECK_SUCCESS TRUE)
|
|
foreach(arg ${REQUESTED_ARGS})
|
|
# Define variables named after the required arguments that are provided by
|
|
# the Kokkos install.
|
|
foreach(requested ${KOKKOS_CHECK_${arg}})
|
|
foreach(provided ${Kokkos_${arg}})
|
|
STRING(TOUPPER ${requested} REQUESTED_UC)
|
|
STRING(TOUPPER ${provided} PROVIDED_UC)
|
|
if(PROVIDED_UC STREQUAL REQUESTED_UC)
|
|
string(REPLACE ";" " " ${requested} "${KOKKOS_CHECK_${arg}}")
|
|
endif()
|
|
endforeach()
|
|
endforeach()
|
|
# Somewhat divert the CMake function below from its original purpose and
|
|
# use it to check that there are variables defined for all required
|
|
# arguments. Success or failure messages will be displayed but we are
|
|
# responsible for signaling failure and skip the build system generation.
|
|
if (KOKKOS_CHECK_RETURN_VALUE)
|
|
set(Kokkos_${arg}_FIND_QUIETLY ON)
|
|
endif()
|
|
find_package_handle_standard_args("Kokkos_${arg}" DEFAULT_MSG
|
|
${KOKKOS_CHECK_${arg}})
|
|
if(NOT Kokkos_${arg}_FOUND)
|
|
set(KOKKOS_CHECK_SUCCESS FALSE)
|
|
endif()
|
|
endforeach()
|
|
if(NOT KOKKOS_CHECK_SUCCESS AND NOT KOKKOS_CHECK_RETURN_VALUE)
|
|
message(FATAL_ERROR "Kokkos does NOT provide all backends and/or architectures requested")
|
|
else()
|
|
set(${KOKKOS_CHECK_RETURN_VALUE} ${KOKKOS_CHECK_SUCCESS} PARENT_SCOPE)
|
|
endif()
|
|
endfunction()
|