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>2017-11-24 18:15:47 +0300
committerHieu Hoang <hieuhoang@gmail.com>2017-11-24 18:15:47 +0300
commit94d80c8cfaeaf39992c242ba468cdcf6335f7d49 (patch)
tree18fbc80955ddbf80d07c54292c2c4f1a05dceee8
parentb52005fe23bec266bfda873ee8b9c3ee9ad140b2 (diff)
more compile flag
-rw-r--r--CMakeLists.txt2
1 files changed, 1 insertions, 1 deletions
diff --git a/CMakeLists.txt b/CMakeLists.txt
index c54049c8..dd71f7f7 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; -gencode=arch=compute_50,code=sm_50; -gencode=arch=compute_35,code=sm_35; -lineinfo; --use_fast_math; -DNDEBUG;)
+ LIST(APPEND CUDA_NVCC_FLAGS --default-stream per-thread; -std=c++11; -O3; -gencode=arch=compute_61,code=sm_61; -gencode=arch=compute_50,code=sm_50; -gencode=arch=compute_35,code=sm_35; -lineinfo; --use_fast_math; -DNDEBUG;)
# add_definitions(-DCUDA_API_PER_THREAD_DEFAULT_STREAM)
add_definitions(-DCUDA)
SET(CUDA_PROPAGATE_HOST_FLAGS OFF)