Skip to content

Commit

Permalink
Add helper script to benchmark all models
Browse files Browse the repository at this point in the history
Simply run:

$ ./extra/bench-all.sh
  • Loading branch information
ggerganov committed Oct 26, 2022
1 parent 2400660 commit a22e574
Showing 1 changed file with 53 additions and 0 deletions.
53 changes: 53 additions & 0 deletions extra/bench-all.sh
Original file line number Diff line number Diff line change
@@ -0,0 +1,53 @@
#!/bin/bash

# Helper script to run the bench tool on all models and print the results in share-able format

printf "Usage: ./bench.sh [n_threads]\n"

if [ -z "$1" ]; then
n_threads=4
else
n_threads=$1
fi

models=( "tiny" "base" "small" "medium" "large" )

printf "\n"
printf "Running benchmark for all models\n"
printf "This can take a while!\n"
printf "\n"

printf "| CPU | OS | Config | Model | Threads | Load [ms] | Encode [ms] |\n"
printf "| --- | -- | ------ | ----- | ------- | --------- | ----------- |\n"

for model in "${models[@]}"; do
# run once to heat-up the cache
./bench -m ./models/ggml-$model.bin -t $n_threads 2>/dev/null 1>/dev/null

# actual run
# store stderr output in a variable in order to parse it later
output=$(./bench -m ./models/ggml-$model.bin -t $n_threads 2>&1)

# parse the output:
load_time=$(echo "$output" | grep "load time" | awk '{print $5}')
encode_time=$(echo "$output" | grep "encode time" | awk '{print $5}')
system_info=$(echo "$output" | grep "system_info")
n_threads=$(echo "$output" | grep "system_info" | awk '{print $4}')

config=""

if [[ $system_info == *"AVX2 = 1"* ]]; then
config="$config AVX2"
fi

if [[ $system_info == *"NEON = 1"* ]]; then
config="$config NEON"
fi

if [[ $system_info == *"BLAS = 1"* ]]; then
config="$config BLAS"
fi

printf "| <todo> | <todo> | $config | $model | $n_threads | $load_time | $encode_time |\n"
done

0 comments on commit a22e574

Please sign in to comment.