99 lines
4.1 KiB
CMake
99 lines
4.1 KiB
CMake
|
# Distributed under the OSI-approved BSD 3-Clause License. See accompanying
|
||
|
|
# file Copyright.txt or https://cmake.org/licensing for details.
|
||
|
|
|
||
|
|
if(CMAKE_CUDA_COMPILER_FORCED)
|
||
|
|
# The compiler configuration was forced by the user.
|
||
|
|
# Assume the user has configured all compiler information.
|
||
|
|
set(CMAKE_CUDA_COMPILER_WORKS TRUE)
|
||
|
|
return()
|
||
|
|
endif()
|
||
|
|
|
||
|
|
include(CMakeTestCompilerCommon)
|
||
|
|
|
||
|
|
# Remove any cached result from an older CMake version.
|
||
|
|
# We now store this in CMakeCUDACompiler.cmake.
|
||
|
|
unset(CMAKE_CUDA_COMPILER_WORKS CACHE)
|
||
|
|
|
||
|
|
# Try to identify the ABI and configure it into CMakeCUDACompiler.cmake
|
||
|
|
include(${CMAKE_ROOT}/Modules/CMakeDetermineCompilerABI.cmake)
|
||
|
|
CMAKE_DETERMINE_COMPILER_ABI(CUDA ${CMAKE_ROOT}/Modules/CMakeCUDACompilerABI.cu)
|
||
|
|
if(CMAKE_CUDA_ABI_COMPILED)
|
||
|
|
# The compiler worked so skip dedicated test below.
|
||
|
|
set(CMAKE_CUDA_COMPILER_WORKS TRUE)
|
||
|
|
message(STATUS "Check for working CUDA compiler: ${CMAKE_CUDA_COMPILER} - skipped")
|
||
|
|
endif()
|
||
|
|
|
||
|
|
# This file is used by EnableLanguage in cmGlobalGenerator to
|
||
|
|
# determine that the selected cuda compiler can actually compile
|
||
|
|
# and link the most basic of programs. If not, a fatal error
|
||
|
|
# is set and cmake stops processing commands and will not generate
|
||
|
|
# any makefiles or projects.
|
||
|
|
if(NOT CMAKE_CUDA_COMPILER_WORKS)
|
||
|
|
PrintTestCompilerStatus("CUDA")
|
||
|
|
file(WRITE ${CMAKE_BINARY_DIR}${CMAKE_FILES_DIRECTORY}/CMakeTmp/main.cu
|
||
|
|
"#ifndef __CUDACC__\n"
|
||
|
|
"# error \"The CMAKE_CUDA_COMPILER is set to an invalid CUDA compiler\"\n"
|
||
|
|
"#endif\n"
|
||
|
|
"int main(){return 0;}\n")
|
||
|
|
|
||
|
|
try_compile(CMAKE_CUDA_COMPILER_WORKS ${CMAKE_BINARY_DIR}
|
||
|
|
${CMAKE_BINARY_DIR}${CMAKE_FILES_DIRECTORY}/CMakeTmp/main.cu
|
||
|
|
OUTPUT_VARIABLE __CMAKE_CUDA_COMPILER_OUTPUT)
|
||
|
|
|
||
|
|
# Move result from cache to normal variable.
|
||
|
|
set(CMAKE_CUDA_COMPILER_WORKS ${CMAKE_CUDA_COMPILER_WORKS})
|
||
|
|
unset(CMAKE_CUDA_COMPILER_WORKS CACHE)
|
||
|
|
if(NOT CMAKE_CUDA_COMPILER_WORKS)
|
||
|
|
PrintTestCompilerResult(CHECK_FAIL "broken")
|
||
|
|
file(APPEND ${CMAKE_BINARY_DIR}${CMAKE_FILES_DIRECTORY}/CMakeError.log
|
||
|
|
"Determining if the CUDA compiler works failed with "
|
||
|
|
"the following output:\n${__CMAKE_CUDA_COMPILER_OUTPUT}\n\n")
|
||
|
|
string(REPLACE "\n" "\n " _output "${__CMAKE_CUDA_COMPILER_OUTPUT}")
|
||
|
|
message(FATAL_ERROR "The CUDA compiler\n \"${CMAKE_CUDA_COMPILER}\"\n"
|
||
|
|
"is not able to compile a simple test program.\nIt fails "
|
||
|
|
"with the following output:\n ${_output}\n\n"
|
||
|
|
"CMake will not be able to correctly generate this project.")
|
||
|
|
endif()
|
||
|
|
PrintTestCompilerResult(CHECK_PASS "works")
|
||
|
|
file(APPEND ${CMAKE_BINARY_DIR}${CMAKE_FILES_DIRECTORY}/CMakeOutput.log
|
||
|
|
"Determining if the CUDA compiler works passed with "
|
||
|
|
"the following output:\n${__CMAKE_CUDA_COMPILER_OUTPUT}\n\n")
|
||
|
|
endif()
|
||
|
|
|
||
|
|
# Try to identify the compiler features
|
||
|
|
include(${CMAKE_ROOT}/Modules/CMakeDetermineCompileFeatures.cmake)
|
||
|
|
CMAKE_DETERMINE_COMPILE_FEATURES(CUDA)
|
||
|
|
|
||
|
|
if("x${CMAKE_CUDA_SIMULATE_ID}" STREQUAL "xMSVC")
|
||
|
|
set(CMAKE_CUDA_IMPLICIT_LINK_LIBRARIES "${CMAKE_CUDA_HOST_IMPLICIT_LINK_LIBRARIES}")
|
||
|
|
set(CMAKE_CUDA_IMPLICIT_LINK_DIRECTORIES "${CMAKE_CUDA_HOST_IMPLICIT_LINK_DIRECTORIES}")
|
||
|
|
endif()
|
||
|
|
|
||
|
|
# Filter out implicit link libraries that should not be passed unconditionally.
|
||
|
|
# See CMAKE_CUDA_IMPLICIT_LINK_LIBRARIES_EXCLUDE in CMakeDetermineCUDACompiler.
|
||
|
|
list(REMOVE_ITEM CMAKE_CUDA_IMPLICIT_LINK_LIBRARIES ${CMAKE_CUDA_IMPLICIT_LINK_LIBRARIES_EXCLUDE})
|
||
|
|
|
||
|
|
if(CMAKE_CUDA_COMPILER_ID STREQUAL "NVIDIA")
|
||
|
|
# Remove the CUDA Toolkit include directories from the set of
|
||
|
|
# implicit system include directories.
|
||
|
|
# This resolves the issue that NVCC doesn't specify these
|
||
|
|
# includes as SYSTEM includes when compiling device code, and sometimes
|
||
|
|
# they contain headers that generate warnings, so let users mark them
|
||
|
|
# as SYSTEM explicitly
|
||
|
|
if(CMAKE_CUDA_TOOLKIT_INCLUDE_DIRECTORIES)
|
||
|
|
list(REMOVE_ITEM CMAKE_CUDA_IMPLICIT_INCLUDE_DIRECTORIES
|
||
|
|
${CMAKE_CUDA_TOOLKIT_INCLUDE_DIRECTORIES}
|
||
|
|
)
|
||
|
|
endif()
|
||
|
|
endif()
|
||
|
|
|
||
|
|
# Re-configure to save learned information.
|
||
|
|
configure_file(
|
||
|
|
${CMAKE_ROOT}/Modules/CMakeCUDACompiler.cmake.in
|
||
|
|
${CMAKE_PLATFORM_INFO_DIR}/CMakeCUDACompiler.cmake
|
||
|
|
@ONLY
|
||
|
|
)
|
||
|
|
include(${CMAKE_PLATFORM_INFO_DIR}/CMakeCUDACompiler.cmake)
|
||
|
|
|
||
|
|
unset(__CMAKE_CUDA_COMPILER_OUTPUT)
|