2023-11-01 06:11:15 +08:00
|
|
|
# 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)
|
|
|
|
|
|
|
|
# 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)
|
2023-11-01 06:34:31 +08:00
|
|
|
PrintTestCompilerStatus("CUDA")
|
2023-11-01 06:11:15 +08:00
|
|
|
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)
|
|
|
|
set(CUDA_TEST_WAS_RUN 1)
|
|
|
|
endif()
|
|
|
|
|
|
|
|
if(NOT CMAKE_CUDA_COMPILER_WORKS)
|
2023-11-01 06:34:31 +08:00
|
|
|
PrintTestCompilerResult(CHECK_FAIL "broken")
|
2023-11-01 06:11:15 +08:00
|
|
|
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.")
|
|
|
|
else()
|
|
|
|
if(CUDA_TEST_WAS_RUN)
|
2023-11-01 06:34:31 +08:00
|
|
|
PrintTestCompilerResult(CHECK_PASS "works")
|
2023-11-01 06:11:15 +08:00
|
|
|
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 ABI and configure it into CMakeCUDACompiler.cmake
|
|
|
|
include(${CMAKE_ROOT}/Modules/CMakeDetermineCompilerABI.cmake)
|
|
|
|
CMAKE_DETERMINE_COMPILER_ABI(CUDA ${CMAKE_ROOT}/Modules/CMakeCUDACompilerABI.cu)
|
2023-11-01 06:34:31 +08:00
|
|
|
# Try to identify the compiler features
|
|
|
|
include(${CMAKE_ROOT}/Modules/CMakeDetermineCompileFeatures.cmake)
|
|
|
|
CMAKE_DETERMINE_COMPILE_FEATURES(CUDA)
|
2023-11-01 06:11:15 +08:00
|
|
|
|
|
|
|
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()
|
|
|
|
|
2023-11-01 06:34:31 +08:00
|
|
|
# Remove the following libraries from CMAKE_CUDA_HOST_IMPLICIT_LINK_LIBRARIES and
|
|
|
|
# CMAKE_CUDA_IMPLICIT_LINK_LIBRARIES
|
|
|
|
#
|
|
|
|
# - cudart
|
|
|
|
# - cudart_static
|
|
|
|
# - cudadevrt
|
|
|
|
#
|
|
|
|
# These are controlled by CMAKE_CUDA_RUNTIME_LIBRARY
|
|
|
|
list(REMOVE_ITEM CMAKE_CUDA_IMPLICIT_LINK_LIBRARIES cudart cudart_static cudadevrt)
|
|
|
|
list(REMOVE_ITEM CMAKE_CUDA_HOST_IMPLICIT_LINK_LIBRARIES cudart cudart_static cudadevrt)
|
|
|
|
|
|
|
|
# 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()
|
|
|
|
|
2023-11-01 06:11:15 +08:00
|
|
|
# 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)
|
|
|
|
endif()
|
|
|
|
|
|
|
|
|
|
|
|
unset(__CMAKE_CUDA_COMPILER_OUTPUT)
|