From: Georgi Gerganov <redacted>
Date: Sat, 29 Apr 2023 09:32:18 +0000 (+0300)
Subject: tests : add "threads" to run-tests.sh
X-Git-Tag: upstream/1.7.4~1472
X-Git-Url: https://git.djapps.eu/?a=commitdiff_plain;h=bab97c83d0c9f040bfe1be02c047f272c1a43841;p=pkg%2Fggml%2Fsources%2Fwhisper.cpp

tests : add "threads" to run-tests.sh
---

diff --git a/tests/run-tests.sh b/tests/run-tests.sh
index eda6bbeb..38fa5cea 100755
--- a/tests/run-tests.sh
+++ b/tests/run-tests.sh
@@ -13,7 +13,7 @@
 #
 # Usage:
 #
-#   ./tests/run-tests.sh <model_name>
+#   ./tests/run-tests.sh <model_name> [threads]
 #
 
 cd `dirname $0`
@@ -32,7 +32,7 @@ function list_models {
 }
 
 if [ $# -eq 0 ]; then
-    printf "Usage: $0 [model]\n\n"
+    printf "Usage: $0 [model] [threads]\n\n"
     printf "No model specified. Aborting\n"
     list_models
     exit 1
@@ -41,6 +41,11 @@ fi
 model=$1
 main="../main"
 
+threads=""
+if [ $# -eq 2 ]; then
+    threads="-t $2"
+fi
+
 if [ ! -f ../models/ggml-$model.bin ]; then
     printf "Model $model not found. Aborting\n"
     list_models
@@ -105,7 +110,7 @@ function run_lang() {
             fi
         fi
 
-        $main -m ../models/ggml-$model.bin -f $fname_dst -l $lang -otxt 2> /dev/null
+        $main -m ../models/ggml-$model.bin $threads -f $fname_dst -l $lang -otxt 2> /dev/null
 
         git diff --no-index --word-diff=color --word-diff-regex=. $lang-$i-ref.txt $fname_dst.txt