chore: add -gencode in CMakeLists.txt (#1603)

Former-commit-id: 2a539125119f471319820255be7d5da0e92116f5

chore: add -gencode in CMakeLists.txt (#1603)
Former-commit-id: 2a539125119f471319820255be7d5da0e92116f5
a5581ba4 · Niu Chong · GitHub · 560354c7 · a5581ba4
隐藏空白更改
内联并排

Showing with 5 addition and 0 deletion

CMakeLists.txt CMakeLists.txt +5 -0

未找到文件。
--- a/CMakeLists.txt
+++ b/CMakeLists.txt
@@ -40,6 +40,11 @@ if (WIN32)
  set(CMAKE_CXX_FLAGS_DEBUG "${CMAKE_CXX_FLAGS_DEBUG} /D_ITERATOR_DEBUG_LEVEL=0")
 else()
  list(APPEND CUDA_NVCC_FLAGS -std=c++11 -w -Wno-deprecated-gpu-targets)
+  list(APPEND CUDA_NVCC_FLAGS -gencode arch=compute_30,code=\"sm_30,compute_30\")
+  list(APPEND CUDA_NVCC_FLAGS -gencode arch=compute_52,code=\"sm_52,compute_52\")
+  list(APPEND CUDA_NVCC_FLAGS -gencode arch=compute_60,code=\"sm_60,compute_60\")
+  list(APPEND CUDA_NVCC_FLAGS -gencode arch=compute_61,code=\"sm_61,compute_61\")
+  list(APPEND CUDA_NVCC_FLAGS -gencode arch=compute_70,code=\"sm_70,compute_70\")
  set(CMAKE_CXX_FLAGS "${CMAKE_CXX_FLAGS} -g -Wall -Wno-sign-compare -Wno-unused-function")
  if (RELEASE_VERSION)
    list(APPEND CUDA_NVCC_FLAGS -O3)