blob: a42dd6f86ec745661798f82db7e18ee3c7a88bc0 (
plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
|
#!/bin/bash
#####################################################################
# SUMMARY: Decode intgemm 8bit avx2
# TAGS: cpu student shortlist intgemm
#####################################################################
# Exit on error
set -e
# Skip if requirements are not met
if [ ! $MRT_MARIAN_USE_MKL ]; then
echo "Marian is not compiled with CPU" 1>&2
exit 100
elif ! grep -q "avx2" /proc/cpuinfo; then
echo "Your CPU does not support AVX2, which is required" 1>&2
exit 100
fi
# Outputs differ on CPUs supporting AVX, AVX2 or AVX512
suffix=avx2
if grep -q "avx512_vnni" /proc/cpuinfo; then
suffix=avx512_vnni
elif grep -q "avx512" /proc/cpuinfo; then
suffix=avx512
fi
prefix=intgemm_8bit_avx2
# Remove previous outputs
rm -f $prefix.out $prefix.$suffix.bin
# Pack the model
$MRT_MARIAN/marian-conv -f $MRT_MODELS/student-eten/model.npz -t $prefix.$suffix.bin --gemm-type intgemm8avx2
test -s $prefix.$suffix.bin
# Run test
$MRT_MARIAN/marian-decoder \
-m $prefix.$suffix.bin -v $MRT_MODELS/student-eten/{vocab.spm,vocab.spm} \
-i newstest2018.src -o $prefix.out \
-b 1 --mini-batch 32 --maxi-batch 100 --maxi-batch-sort src -w 128 \
--shortlist $MRT_MODELS/student-eten/lex.s2t 50 50 --cpu-threads 1 \
--quiet-translation
# Print current and expected BLEU for debugging
python3 $MRT_TOOLS/sacrebleu/sacrebleu.py newstest2018.ref < $prefix.out | tee $prefix.out.bleu
# BLEU scores calculated on AVX, AVX2, AVX512 should be very similar, but does not have to be identical
tail -n1 $prefix.*.expected.bleu || true
# Compare with the expected output, allow up to 20 different lines
$MRT_TOOLS/diff.sh $prefix.out $prefix.$suffix.expected 20 > $prefix.diff
# Compare BLEU scores
cut -f3 -d' ' $prefix.out.bleu > $prefix.out.bleu_score
cut -f3 -d' ' $prefix.$suffix.expected.bleu > $prefix.out.bleu_score.expected
$MRT_TOOLS/diff-nums.py -a -p 0.9 $prefix.out.bleu_score{,.expected}
# Exit with success code
exit 0
|