Merge branch 'dev_bert' into dev_mem_sharing_for_variable_op

Former-commit-id: dca95c0e8bd46e73e52fe09d0be388c881b47c82

Merge branch 'dev_bert' into dev_mem_sharing_for_variable_op
Former-commit-id: dca95c0e8bd46e73e52fe09d0be388c881b47c82
e3978c9f · Xinqi · 8e631a2c · a5581ba4 · e3978c9f
隐藏空白更改
内联并排

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)