Skip to content

Commit db582f3

Browse files
LuLu
authored andcommitted
move message out
1 parent 79faee4 commit db582f3

File tree

1 file changed

+12
-19
lines changed

1 file changed

+12
-19
lines changed

source/op/cuda/CMakeLists.txt

Lines changed: 12 additions & 19 deletions
Original file line numberDiff line numberDiff line change
@@ -17,21 +17,20 @@ SET(CMAKE_CUDA_STANDARD 11)
1717
# for searching device compute capability, https://developer.nvidia.com/cuda-gpus
1818
include_directories(cub)
1919

20+
message(STATUS "CUDA major version is " ${CUDA_VERSION_MAJOR})
21+
2022
if (${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
)
4746
elseif (${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
)
6158
elseif (${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
)
7469
elseif (${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

Comments
 (0)