From 00c83b6b2283daf62a1632c02e9b17c1043c2c43 Mon Sep 17 00:00:00 2001 From: denghuilu Date: Wed, 5 Jan 2022 14:26:15 +0800 Subject: [PATCH] fix nvcc warning when using cuda-11.x toolkit --- source/lib/src/cuda/CMakeLists.txt | 6 ++---- 1 file changed, 2 insertions(+), 4 deletions(-) diff --git a/source/lib/src/cuda/CMakeLists.txt b/source/lib/src/cuda/CMakeLists.txt index eb1d1893bb..6941ba9532 100644 --- a/source/lib/src/cuda/CMakeLists.txt +++ b/source/lib/src/cuda/CMakeLists.txt @@ -36,8 +36,7 @@ if (${CUDA_VERSION_MAJOR} GREATER "11") ) elseif (${CUDA_VERSION_MAJOR} STREQUAL "11" AND ${CUDA_VERSION_MINOR} GREATER "0") # nvcc flags - set(CUDA_NVCC_FLAGS -gencode arch=compute_50,code=sm_50; - -gencode arch=compute_52,code=sm_52; # Tesla M40, Tesla M40, Quadro M6000... + set(CUDA_NVCC_FLAGS -gencode arch=compute_52,code=sm_52; # Tesla M40, Tesla M40, Quadro M6000... -gencode arch=compute_53,code=sm_53; -gencode arch=compute_60,code=sm_60; # Pascal – GP100/Tesla P100 – DGX-1 (Generic Pascal) -gencode arch=compute_61,code=sm_61; # Pascal - GTX 1080, GTX 1070, GTX 1060, GTX 1050, GTX 1030, Titan Xp, Tesla P40, Tesla P4, Discrete GPU on the NVIDIA Drive PX2 @@ -49,8 +48,7 @@ elseif (${CUDA_VERSION_MAJOR} STREQUAL "11" AND ${CUDA_VERSION_MINOR} GREATER "0 ) elseif (${CUDA_VERSION_MAJOR} STREQUAL "11" AND ${CUDA_VERSION_MINOR} STREQUAL "0") # nvcc flags - set(CUDA_NVCC_FLAGS -gencode arch=compute_50,code=sm_50; - -gencode arch=compute_52,code=sm_52; # Tesla M40, Tesla M40, Quadro M6000... + set(CUDA_NVCC_FLAGS -gencode arch=compute_52,code=sm_52; # Tesla M40, Tesla M40, Quadro M6000... -gencode arch=compute_53,code=sm_53; -gencode arch=compute_60,code=sm_60; # Pascal – GP100/Tesla P100 – DGX-1 (Generic Pascal) -gencode arch=compute_61,code=sm_61; # Pascal - GTX 1080, GTX 1070, GTX 1060, GTX 1050, GTX 1030, Titan Xp, Tesla P40, Tesla P4, Discrete GPU on the NVIDIA Drive PX2