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:
authorTomasz Dwojak <t.dwojak@amu.edu.pl>2016-10-05 13:23:59 +0300
committerTomasz Dwojak <t.dwojak@amu.edu.pl>2016-10-05 16:20:15 +0300
commit9f5d666848f09ea1bd861ef7c8befa2180ca077d (patch)
tree1cb882f73945dd37a1d1a9ee9cca0cb51a4d5620 /CMakeLists.txt
parentf4c508c96953dfda235d5a7dddd6756e69193cf4 (diff)
Add mode (CPU|GPU) option
Diffstat (limited to 'CMakeLists.txt')
-rw-r--r--CMakeLists.txt1
1 files changed, 1 insertions, 0 deletions
diff --git a/CMakeLists.txt b/CMakeLists.txt
index 587691d2..5201d1b7 100644
--- a/CMakeLists.txt
+++ b/CMakeLists.txt
@@ -18,6 +18,7 @@ find_package(CUDA)
if(CUDA_FOUND)
LIST(APPEND CUDA_NVCC_FLAGS --default-stream per-thread; -std=c++11; -g; -O3; -arch=sm_35; -lineinfo; --use_fast_math;)
add_definitions(-DCUDA_API_PER_THREAD_DEFAULT_STREAM)
+ add_definitions(-DCUDA)
SET(CUDA_PROPAGATE_HOST_FLAGS OFF)
else(CUDA_FOUND)
add_definitions(-DNO_CUDA)