mirror of
https://github.com/Kitware/CMake.git
synced 2026-01-05 21:31:08 -06:00
5df21adf46
When crosscompiling we pass the sysroot.
We need to try various architecture flags. Clang doesn't automatically
select one that works. First try the ones that are more likely to work
for modern installations:
* <=sm_50 is deprecated since CUDA 10.2, try sm_52 first for
future compatibility.
* <=sm_20 is removed since CUDA 9.0, try sm_30.
Otherwise fallback to Clang's current default. Currently that's `sm_20`,
the lowest it supports.
Separable compilation isn't supported yet.
Fixes: #16586
111 lines
4.3 KiB
CMake
111 lines
4.3 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)
|
|
|
|
# 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)
|
|
set(CUDA_TEST_WAS_RUN 1)
|
|
endif()
|
|
|
|
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.")
|
|
else()
|
|
if(CUDA_TEST_WAS_RUN)
|
|
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 ABI and configure it into CMakeCUDACompiler.cmake
|
|
include(${CMAKE_ROOT}/Modules/CMakeDetermineCompilerABI.cmake)
|
|
CMAKE_DETERMINE_COMPILER_ABI(CUDA ${CMAKE_ROOT}/Modules/CMakeCUDACompilerABI.cu)
|
|
# 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()
|
|
|
|
# Remove the following libraries from CMAKE_CUDA_HOST_IMPLICIT_LINK_LIBRARIES and
|
|
# CMAKE_CUDA_IMPLICIT_LINK_LIBRARIES
|
|
#
|
|
# - cudart
|
|
# - cudart_static
|
|
# - cudadevrt
|
|
#
|
|
# Additionally on Linux:
|
|
# - rt
|
|
# - pthread
|
|
# - dl
|
|
#
|
|
# These are controlled by CMAKE_CUDA_RUNTIME_LIBRARY
|
|
list(REMOVE_ITEM CMAKE_CUDA_IMPLICIT_LINK_LIBRARIES cudart cudart_static cudadevrt rt pthread dl)
|
|
list(REMOVE_ITEM CMAKE_CUDA_HOST_IMPLICIT_LINK_LIBRARIES cudart cudart_static cudadevrt rt pthread dl)
|
|
|
|
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)
|
|
endif()
|
|
|
|
|
|
unset(__CMAKE_CUDA_COMPILER_OUTPUT)
|