yams/tests/benchmarks/scripts/benchmark_mode_manifest.sh
Trevon f90c2819bd
perf: improve ingestion optimization and profiling infrastructure (#52)
* perf: improving profile and assertions in ingestion and retrieval paths

* perf: fix tests and profiling infrastructure

* perf: improving profiling implementation and optimizations in ingestion path

* perf: ingestion optimizations and profiling improvements

* perf(ingest): cache dispatch policy resolution

* perf: improving measurements of subsystems

* perf(storage): batch ref-count commits

* perf(ingest): route metadata writes through dedicated writers

* perf: additional optimizations and benchmark docs update

* docs(bench): add profiling bundle helpers

* docs: make pipeline diagram user-facing
2026-06-19 17:32:19 -04:00

64 lines
2.1 KiB
Bash
Executable file

#!/bin/bash
# benchmark_mode_manifest.sh — write a small JSON manifest describing
# benchmark fidelity / realism settings for a run.
set -euo pipefail
OUTPUT_PATH="${1:-}"
if [[ -z "$OUTPUT_PATH" ]]; then
echo "usage: $0 <output-path>" >&2
exit 1
fi
RUNNER_NAME="${RUNNER_NAME:-unknown}"
BENCH_BIN="${BENCH_BIN:-unknown}"
BUILD_DIR="${BUILD_DIR:-unknown}"
DATASET="${YAMS_BENCH_DATASET:-unknown}"
CORPUS_SIZE="${YAMS_BENCH_CORPUS_SIZE:-unknown}"
NUM_QUERIES="${YAMS_BENCH_NUM_QUERIES:-unknown}"
TOPK="${YAMS_BENCH_TOPK:-unknown}"
FORCE_MOCK_EMBEDDINGS="${YAMS_BENCH_FORCE_MOCK_EMBEDDINGS:-0}"
USE_WARM_DATA_DIR="${YAMS_BENCH_WARM_DATA_DIR:-0}"
PLUGIN_DIR="${YAMS_PLUGIN_DIR:-}"
SEARCH_STAGE_TRACE="${YAMS_SEARCH_STAGE_TRACE:-0}"
OPT_LOOP="${YAMS_BENCH_OPT_LOOP:-0}"
DISABLE_VECTORS="${YAMS_DISABLE_VECTORS:-0}"
MODE_LABEL="synthetic"
if [[ "$FORCE_MOCK_EMBEDDINGS" != "1" && "$DISABLE_VECTORS" != "1" ]]; then
MODE_LABEL="live-ish"
fi
if [[ -n "$PLUGIN_DIR" && "$FORCE_MOCK_EMBEDDINGS" != "1" ]]; then
MODE_LABEL="daemon-faithful"
fi
cat >"$OUTPUT_PATH" <<EOF
{
"runner": "$RUNNER_NAME",
"benchmark_binary": "$BENCH_BIN",
"build_dir": "$BUILD_DIR",
"mode_label": "$MODE_LABEL",
"fidelity": {
"real_daemon_startup": false,
"real_plugin_discovery": $([[ -n "$PLUGIN_DIR" ]] && echo true || echo false),
"real_embeddings": $([[ "$FORCE_MOCK_EMBEDDINGS" != "1" && "$DISABLE_VECTORS" != "1" ]] && echo true || echo false),
"real_post_ingest_queues": false,
"warm_data_dir_reuse": $([[ "$USE_WARM_DATA_DIR" == "1" ]] && echo true || echo false),
"synthetic_dataset": $([[ "$DATASET" == synthetic* ]] && echo true || echo false),
"search_stage_trace": $([[ "$SEARCH_STAGE_TRACE" == "1" ]] && echo true || echo false),
"optimization_loop": $([[ "$OPT_LOOP" == "1" ]] && echo true || echo false)
},
"inputs": {
"dataset": "$DATASET",
"corpus_size": "$CORPUS_SIZE",
"num_queries": "$NUM_QUERIES",
"topk": "$TOPK"
},
"notes": [
"mode_label is heuristic: daemon-faithful > live-ish > synthetic",
"Use this manifest to compare benchmark realism before interpreting latency deltas"
]
}
EOF
echo "$OUTPUT_PATH"