|
|
|
@ -65,7 +65,7 @@ if(CUDA_FOUND) |
|
|
|
|
if(CUDA_GENERATION STREQUAL "Fermi") |
|
|
|
|
set(__cuda_arch_bin "2.0") |
|
|
|
|
elseif(CUDA_GENERATION STREQUAL "Kepler") |
|
|
|
|
set(__cuda_arch_bin "3.0 3.5") |
|
|
|
|
set(__cuda_arch_bin "3.0 3.5 3.7") |
|
|
|
|
elseif(CUDA_GENERATION STREQUAL "Maxwell") |
|
|
|
|
set(__cuda_arch_bin "5.0") |
|
|
|
|
elseif(CUDA_GENERATION STREQUAL "Pascal") |
|
|
|
@ -92,9 +92,9 @@ if(CUDA_FOUND) |
|
|
|
|
set(__cuda_arch_ptx "") |
|
|
|
|
else() |
|
|
|
|
if(${CUDA_VERSION} VERSION_LESS "8.0") |
|
|
|
|
set(__cuda_arch_bin "2.0 3.0 3.5 5.0") |
|
|
|
|
set(__cuda_arch_bin "2.0 3.0 3.5 3.7 5.0") |
|
|
|
|
else() |
|
|
|
|
set(__cuda_arch_bin "2.0 3.0 3.5 5.0 6.0") |
|
|
|
|
set(__cuda_arch_bin "2.0 3.0 3.5 3.7 5.0 6.0") |
|
|
|
|
endif() |
|
|
|
|
set(__cuda_arch_ptx "") |
|
|
|
|
endif() |
|
|
|
|