@ -81,13 +81,9 @@ function(specify_cuda_arch cuda_version cuda_arch)
list ( APPEND __arch_flags " -gencode arch=compute_${cuda_arch},code=sm_${cuda_arch}" )
endif ( )
endforeach ( )
elseif ( ${ cuda_version } VERSION_GREATER "7.0" )
foreach ( capability 52 53 )
if ( ${ cuda_arch } STREQUAL ${ capability } )
elseif ( ${ cuda_version } VERSION_GREATER "7.0" and ${ cuda_arch } STREQUAL "53" )
list ( APPEND __arch_flags " -gencode arch=compute_${cuda_arch},code=sm_${cuda_arch}" )
endif ( )
endforeach ( )
endif ( )
endfunction ( )
# C o m m o n g p u a r c h i t e c t u r e s : K e p l e r , M a x w e l l
@ -95,6 +91,10 @@ foreach(capability 30 35 50)
list ( APPEND __arch_flags " -gencode arch=compute_${capability},code=sm_${capability}" )
endforeach ( )
if ( CUDA_VERSION VERSION_GREATER "7.0" )
list ( APPEND __arch_flags " -gencode arch=compute_52,code=sm_52" )
endif ( )
# M o d e r n g p u a r c h i t e c t u r e s : P a s c a l
if ( CUDA_VERSION VERSION_GREATER "8.0" )
list ( APPEND __arch_flags " -gencode arch=compute_60,code=sm_60" )