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()