From f14054ec2b49440ad488c3e28612e7a76780bc5f Mon Sep 17 00:00:00 2001 From: AlexeyAB Date: Thu, 30 Apr 2020 02:20:42 +0300 Subject: [PATCH] Fixed Cmake for CUDA CC 7.0, 7.2, 7.5 --- CMakeLists.txt | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/CMakeLists.txt b/CMakeLists.txt index 0b354dc1..700510fd 100644 --- a/CMakeLists.txt +++ b/CMakeLists.txt @@ -65,9 +65,9 @@ else() else() cuda_select_nvcc_arch_flags(CUDA_ARCH_FLAGS ${CUDA_ARCHITECTURES}) message(STATUS "Building with CUDA flags: " "${CUDA_ARCH_FLAGS}") - if (NOT "arch=compute_70,code=sm_70" IN_LIST CUDA_ARCH_FLAGS) + if (NOT "arch=compute_70,code=sm_70" IN_LIST CUDA_ARCH_FLAGS AND NOT "arch=compute_72,code=sm_72" IN_LIST CUDA_ARCH_FLAGS AND NOT "arch=compute_75,code=sm_75" IN_LIST CUDA_ARCH_FLAGS) set(ENABLE_CUDNN_HALF "FALSE" CACHE BOOL "Enable CUDNN Half precision" FORCE) - message(STATUS "Your setup does not supports half precision (it requires CC >= 7.5)") + message(STATUS "Your setup does not supports half precision (it requires CC >= 7.0)") endif() endif() else()