|
|
|
@ -8,7 +8,7 @@ include(CheckCXXSymbolExists)
|
|
|
|
|
# is_c: is C flag or C++ flag, bool type.
|
|
|
|
|
# src_list: The list name which the flag name will be append to.
|
|
|
|
|
# flag_name: the flag name for compiler, such as '-Werror' '-Wall' etc
|
|
|
|
|
# rest arguments: not used.
|
|
|
|
|
# rest arguments: not used.
|
|
|
|
|
function(safe_set_flag is_c src_list flag_name)
|
|
|
|
|
string(REPLACE "-" "_" safe_name ${flag_name})
|
|
|
|
|
string(REPLACE "=" "_" safe_name ${safe_name})
|
|
|
|
@ -44,7 +44,7 @@ CHECK_CXX_SYMBOL_EXISTS(UINT64_MAX "stdint.h" UINT64_MAX_EXISTS)
|
|
|
|
|
if(NOT UINT64_MAX_EXISTS)
|
|
|
|
|
set(CMAKE_REQUIRED_DEFINITIONS -D__STDC_LIMIT_MACROS)
|
|
|
|
|
CHECK_CXX_SYMBOL_EXISTS(UINT64_MAX "stdint.h" UINT64_MAX_EXISTS_HERE)
|
|
|
|
|
if(UINT64_MAX_EXISTS_HERE)
|
|
|
|
|
if(UINT64_MAX_EXISTS_HERE)
|
|
|
|
|
set(CMAKE_REQUIRED_DEFINITIONS)
|
|
|
|
|
add_definitions(-D__STDC_LIMIT_MACROS)
|
|
|
|
|
else()
|
|
|
|
@ -75,12 +75,13 @@ endforeach()
|
|
|
|
|
# So, don't set these flags here.
|
|
|
|
|
|
|
|
|
|
foreach(capability 30 35 50)
|
|
|
|
|
list(APPEND __arch_flags "-gencode arch=compute_${capability},code=sm_${capability}")
|
|
|
|
|
list(APPEND __arch_flags " -gencode arch=compute_${capability},code=sm_${capability}")
|
|
|
|
|
endforeach()
|
|
|
|
|
|
|
|
|
|
message(${__arch_flags})
|
|
|
|
|
if (CUDA_VERSION VERSION_GREATER "7.0")
|
|
|
|
|
list(APPEND __arch_flags "-gencode arch=compute_52,code=sm_52")
|
|
|
|
|
list(APPEND __arch_flags " -gencode arch=compute_52,code=sm_52")
|
|
|
|
|
endif()
|
|
|
|
|
|
|
|
|
|
set(CUDA_NVCC_FLAGS ${__arch_flags} ${CUDA_NVCC_FLAGS})
|
|
|
|
|
|
|
|
|
|
set(CUDA_NVCC_FLAGS ${__arch_flags} ${CUDA_NVCC_FLAGS})
|
|
|
|
|