Skip to content

Commit 7cf77b2

Browse files
authored
Merge pull request #1 from Xreki/support_cuda80
Support cuda 8.0.
2 parents 0c9475f + 5bc8250 commit 7cf77b2

File tree

1 file changed

+5
-0
lines changed

1 file changed

+5
-0
lines changed

CMakeLists.txt

Lines changed: 5 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -45,6 +45,11 @@ set(CUDA_NVCC_FLAGS "${CUDA_NVCC_FLAGS} -gencode arch=compute_35,code=sm_35")
4545
set(CUDA_NVCC_FLAGS "${CUDA_NVCC_FLAGS} -gencode arch=compute_50,code=sm_50")
4646
set(CUDA_NVCC_FLAGS "${CUDA_NVCC_FLAGS} -gencode arch=compute_52,code=sm_52")
4747

48+
# Modern gpu architectures: Pascal
49+
if (CUDA_VERSION VERSION_GREATER "8.0" OR CUDA_VERSION VERSION_EQUAL "8.0")
50+
set(CUDA_NVCC_FLAGS "${CUDA_NVCC_FLAGS} -gencode arch=compute_60,code=sm_60")
51+
endif()
52+
4853
IF(NOT APPLE)
4954
set(CUDA_NVCC_FLAGS "${CUDA_NVCC_FLAGS} --std=c++11")
5055
set(CUDA_NVCC_FLAGS "${CUDA_NVCC_FLAGS} -Xcompiler -fopenmp")

0 commit comments

Comments
 (0)