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

c1133028 · Zhaolong Xing · GitHub · 1d3b0134 · c1133028
显示空白变更内容
内联并排

Showing with 6 addition and 2 deletion

cmake/cuda.cmake cmake/cuda.cmake +6 -2

未找到文件。
--- 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")
+    if (NOT ${CUDA_VERSION} LESS 10.0)
      add_definitions("-DSUPPORTS_CUDA_FP16")
+    endif()
    set(cuda_arch_bin "70")
  elseif(${CUDA_ARCH_NAME} STREQUAL "Turing")
+    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})