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

github.com/marian-nmt/marian-regression-tests.git - Unnamed repository; edit this file 'description' to name the repository.
summaryrefslogtreecommitdiff
diff options
context:
space:
mode:
authorTomasz Dwojak <t.dwojak@amu.edu.pl>2017-11-30 12:56:29 +0300
committerTomasz Dwojak <t.dwojak@amu.edu.pl>2017-11-30 12:56:29 +0300
commitbb365248eb2ad0b5edc42a4e880c389fb106fa85 (patch)
treee7829cfa97920768e3aa1273d4403801dbbcd8af /Makefile
parent8eb15379f12a9aa5350057fe378cf1542fada18f (diff)
Set USE_CUDNN in makefile
Diffstat (limited to 'Makefile')
-rw-r--r--Makefile4
1 files changed, 3 insertions, 1 deletions
diff --git a/Makefile b/Makefile
index 8980396..7d6edc8 100644
--- a/Makefile
+++ b/Makefile
@@ -6,11 +6,13 @@ GIT_MOSES_SCRIPTS=http://github.com/marian-nmt/moses-scripts.git
GIT_SUBWORD_NMT=http://github.com/rsennrich/subword-nmt.git
BRANCH=master
+USE_CUDNN=ON
.PHONY: tools/marian-dev tools/marian install tools models data run
.SECONDARY:
+
#####################################################################
install: tools models data
@@ -24,7 +26,7 @@ tools: tools/marian
tools/marian:
git -C $@ pull || git clone $(GIT_MARIAN_DEV) -b $(BRANCH) $@
- mkdir -p $@/build && cd $@/build && cmake .. -DCOMPILE_EXAMPLES=ON -DUSE_CUDNN=ON && make -j$(THREADS)
+ mkdir -p $@/build && cd $@/build && cmake .. -DCOMPILE_EXAMPLES=ON -DUSE_CUDNN=$(USE_CUDNN) && make -j$(THREADS)
models:
mkdir -p $@