Fix errors for high version GPU.

ISSUE=4622423 git-svn-id: https://svn.baidu.com/idl/trunk/paddle@1503 1ad973e4-5ce8-4261-8a94-b56d1f490c56
9 years ago · 0b02a221ff
parent 6968191ede
commit 0b02a221ff
1 changed files with 6 additions and 5 deletions
--- a/cmake/flags.cmake
+++ b/cmake/flags.cmake
@ -78,9 +78,10 @@ foreach(capability 30 35 50)
    list(APPEND __arch_flags " -gencode arch=compute_${capability},code=sm_${capability}")
 endforeach()

+message(${__arch_flags})
 if (CUDA_VERSION VERSION_GREATER "7.0")
    list(APPEND __arch_flags " -gencode arch=compute_52,code=sm_52")
 endif()

-set(CUDA_NVCC_FLAGS ${__arch_flags} ${CUDA_NVCC_FLAGS})

+set(CUDA_NVCC_FLAGS ${__arch_flags} ${CUDA_NVCC_FLAGS})