@@ -17,21 +17,20 @@ SET(CMAKE_CUDA_STANDARD 11)
1717# for searching device compute capability, https://developer.nvidia.com/cuda-gpus
1818include_directories (cub)
1919
20+ message (STATUS "CUDA major version is " ${CUDA_VERSION_MAJOR} )
21+
2022if (${CUDA_VERSION_MAJOR} GREATER "10" )
21- message (STATUS "CUDA major version is " ${CUDA_VERSION_MAJOR} )
22- # nvcc flags
23- set (CUDA_NVCC_FLAGS -gencode arch=compute_50,code=sm_50;
24- -gencode arch=compute_52,code=sm_52; # Tesla M40, Tesla M40, Quadro M6000...
25- -gencode arch=compute_53,code=sm_53;
26- -gencode arch=compute_60,code=sm_60; # Pascal – GP100/Tesla P100 – DGX-1 (Generic Pascal)
27- -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
28- -gencode arch=compute_70,code=sm_70; # Volta - GV100/Tesla V100, GTX 1180 (GV104)
29- -gencode arch=compute_75,code=sm_75; # Turing - RTX 2080, Titan RTX, Quadro R8000
30- -O3; -Xcompiler -fPIC;
31- )
32- elseif (${CUDA_VERSION_MAJOR} STREQUAL "10" )
33- message (STATUS "CUDA major version is " ${CUDA_VERSION_MAJOR} )
3423 # nvcc flags
24+ set (CUDA_NVCC_FLAGS -gencode arch=compute_50,code=sm_50;
25+ -gencode arch=compute_52,code=sm_52; # Tesla M40, Tesla M40, Quadro M6000...
26+ -gencode arch=compute_53,code=sm_53;
27+ -gencode arch=compute_60,code=sm_60; # Pascal – GP100/Tesla P100 – DGX-1 (Generic Pascal)
28+ -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
29+ -gencode arch=compute_70,code=sm_70; # Volta - GV100/Tesla V100, GTX 1180 (GV104)
30+ -gencode arch=compute_75,code=sm_75; # Turing - RTX 2080, Titan RTX, Quadro R8000
31+ -O3; -Xcompiler -fPIC;
32+ )
33+ elseif (${CUDA_VERSION_MAJOR} STREQUAL "10" )
3534 set (CUDA_NVCC_FLAGS -gencode arch=compute_30,code=sm_30; # Tesla K10, Quadro K600 K420 K410,
3635 -gencode arch=compute_35,code=sm_35; # Tesla K20 K40, TITAN Z Black, GTX 780Ti 780
3736 -gencode arch=compute_37,code=sm_37; # Tesla K80
@@ -45,8 +44,6 @@ elseif (${CUDA_VERSION_MAJOR} STREQUAL "10")
4544 -O3; -Xcompiler -fPIC;
4645 )
4746elseif (${CUDA_VERSION_MAJOR} STREQUAL "9" )
48- message (STATUS "CUDA major version is " ${CUDA_VERSION_MAJOR} )
49- # nvcc flags
5047 set (CUDA_NVCC_FLAGS -gencode arch=compute_30,code=sm_30;
5148 -gencode arch=compute_35,code=sm_35;
5249 -gencode arch=compute_37,code=sm_37;
@@ -59,8 +56,6 @@ elseif (${CUDA_VERSION_MAJOR} STREQUAL "9")
5956 -O3; -Xcompiler -fPIC;
6057 )
6158elseif (${CUDA_VERSION_MAJOR} STREQUAL "8" )
62- message (STATUS "CUDA major version is " ${CUDA_VERSION_MAJOR} )
63- # nvcc flags
6459 set (CUDA_NVCC_FLAGS -gencode arch=compute_30,code=sm_30;
6560 -gencode arch=compute_35,code=sm_35;
6661 -gencode arch=compute_37,code=sm_37;
@@ -72,8 +67,6 @@ elseif (${CUDA_VERSION_MAJOR} STREQUAL "8")
7267 -O3; -Xcompiler -fPIC;
7368 )
7469elseif (${CUDA_VERSION_MAJOR} STREQUAL "7" )
75- message (STATUS "CUDA major version is " ${CUDA_VERSION_MAJOR} )
76- # nvcc flags
7770 set (CUDA_NVCC_FLAGS -gencode arch=compute_30,code=sm_30;
7871 -gencode arch=compute_35,code=sm_35;
7972 -gencode arch=compute_37,code=sm_37;
0 commit comments