m53297601
/
Paddle
1
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity

modify flags.cmake to support modern gpu arch

Browse Source 
Change-Id: Ie237127bb0f537998827ab29ffde2526c21c6dd1
avx_docs
liaogang 9 years ago committed by Yu Yang
parent eb26ebf99d
commit 2fe39f8c7a
1 changed files with 8 additions and 3 deletions
Show Stats Download Patch File Download Diff File

11
cmake/flags.cmake
Unescape View File

@ -76,7 +76,7 @@ endforeach()
function(specify_cuda_arch cuda_version cuda_arch) function(specify_cuda_arch cuda_version cuda_arch)
if(${cuda_version} VERSION_GREATER "8.0") if(${cuda_version} VERSION_GREATER "8.0")
foreach(capability 60 61 62) foreach(capability 61 62)
if(${cuda_arch} STREQUAL ${capability}) if(${cuda_arch} STREQUAL ${capability})
list(APPEND __arch_flags " -gencode arch=compute_${cuda_arch},code=sm_${cuda_arch}") list(APPEND __arch_flags " -gencode arch=compute_${cuda_arch},code=sm_${cuda_arch}")
endif() endif()
@ -90,12 +90,17 @@ function(specify_cuda_arch cuda_version cuda_arch)
endif() endif()
endfunction() endfunction()
# Common cuda architectures # Common gpu architectures: Kepler, Maxwell
foreach(capability 30 35 50) foreach(capability 30 35 50)
list(APPEND __arch_flags " -gencode arch=compute_${capability},code=sm_${capability}") list(APPEND __arch_flags " -gencode arch=compute_${capability},code=sm_${capability}")
endforeach() endforeach()
# Custom cuda architecture # Modern gpu architectures: Pascal
if (CUDA_VERSION VERSION_GREATER "8.0")
list(APPEND __arch_flags " -gencode arch=compute_60,code=sm_60")
endif()
# Custom gpu architecture
set(CUDA_ARCH) set(CUDA_ARCH)
if(CUDA_ARCH) if(CUDA_ARCH)

Write Preview
Loading…
Cancel
Save