fix cuda9, volta, turing compile error (#23730)

5 years ago · c113302826
parent 1d3b0134ca
commit c113302826
1 changed files with 6 additions and 2 deletions
--- a/cmake/cuda.cmake
+++ b/cmake/cuda.cmake
@ -101,10 +101,14 @@ function(select_nvcc_arch_flags out_variable)
  elseif(${CUDA_ARCH_NAME} STREQUAL "Pascal")
    set(cuda_arch_bin "60 61")
  elseif(${CUDA_ARCH_NAME} STREQUAL "Volta")
-    add_definitions("-DSUPPORTS_CUDA_FP16")
+    if (NOT ${CUDA_VERSION} LESS 10.0)
+      add_definitions("-DSUPPORTS_CUDA_FP16")
+    endif()
    set(cuda_arch_bin "70")
  elseif(${CUDA_ARCH_NAME} STREQUAL "Turing")
-    add_definitions("-DSUPPORTS_CUDA_FP16")
+    if (NOT ${CUDA_VERSION} LESS 10.0)
+      add_definitions("-DSUPPORTS_CUDA_FP16")
+    endif()
    set(cuda_arch_bin "75")
  elseif(${CUDA_ARCH_NAME} STREQUAL "All")
    set(cuda_arch_bin ${paddle_known_gpu_archs})