Skip to content

Commit b7127c0

Browse files
author
Carsten Griwodz
committed
cuda version condition
1 parent a3028b1 commit b7127c0

File tree

1 file changed

+4
-0
lines changed

1 file changed

+4
-0
lines changed

CMakeLists.txt

Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -81,10 +81,14 @@ set(CUDA_NVCC_FLAGS "${CUDA_NVCC_FLAGS};-gencode;arch=compute_30,code=sm
8181
set(CUDA_NVCC_FLAGS "${CUDA_NVCC_FLAGS};-gencode;arch=compute_35,code=sm_35")
8282
set(CUDA_NVCC_FLAGS "${CUDA_NVCC_FLAGS};-gencode;arch=compute_50,code=sm_50")
8383
set(CUDA_NVCC_FLAGS "${CUDA_NVCC_FLAGS};-gencode;arch=compute_52,code=sm_52")
84+
if(CUDA_VERSION_MAJOR GREATER 7)
8485
set(CUDA_NVCC_FLAGS "${CUDA_NVCC_FLAGS};-gencode;arch=compute_60,code=sm_60")
8586
set(CUDA_NVCC_FLAGS "${CUDA_NVCC_FLAGS};-gencode;arch=compute_61,code=sm_61")
8687
set(CUDA_NVCC_FLAGS "${CUDA_NVCC_FLAGS};-gencode;arch=compute_62,code=sm_62")
8788
set(CUDA_NVCC_FLAGS "${CUDA_NVCC_FLAGS};-gencode;arch=compute_62,code=compute_62")
89+
else(CUDA_VERSION_MAJOR GREATER 7)
90+
set(CUDA_NVCC_FLAGS "${CUDA_NVCC_FLAGS};-gencode;arch=compute_52,code=compute_52")
91+
endif(CUDA_VERSION_MAJOR GREATER 7)
8892

8993
# default stream legacy implies that the 0 stream synchronizes all streams
9094
# default stream per-thread implies that each host thread has one non-synchronizing 0-stream

0 commit comments

Comments
 (0)