Fixed Cmake for CUDA CC 7.0, 7.2, 7.5

Tossy0423 · Apr 29, 2020 · f14054e · f14054e
1 parent 36c73c5
commit f14054e
Showing 1 changed file with 2 additions and 2 deletions.
diff --git 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()