#!/usr/bin/env bash
# Serve Qwen3.8-27B IU4 Kairic Edge v1.2 as frank-default-large.
# The Kairic runtime requires its authoritative GGUF and all three Prompt Forge
# sidecars. Compatibility mode is enabled for Qwen's recommended sampling.
set -euo pipefail

SCRIPT_DIR="$(cd -- "$(dirname -- "${BASH_SOURCE[0]}")" && pwd)"
ENV_FILE="${FRANK_LARGE_ENV_FILE:-$SCRIPT_DIR/frank-default-large.env}"
if [[ -f "$ENV_FILE" ]]; then
    set -a
    # shellcheck disable=SC1090
    source "$ENV_FILE"
    set +a
fi

: "${LLAMA_BIN:=/home/dank/ciru-qwen38/ROCmFPX/build-kairic/bin/llama-server}"
: "${MODEL_DIR:=/home/dank/models/Kairic-Qwen3.8-27B-IU4-Edge}"
: "${MODEL_FILE:=Qwen3.8-27B-IU4-Kairic-Edge.gguf}"
: "${KAIRIC_FFN_SIDECAR:=$MODEL_DIR/Qwen3.8-27B-Kairic-IU4-FFN.pfs}"
: "${KAIRIC_GDN_SIDECAR:=$MODEL_DIR/Qwen3.8-27B-Kairic-IU4-GDN.pfs}"
: "${KAIRIC_GDN_OUTPUT_SIDECAR:=$MODEL_DIR/Qwen3.8-27B-Kairic-IU4-GDN-Output.pfs}"
: "${ALIAS:=frank-default-large}"
: "${BIND_ADDR:=0.0.0.0}"
: "${LISTEN_PORT:=7770}"
: "${CTX_SIZE:=262144}"
: "${PARALLEL:=1}"
: "${BATCH_SIZE:=2048}"
: "${UBATCH_SIZE:=512}"
: "${NUM_THREADS:=16}"
: "${THREAD_BATCH:=32}"
: "${CACHE_TYPE_K:=f16}"
: "${CACHE_TYPE_V:=f16}"
: "${CTX_CHECKPOINTS:=32}"
: "${CACHE_RAM:=8192}"
: "${MTP_N_MAX:=4}"
: "${MTP_N_MIN:=0}"
: "${MTP_P_MIN:=0.0}"
: "${MTP_P_SPLIT:=0.10}"
: "${TEMPERATURE:=0.7}"
: "${TOP_K:=20}"
: "${TOP_P:=0.8}"
: "${MIN_P:=0.0}"
: "${PRESENCE_PENALTY:=1.5}"
: "${REASONING:=off}"
: "${REASONING_FORMAT:=none}"
: "${KAIRIC_EDGE_COMPATIBILITY_MODE:=1}"
: "${HSA_OVERRIDE_GFX_VERSION:=11.5.1}"

MODEL="$MODEL_DIR/$MODEL_FILE"
for f in "$LLAMA_BIN" "$MODEL" "$KAIRIC_FFN_SIDECAR" "$KAIRIC_GDN_SIDECAR" "$KAIRIC_GDN_OUTPUT_SIDECAR"; do
    if [[ ! -r "$f" ]]; then
        echo "missing or unreadable required artifact: $f" >&2
        exit 1
    fi
done
if [[ ! -x "$LLAMA_BIN" ]]; then
    echo "missing executable runtime: $LLAMA_BIN" >&2
    exit 1
fi
if [[ "$KAIRIC_EDGE_COMPATIBILITY_MODE" != 1 ]]; then
    echo "KAIRIC_EDGE_COMPATIBILITY_MODE must be 1 for this deployment" >&2
    exit 1
fi

SERVER_DIR="$(cd -- "$(dirname -- "$LLAMA_BIN")" && pwd)"
ROCM_PATH="${ROCM_PATH:-/opt/rocm}"
LIBRARY_PATH="$SERVER_DIR:$ROCM_PATH/lib:$ROCM_PATH/lib/rocm_sysdeps/lib:$ROCM_PATH/lib/llvm/lib:$ROCM_PATH/llvm/lib${LD_LIBRARY_PATH:+:$LD_LIBRARY_PATH}"

exec /usr/bin/env \
    ROCM_PATH="$ROCM_PATH" \
    HIP_VISIBLE_DEVICES=0 \
    HSA_OVERRIDE_GFX_VERSION="$HSA_OVERRIDE_GFX_VERSION" \
    LD_LIBRARY_PATH="$LIBRARY_PATH" \
    GGML_CUDA_GRAPH_OPT=0 \
    LLAMA_TARGET_GREEDY_ARGMAX_FASTPATH=0 \
    LLAMA_MTP_CPU_ARGMAX_FASTPATH=1 \
    PROMPTFORGE_TARGET_ONLY=0 \
    PROMPTFORGE_MODE=iu4_ffn \
    PROMPTFORGE_IU4_SIDECAR="$KAIRIC_FFN_SIDECAR" \
    PROMPTFORGE_SIDECAR="$KAIRIC_FFN_SIDECAR" \
    PROMPTFORGE_IU4_HADAMARD=1 \
    PROMPTFORGE_IU4_SEGMENTED=0 \
    PROMPTFORGE_ENABLE_FFN_KEEPERS=0 \
    PROMPTFORGE_FFN_KEEPERS=late6 \
    PROMPTFORGE_ENABLE_GDN=1 \
    PROMPTFORGE_ENABLE_GDN_W8=0 \
    PROMPTFORGE_GDN_SIDECAR_OVERRIDE="$KAIRIC_GDN_SIDECAR" \
    PROMPTFORGE_GDN_SIDECAR="$KAIRIC_GDN_SIDECAR" \
    PROMPTFORGE_GDN_IU4_HADAMARD=1 \
    PROMPTFORGE_ENABLE_GDN_KEEPERS=0 \
    PROMPTFORGE_ENABLE_GDN_OUTPUT=1 \
    PROMPTFORGE_GDN_OUTPUT_SIDECAR="$KAIRIC_GDN_OUTPUT_SIDECAR" \
    PROMPTFORGE_GDN_OUTPUT_KEEPERS=v3_lateq6 \
    PROMPTFORGE_ENABLE_SMALLM_IU4=1 \
    PROMPTFORGE_ENABLE_SMALLM_GDN_IU4=1 \
    PROMPTFORGE_ENABLE_SMALLM_GDN_OUTPUT_IU4=0 \
    PROMPTFORGE_ENABLE_IU4_DECODE_M2_M5=0 \
    PROMPTFORGE_OUTPUT_K8_STRICT_GREEDY=0 \
    "$LLAMA_BIN" -m "$MODEL" --alias "$ALIAS" \
    --host "$BIND_ADDR" --port "$LISTEN_PORT" --jinja \
    -dev ROCm0 -ngl 999 -np "$PARALLEL" -c "$CTX_SIZE" \
    -b "$BATCH_SIZE" -ub "$UBATCH_SIZE" -fa on \
    -ctk "$CACHE_TYPE_K" -ctv "$CACHE_TYPE_V" -t "$NUM_THREADS" -tb "$THREAD_BATCH" \
    -ctxcp "$CTX_CHECKPOINTS" --cache-ram "$CACHE_RAM" --cache-prompt --cache-idle-slots --metrics \
    --kairic-edge --spec-type draft-mtp \
    --spec-draft-device ROCm0 --spec-draft-ngl 999 \
    --spec-draft-type-k "$CACHE_TYPE_K" --spec-draft-type-v "$CACHE_TYPE_V" \
    --spec-draft-n-max "$MTP_N_MAX" --spec-draft-n-min "$MTP_N_MIN" \
    --spec-draft-p-min "$MTP_P_MIN" --spec-draft-p-split "$MTP_P_SPLIT" \
    --spec-draft-backend-sampling \
    --temp "$TEMPERATURE" --top-p "$TOP_P" --top-k "$TOP_K" --min-p "$MIN_P" \
    --presence-penalty "$PRESENCE_PENALTY" \
    --reasoning "$REASONING" --reasoning-format "$REASONING_FORMAT" --reasoning-budget -1