35 lines
1.5 KiB
Bash
Executable File
35 lines
1.5 KiB
Bash
Executable File
#!/usr/bin/env bash
|
|
# Benchmark dispatcher
|
|
set -euo pipefail
|
|
|
|
SCRIPT_DIR="$(cd "$(dirname "${BASH_SOURCE[0]}")/.." && pwd)"
|
|
|
|
case "${1:-help}" in
|
|
setup) exec bash "$SCRIPT_DIR/scripts/benchmark/setup.sh" ;;
|
|
baseline) exec bash "$SCRIPT_DIR/scripts/benchmark/run-baseline.sh" "${@:2}" ;;
|
|
run) exec bash "$SCRIPT_DIR/scripts/benchmark/run-suite.sh" "${@:2}" ;;
|
|
compare) exec bash "$SCRIPT_DIR/scripts/benchmark/compare.sh" "${@:2}" ;;
|
|
*)
|
|
echo "Usage: benchmark <command> [options]"
|
|
echo ""
|
|
echo "Commands:"
|
|
echo " setup Ensure toolboxes and test models are ready"
|
|
echo " baseline Capture pre-optimization baseline"
|
|
echo " run Run full benchmark suite"
|
|
echo " compare Compare two runs (DIR1 DIR2)"
|
|
echo ""
|
|
echo "Filtering options (baseline and run):"
|
|
echo " --max-size GB Only models up to this file size"
|
|
echo " --category LIST Comma-separated: smoke,dense,moe"
|
|
echo " --skip-longctx Skip long-context (32K) tests"
|
|
echo " --reps N Standard test repetitions (default: 5)"
|
|
echo " --kv-types LIST KV cache sweep (e.g. f16,q8_0,q4_0 or q4_0:q8_0)"
|
|
echo ""
|
|
echo "Examples:"
|
|
echo " benchmark baseline --max-size 20 --skip-longctx"
|
|
echo " benchmark run --tag post-opt --category moe"
|
|
echo " benchmark run --tag kv-sweep --kv-types f16,q8_0,q4_0 --context 131072"
|
|
exit 1
|
|
;;
|
|
esac
|