69 lines
2.0 KiB
Bash
69 lines
2.0 KiB
Bash
|
#!/usr/bin/env bash
|
||
|
|
set -euo pipefail
|
||
|
|
|
||
|
|
# Larson triad (system/mimalloc/HAKMEM), CSV保存
|
||
|
|
# Usage: benchmarks/scripts/run_larson_matrix.sh [dur_csv] [threads_csv] [reps]
|
||
|
|
# dur_csv default: 2,10 threads_csv default: 1,4 reps default: 5
|
||
|
|
|
||
|
|
ROOT_DIR="$(cd "$(dirname "${BASH_SOURCE[0]}")"/../.. && pwd)"
|
||
|
|
cd "$ROOT_DIR"
|
||
|
|
|
||
|
|
dur_csv=${1:-"2,10"}
|
||
|
|
thr_csv=${2:-"1,4"}
|
||
|
|
reps=${3:-5}
|
||
|
|
|
||
|
|
MIN=8; MAX=128; CHUNKS=1024; ROUNDS=1; SEED=12345
|
||
|
|
MI_LIB_DEFAULT="mimalloc-bench/extern/mi/out/release/libmimalloc.so"
|
||
|
|
MI_LIB="${MIMALLOC_SO:-$MI_LIB_DEFAULT}"
|
||
|
|
|
||
|
|
[[ -x ./larson_system ]] || make -s larson_system >/dev/null
|
||
|
|
if [[ -f "$MI_LIB" ]]; then
|
||
|
|
[[ -x ./larson_mi ]] || make -s larson_mi >/dev/null
|
||
|
|
HAVE_MI=1
|
||
|
|
else
|
||
|
|
HAVE_MI=0
|
||
|
|
fi
|
||
|
|
[[ -x ./larson_hakmem ]] || make -s larson_hakmem >/dev/null
|
||
|
|
|
||
|
|
TS=$(date +%Y%m%d_%H%M%S)
|
||
|
|
OUTDIR="bench_results/auto/larson_${TS}"
|
||
|
|
mkdir -p "$OUTDIR"
|
||
|
|
CSV="$OUTDIR/results.csv"
|
||
|
|
echo "ts,scenario,dur_s,threads,allocator,env,rep,throughput_ops_s" >"$CSV"
|
||
|
|
|
||
|
|
IFS=',' read -ra DLIST <<<"$dur_csv"
|
||
|
|
IFS=',' read -ra TLIST <<<"$thr_csv"
|
||
|
|
|
||
|
|
extract_ops_s() {
|
||
|
|
awk '/Throughput =/{print $3}' | tail -n1
|
||
|
|
}
|
||
|
|
|
||
|
|
run_case() {
|
||
|
|
local dur="$1"; shift
|
||
|
|
local thr="$1"; shift
|
||
|
|
local alloc="$1"; shift
|
||
|
|
local envstr="$1"; shift
|
||
|
|
local rep="$2"; shift
|
||
|
|
local ts=$(date +%H%M%S)
|
||
|
|
local out
|
||
|
|
out=$($envstr ./larson_${alloc} "$dur" "$MIN" "$MAX" "$CHUNKS" "$ROUNDS" "$SEED" "$thr" 2>/dev/null || true)
|
||
|
|
local tput=$(echo "$out" | extract_ops_s)
|
||
|
|
if [[ -n "${tput:-}" ]]; then
|
||
|
|
echo "$ts,larson,$dur,$thr,$alloc,$(echo "$envstr" | sed 's/,/;/g'),$rep,$tput" >>"$CSV"
|
||
|
|
fi
|
||
|
|
}
|
||
|
|
|
||
|
|
echo "[info] writing CSV to $CSV"
|
||
|
|
for d in "${DLIST[@]}"; do
|
||
|
|
for t in "${TLIST[@]}"; do
|
||
|
|
for ((i=1;i<=reps;i++)); do run_case "$d" "$t" system "env -i" "$i"; done
|
||
|
|
if (( HAVE_MI == 1 )); then
|
||
|
|
for ((i=1;i<=reps;i++)); do run_case "$d" "$t" mi "env -i LD_LIBRARY_PATH=$(dirname "$MI_LIB")" "$i"; done
|
||
|
|
fi
|
||
|
|
for ((i=1;i<=reps;i++)); do run_case "$d" "$t" hakmem "env -i HAKMEM_WRAP_TINY=1" "$i"; done
|
||
|
|
done
|
||
|
|
done
|
||
|
|
|
||
|
|
echo "[done] $CSV"
|
||
|
|
|