Skip to content

Commit

Permalink
Merge pull request #86 from marian-nmt/usecpu
Browse files Browse the repository at this point in the history
Rename CPU flag from MRT_MARIAN_USE_MKL to MRT_MARIAN_USE_CPU
  • Loading branch information
snukky authored Mar 28, 2023
2 parents 2a8bed3 + efa0fd8 commit 0a00605
Show file tree
Hide file tree
Showing 20 changed files with 31 additions and 31 deletions.
4 changes: 2 additions & 2 deletions run_mrt.sh
Original file line number Diff line number Diff line change
Expand Up @@ -109,7 +109,7 @@ fi
# Check Marian compilation settings
export MRT_MARIAN_BUILD_TYPE=$(cat $MRT_ROOT/cmake.log | grep -i "CMAKE_BUILD_TYPE=" | cut -f2 -d=)
export MRT_MARIAN_COMPILER=$(cat $MRT_ROOT/cmake.log | grep -i "CMAKE_CXX_COMPILER=" | cut -f2 -d=)
export MRT_MARIAN_USE_MKL=$(cat $MRT_ROOT/cmake.log | egrep -i "COMPILE_CPU=(true|on|1)" | cat)
export MRT_MARIAN_USE_CPU=$(cat $MRT_ROOT/cmake.log | egrep -i "COMPILE_CPU=(true|on|1)" | cat)
export MRT_MARIAN_USE_CUDA=$(cat $MRT_ROOT/cmake.log | egrep -i "COMPILE_CUDA=(true|on|1)" | cat)
export MRT_MARIAN_USE_CUDNN=$(cat $MRT_ROOT/cmake.log | egrep -i "USE_CUDNN=(true|on|1)" | cat)
export MRT_MARIAN_USE_SENTENCEPIECE=$(cat $MRT_ROOT/cmake.log | egrep -i "USE_SENTENCEPIECE=(true|on|1)" | cat)
Expand All @@ -118,7 +118,7 @@ export MRT_MARIAN_USE_UNITTESTS=$(cat $MRT_ROOT/cmake.log | egrep -i "COMPIL

log "Build type: $MRT_MARIAN_BUILD_TYPE"
log "Using compiler: $MRT_MARIAN_COMPILER"
log "Using MKL: $MRT_MARIAN_USE_MKL"
log "Using CPU: $MRT_MARIAN_USE_CPU"
log "Using CUDNN: $MRT_MARIAN_USE_CUDNN"
log "Using SentencePiece: $MRT_MARIAN_USE_SENTENCEPIECE"
log "Using FBGEMM: $MRT_MARIAN_USE_FBGEMM"
Expand Down
4 changes: 2 additions & 2 deletions tests/decoder/align/test_align_cpu.sh
Original file line number Diff line number Diff line change
Expand Up @@ -8,8 +8,8 @@
# Exit on error
set -e

# Skip if no MKL found
if [ ! $MRT_MARIAN_USE_MKL ]; then
# Skip if no CPU found
if [ ! $MRT_MARIAN_USE_CPU ]; then
exit 100
fi

Expand Down
2 changes: 1 addition & 1 deletion tests/decoder/intgemm/_test_intgemm_16bit.sh
Original file line number Diff line number Diff line change
Expand Up @@ -9,7 +9,7 @@
set -e

# Skip if requirements are not met
if [ ! $MRT_MARIAN_USE_MKL ]; then
if [ ! $MRT_MARIAN_USE_CPU ]; then
echo "Marian is not compiled with CPU" 1>&2
exit 100
elif ! grep -q "avx" /proc/cpuinfo; then
Expand Down
2 changes: 1 addition & 1 deletion tests/decoder/intgemm/_test_intgemm_8bit.sh
Original file line number Diff line number Diff line change
Expand Up @@ -9,7 +9,7 @@
set -e

# Skip if requirements are not met
if [ ! $MRT_MARIAN_USE_MKL ]; then
if [ ! $MRT_MARIAN_USE_CPU ]; then
echo "Marian is not compiled with CPU" 1>&2
exit 100
elif ! grep -q "avx" /proc/cpuinfo; then
Expand Down
2 changes: 1 addition & 1 deletion tests/decoder/intgemm/_test_intgemm_8bit_shifted.sh
Original file line number Diff line number Diff line change
Expand Up @@ -9,7 +9,7 @@
set -e

# Skip if requirements are not met
if [ ! $MRT_MARIAN_USE_MKL ]; then
if [ ! $MRT_MARIAN_USE_CPU ]; then
echo "Marian is not compiled with CPU" 1>&2
exit 100
elif ! grep -q "avx" /proc/cpuinfo; then
Expand Down
2 changes: 1 addition & 1 deletion tests/decoder/intgemm/test_intgemm_16bit.sh
Original file line number Diff line number Diff line change
Expand Up @@ -9,7 +9,7 @@
set -e

# Skip if requirements are not met
if [ ! $MRT_MARIAN_USE_MKL ]; then
if [ ! $MRT_MARIAN_USE_CPU ]; then
echo "Marian is not compiled with CPU" 1>&2
exit 100
elif ! grep -q "avx" /proc/cpuinfo; then
Expand Down
2 changes: 1 addition & 1 deletion tests/decoder/intgemm/test_intgemm_16bit_avx2.sh
Original file line number Diff line number Diff line change
Expand Up @@ -9,7 +9,7 @@
set -e

# Skip if requirements are not met
if [ ! $MRT_MARIAN_USE_MKL ]; then
if [ ! $MRT_MARIAN_USE_CPU ]; then
echo "Marian is not compiled with CPU" 1>&2
exit 100
elif ! grep -q "avx2" /proc/cpuinfo; then
Expand Down
2 changes: 1 addition & 1 deletion tests/decoder/intgemm/test_intgemm_16bit_sse2.sh
Original file line number Diff line number Diff line change
Expand Up @@ -9,7 +9,7 @@
set -e

# Skip if requirements are not met
if [ ! $MRT_MARIAN_USE_MKL ]; then
if [ ! $MRT_MARIAN_USE_CPU ]; then
echo "Marian is not compiled with CPU" 1>&2
exit 100
elif ! grep -q "sse2" /proc/cpuinfo; then
Expand Down
2 changes: 1 addition & 1 deletion tests/decoder/intgemm/test_intgemm_8bit.sh
Original file line number Diff line number Diff line change
Expand Up @@ -9,7 +9,7 @@
set -e

# Skip if requirements are not met
if [ ! $MRT_MARIAN_USE_MKL ]; then
if [ ! $MRT_MARIAN_USE_CPU ]; then
echo "Marian is not compiled with CPU" 1>&2
exit 100
elif ! grep -q "avx" /proc/cpuinfo; then
Expand Down
2 changes: 1 addition & 1 deletion tests/decoder/intgemm/test_intgemm_8bit_avx2.sh
Original file line number Diff line number Diff line change
Expand Up @@ -9,7 +9,7 @@
set -e

# Skip if requirements are not met
if [ ! $MRT_MARIAN_USE_MKL ]; then
if [ ! $MRT_MARIAN_USE_CPU ]; then
echo "Marian is not compiled with CPU" 1>&2
exit 100
elif ! grep -q "avx2" /proc/cpuinfo; then
Expand Down
2 changes: 1 addition & 1 deletion tests/decoder/intgemm/test_intgemm_8bit_ssse3.sh
Original file line number Diff line number Diff line change
Expand Up @@ -9,7 +9,7 @@
set -e

# Skip if requirements are not met
if [ ! $MRT_MARIAN_USE_MKL ]; then
if [ ! $MRT_MARIAN_USE_CPU ]; then
echo "Marian is not compiled with CPU" 1>&2
exit 100
elif ! grep -q "ssse3" /proc/cpuinfo; then
Expand Down
4 changes: 2 additions & 2 deletions tests/decoder/wmt16/test_ende_cpu.sh
Original file line number Diff line number Diff line change
Expand Up @@ -8,8 +8,8 @@
# Exit on error
set -e

# Skip if no MKL found
if [ ! $MRT_MARIAN_USE_MKL ]; then
# Skip if no CPU found
if [ ! $MRT_MARIAN_USE_CPU ]; then
exit 100
fi

Expand Down
4 changes: 2 additions & 2 deletions tests/models/transformer/test_hard_aligns_cpu.sh
Original file line number Diff line number Diff line change
Expand Up @@ -8,8 +8,8 @@
# Exit on error
set -e

# Skip if no MKL found
if [ ! $MRT_MARIAN_USE_MKL ]; then
# Skip if no CPU found
if [ ! $MRT_MARIAN_USE_CPU ]; then
exit 100
fi

Expand Down
4 changes: 2 additions & 2 deletions tests/models/wnmt18/test_student_small.sh
Original file line number Diff line number Diff line change
Expand Up @@ -8,8 +8,8 @@
# Exit on error
set -e

# Skip if no MKL found
if [ ! $MRT_MARIAN_USE_MKL ]; then
# Skip if no CPU found
if [ ! $MRT_MARIAN_USE_CPU ]; then
exit 100
fi

Expand Down
4 changes: 2 additions & 2 deletions tests/models/wnmt18/test_student_small_aan.sh
Original file line number Diff line number Diff line change
Expand Up @@ -8,8 +8,8 @@
# Exit on error
set -e

# Skip if no MKL found
if [ ! $MRT_MARIAN_USE_MKL ]; then
# Skip if no CPU found
if [ ! $MRT_MARIAN_USE_CPU ]; then
exit 100
fi

Expand Down
4 changes: 2 additions & 2 deletions tests/models/wnmt18/test_student_small_aan_intgemm16.sh
Original file line number Diff line number Diff line change
Expand Up @@ -8,8 +8,8 @@
# Exit on error
set -e

# Skip if no MKL found
if [ ! $MRT_MARIAN_USE_MKL ]; then
# Skip if no CPU found
if [ ! $MRT_MARIAN_USE_CPU ]; then
exit 100
fi

Expand Down
4 changes: 2 additions & 2 deletions tests/models/wnmt18/test_student_small_aan_intgemm8.sh
Original file line number Diff line number Diff line change
Expand Up @@ -8,8 +8,8 @@
# Exit on error
set -e

# Skip if no MKL found
if [ ! $MRT_MARIAN_USE_MKL ]; then
# Skip if no CPU found
if [ ! $MRT_MARIAN_USE_CPU ]; then
exit 100
fi

Expand Down
4 changes: 2 additions & 2 deletions tests/scorer/scores/test_scores_cpu.sh
Original file line number Diff line number Diff line change
Expand Up @@ -8,8 +8,8 @@
# Exit on error
set -e

# Skip if no MKL found
if [ ! $MRT_MARIAN_USE_MKL ]; then
# Skip if no CPU found
if [ ! $MRT_MARIAN_USE_CPU ]; then
exit 100
fi

Expand Down
4 changes: 2 additions & 2 deletions tests/server/test_ende_cpu.sh
Original file line number Diff line number Diff line change
Expand Up @@ -8,8 +8,8 @@
# Exit on error
set -e

# Skip if no MKL found
if [ ! $MRT_MARIAN_USE_MKL ]; then
# Skip if no CPU found
if [ ! $MRT_MARIAN_USE_CPU ]; then
exit 100
fi

Expand Down
4 changes: 2 additions & 2 deletions tests/training/restoring/multi-gpu/test_adam_sync_cpu.sh
Original file line number Diff line number Diff line change
Expand Up @@ -8,8 +8,8 @@
# Exit on error
set -e

# Skip if no MKL found
if [ ! $MRT_MARIAN_USE_MKL ]; then
# Skip if no CPU found
if [ ! $MRT_MARIAN_USE_CPU ]; then
exit 100
fi

Expand Down

0 comments on commit 0a00605

Please sign in to comment.