diff options
author | Tomasz Dwojak <t.dwojak@amu.edu.pl> | 2017-11-30 12:56:29 +0300 |
---|---|---|
committer | Tomasz Dwojak <t.dwojak@amu.edu.pl> | 2017-11-30 12:56:29 +0300 |
commit | bb365248eb2ad0b5edc42a4e880c389fb106fa85 (patch) | |
tree | e7829cfa97920768e3aa1273d4403801dbbcd8af /Makefile | |
parent | 8eb15379f12a9aa5350057fe378cf1542fada18f (diff) |
Set USE_CUDNN in makefile
Diffstat (limited to 'Makefile')
-rw-r--r-- | Makefile | 4 |
1 files changed, 3 insertions, 1 deletions
@@ -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 $@ |