mirror of
https://github.com/ggerganov/llama.cpp.git
synced 2025-01-12 13:27:21 +01:00
scripts : add helpers script for bench comparing commits (#5521)
* scripts : add helpers script for bench comparing commits * scripts : detect CUDA * set flags after checking the command line * fix make flags --------- Co-authored-by: slaren <slarengh@gmail.com>
This commit is contained in:
parent
4cb0727698
commit
d2819d5577
37
scripts/compare-commits.sh
Executable file
37
scripts/compare-commits.sh
Executable file
@ -0,0 +1,37 @@
|
|||||||
|
#!/bin/bash
|
||||||
|
|
||||||
|
if [ $# -lt 2 ]; then
|
||||||
|
echo "usage: ./scripts/compare-commits.sh <commit1> <commit2> [additional llama-bench arguments]"
|
||||||
|
exit 1
|
||||||
|
fi
|
||||||
|
|
||||||
|
set -e
|
||||||
|
set -x
|
||||||
|
|
||||||
|
bench_args="${@:3}"
|
||||||
|
|
||||||
|
rm -f llama-bench.sqlite
|
||||||
|
|
||||||
|
backend="cpu"
|
||||||
|
|
||||||
|
if [[ "$OSTYPE" == "darwin"* ]]; then
|
||||||
|
backend="metal"
|
||||||
|
elif command -v nvcc &> /dev/null; then
|
||||||
|
backend="cuda"
|
||||||
|
fi
|
||||||
|
|
||||||
|
make_opts=""
|
||||||
|
|
||||||
|
if [[ "$backend" == "cuda" ]]; then
|
||||||
|
make_opts="LLAMA_CUBLAS=1"
|
||||||
|
fi
|
||||||
|
|
||||||
|
git checkout $1
|
||||||
|
make clean && make -j32 $make_opts llama-bench
|
||||||
|
./llama-bench -o sql $bench_args | tee /dev/tty | sqlite3 llama-bench.sqlite
|
||||||
|
|
||||||
|
git checkout $2
|
||||||
|
make clean && make -j32 $make_opts llama-bench
|
||||||
|
./llama-bench -o sql $bench_args | tee /dev/tty | sqlite3 llama-bench.sqlite
|
||||||
|
|
||||||
|
./scripts/compare-llama-bench.py -b $1 -c $2
|
Loading…
x
Reference in New Issue
Block a user