Skip to content

Commit ba4755d

Browse files
committed
Add benchmark automation scripts and corpus preparation tools
1 parent 1183ddc commit ba4755d

5 files changed

Lines changed: 4062 additions & 0 deletions

File tree

.gitignore

Lines changed: 7 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -20,3 +20,10 @@ DerivedData/
2020
*.pid
2121
curl_out.txt
2222
sample.txt
23+
benchmark_*.txt
24+
*.out
25+
*.json
26+
tmp-*/
27+
server_pid.txt
28+
extreme_benchmark_out.txt
29+

run_turbo.sh

Lines changed: 11 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,11 @@
1+
#!/bin/bash
2+
model="mlx-community/gemma-4-e4b-it-8bit"
3+
echo "Starting TurboKV Server..."
4+
.build/debug/SwiftLM --model $model --port 5420 --turbo-kv > server_turbo.log 2>&1 &
5+
SERVER_PID=$!
6+
sleep 15
7+
echo "Running Python Benchmark..."
8+
python3 tests/run_benchmarks.py --port 5420 --model $model --concurrency 1 --max-tokens 5 --input-multiplier 2500 > bench_turbo.log 2>&1
9+
kill $SERVER_PID
10+
echo "(Done)"
11+
cat bench_turbo.log

0 commit comments

Comments
 (0)