Welcome to mirror list, hosted at ThFree Co, Russian Federation.

github.com/marian-nmt/marian.git - Unnamed repository; edit this file 'description' to name the repository.
summaryrefslogtreecommitdiff
diff options
context:
space:
mode:
authorHieu Hoang <hieuhoang@gmail.com>2018-01-19 15:11:29 +0300
committerHieu Hoang <hieuhoang@gmail.com>2018-01-19 15:11:29 +0300
commit6a210258384e942c8874aab8f9916846fd5556cb (patch)
tree0255e2b7cc3b863221eb49919294af078eb29132
parentc62c7619b86bd5c528ab7b9d56a006d3136e4949 (diff)
runs on gtx 1060
-rw-r--r--CMakeLists.txt2
1 files changed, 1 insertions, 1 deletions
diff --git a/CMakeLists.txt b/CMakeLists.txt
index 2697ad04..1d278448 100644
--- a/CMakeLists.txt
+++ b/CMakeLists.txt
@@ -38,7 +38,7 @@ endif(FPGA)
if(CUDA)
find_package(CUDA)
if(CUDA_FOUND)
- LIST(APPEND CUDA_NVCC_FLAGS --default-stream per-thread; -std=c++11; -O3; -arch=sm_70; -lineinfo; --use_fast_math; )
+ LIST(APPEND CUDA_NVCC_FLAGS --default-stream per-thread; -std=c++11; -O3; -arch=sm_61; -gencode=arch=compute_61,code=compute_61; -gencode=arch=compute_70,code=compute_70; -lineinfo; --use_fast_math; )
# add_definitions(-DCUDA_API_PER_THREAD_DEFAULT_STREAM)
add_definitions(-DCUDA)
SET(CUDA_PROPAGATE_HOST_FLAGS OFF)