diff options
author | jasjuang <jasjuang@gmail.com> | 2018-02-13 16:57:18 -0800 |
---|---|---|
committer | jasjuang <jasjuang@gmail.com> | 2018-02-13 16:57:18 -0800 |
commit | 37e4289024d80632e2c721e865c14be63aab9d8e (patch) | |
tree | 66611e75c2c26a5a70ea7b83126f07c30c941963 /cmake | |
parent | a44c444ee4ae0e7c0aa77118213d34bb26e9f2e6 (diff) | |
download | caffe-37e4289024d80632e2c721e865c14be63aab9d8e.tar.gz caffe-37e4289024d80632e2c721e865c14be63aab9d8e.tar.bz2 caffe-37e4289024d80632e2c721e865c14be63aab9d8e.zip |
fix cuda 9.1 compilation
Diffstat (limited to 'cmake')
-rw-r--r-- | cmake/Cuda.cmake | 6 |
1 files changed, 6 insertions, 0 deletions
diff --git a/cmake/Cuda.cmake b/cmake/Cuda.cmake index 54e26fd5..9325674a 100644 --- a/cmake/Cuda.cmake +++ b/cmake/Cuda.cmake @@ -109,6 +109,12 @@ function(caffe_select_nvcc_arch_flags out_variable) set(__nvcc_flags "") set(__nvcc_archs_readable "") + string(COMPARE GREATER_EQUAL "${CUDA_VERSION}" "9.1" iscudanewerthan91) + if(iscudanewerthan91) + string(REPLACE "21(20)" "" __cuda_arch_bin "${__cuda_arch_bin}") + string(REPLACE "20" "" __cuda_arch_bin "${__cuda_arch_bin}") + endif() + # Tell NVCC to add binaries for the specified GPUs foreach(__arch ${__cuda_arch_bin}) if(__arch MATCHES "([0-9]+)\\(([0-9]+)\\)") |