@ -81,12 +81,8 @@ 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 } )
list ( APPEND __arch_flags " -gencode arch=compute_${cuda_arch},code=sm_${cuda_arch}" )
endif ( )
endforeach ( )
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 ( )
endfunction ( )
@ -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" )