cmake: add support for Nvidia GPU Tesla K80

This commit is contained in:
Stephane Thiell 2016-09-01 12:07:32 -07:00
parent 671e95d59e
commit 3335963987

View File

@ -65,7 +65,7 @@ if(CUDA_FOUND)
if(CUDA_GENERATION STREQUAL "Fermi") if(CUDA_GENERATION STREQUAL "Fermi")
set(__cuda_arch_bin "2.0") set(__cuda_arch_bin "2.0")
elseif(CUDA_GENERATION STREQUAL "Kepler") 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") elseif(CUDA_GENERATION STREQUAL "Maxwell")
set(__cuda_arch_bin "5.0") set(__cuda_arch_bin "5.0")
elseif(CUDA_GENERATION STREQUAL "Pascal") elseif(CUDA_GENERATION STREQUAL "Pascal")
@ -92,9 +92,9 @@ if(CUDA_FOUND)
set(__cuda_arch_ptx "") set(__cuda_arch_ptx "")
else() else()
if(${CUDA_VERSION} VERSION_LESS "8.0") 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() 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() endif()
set(__cuda_arch_ptx "") set(__cuda_arch_ptx "")
endif() endif()