K3-Stuff / scripts /27_mini.sh
TessaCoil's picture
Upload folder using huggingface_hub
ddf8c5b verified
Raw
History Blame Contribute Delete
1.19 kB
#!/usr/bin/env bash
# k3-test/27_mini.sh — minimal FA/B-thread knob test, cheap prompts, max_itt
SCRIPT_DIR="$(cd "$(dirname "${BASH_SOURCE[0]}")" && pwd)"
export K3TEST_ROOT="$SCRIPT_DIR"
source "$SCRIPT_DIR/lib/common.sh"
source "$K3TEST_ROOT/lib/server.sh"
set -uo pipefail
[[ -f $K3_MODEL_PATH ]] || die "K3 missing"
MINI_PROMPT="$K3TEST_ROOT/prompts/gen/coding_64tok.txt"
[[ -f $MINI_PROMPT ]] || MINI_PROMPT="$K3TEST_ROOT/prompts/coding.txt"
BASE_SPLIT="0.3,1,1,1,1,1,1,1"
FA_FLAGS=()
[[ "${SRV_FA:-0}" == "1" ]] && FA_FLAGS=(-fa on)
BASE=( -m "$K3_MODEL_PATH" -ngl 999 --tensor-split "$BASE_SPLIT" "${FA_FLAGS[@]}" --cpu-moe -b 256 -ub 256 )
THREADS="${THREADS:-112}"
PRED="${PRED:-64}"
SEL="${SRV_GROUPS:-M}"
# mini group M: fa on/off + cold/warm pair (~in 1-3 min per probe)
if [[ " $SEL " == *" M "* ]]; then
info "### mini: fa=$( [[ ${SRV_FA:-0} == 1 ]] && echo ON || echo OFF )"
if start_server "M" "${BASE[@]}" -np 1 -t "$THREADS"; then
server_completion "mini_cold_fa${SRV_FA:-0}" "$MINI_PROMPT" "$PRED"
server_completion "mini_warm_fa${SRV_FA:-0}" "$MINI_PROMPT" "$PRED"
stop_server
fi
echo "$(date +%s) mini_done" >> "$LOG_ROOT/k3_suite_status.txt"
fi