diff options
| author | Roman Grundkiewicz <rgrundki@exseed.ed.ac.uk> | 2018-02-27 13:52:07 +0300 |
|---|---|---|
| committer | Roman Grundkiewicz <rgrundki@exseed.ed.ac.uk> | 2018-02-27 13:52:07 +0300 |
| commit | a0441c69c0b26357e39a6dd9219d97c58526f056 (patch) | |
| tree | 769dc7c33fc945f9689256a2fff9cd79e0628755 /tests/decoder | |
| parent | 91318a80c91b2f23edd3dc1ba661f10020b298f8 (diff) | |
Add the first test for CPU decoding
Diffstat (limited to 'tests/decoder')
| -rw-r--r-- | tests/decoder/wmt16/test_ende_cpu.sh | 16 |
1 files changed, 16 insertions, 0 deletions
diff --git a/tests/decoder/wmt16/test_ende_cpu.sh b/tests/decoder/wmt16/test_ende_cpu.sh new file mode 100644 index 0000000..5bf5723 --- /dev/null +++ b/tests/decoder/wmt16/test_ende_cpu.sh @@ -0,0 +1,16 @@ +#!/bin/bash + +# Exit on error +set -e + +# Skip if no CUDNN found +if [ ! $MRT_MARIAN_USE_MKL ]; then + exit 100 +fi + +# Test code goes here +$MRT_MARIAN/build/marian-decoder -c $MRT_MODELS/wmt16_systems/marian.en-de.yml --cpu-threads 4 < text.in > text_cpu.out +diff text_cpu.out text.expected > text_cpu.diff + +# Exit with success code +exit 0 |