large: ubatch 4096 (match v3), add --no-mmap; sync bench harness prod config

This commit is contained in:
Frank committed 2026-08-22 03:02:16 -07:00
1 parent 0f2bb636d7
commit bd9d75af3f
3 files changed
+19 -12

No files matched your search

+4 -3
View File
@@ -7,9 +7,10 @@
# greedy argmax fast path ON) # greedy argmax fast path ON)
# nommap runner + --no-mmap # nommap runner + --no-mmap
# nofast runner + fast path OFF # nofast runner + fast path OFF
# prod production profile (b4096/ub2048, reasoning on/deepseek, # prod production profile (b4096/ub4096, --no-mmap, reasoning on/deepseek,
# fast path OFF, preserve_thinking) # fast path OFF, preserve_thinking)
# prodnommap prod + --no-mmap # prodnommap (obsolete: identical to prod since ubatch 4096 + --no-mmap
# moved into the production profile)
# #
# Usage: ./scripts/bench-kairic-ab.sh <cfg> [ctx] [prompt_chars] # Usage: ./scripts/bench-kairic-ab.sh <cfg> [ctx] [prompt_chars]
# Example: ./scripts/bench-kairic-ab.sh prod 16384 53000 # Example: ./scripts/bench-kairic-ab.sh prod 16384 53000
@@ -61,7 +62,7 @@ case "$CFG" in
runner) export LLAMA_TARGET_GREEDY_ARGMAX_FASTPATH=1; B=2048; UB=512; REA=off; RFMT=none; EXTRA="" ;; runner) export LLAMA_TARGET_GREEDY_ARGMAX_FASTPATH=1; B=2048; UB=512; REA=off; RFMT=none; EXTRA="" ;;
nommap) export LLAMA_TARGET_GREEDY_ARGMAX_FASTPATH=1; B=2048; UB=512; REA=off; RFMT=none; EXTRA=""; MMAP="--no-mmap" ;; nommap) export LLAMA_TARGET_GREEDY_ARGMAX_FASTPATH=1; B=2048; UB=512; REA=off; RFMT=none; EXTRA=""; MMAP="--no-mmap" ;;
nofast) export LLAMA_TARGET_GREEDY_ARGMAX_FASTPATH=0; B=2048; UB=512; REA=off; RFMT=none; EXTRA="" ;; nofast) export LLAMA_TARGET_GREEDY_ARGMAX_FASTPATH=0; B=2048; UB=512; REA=off; RFMT=none; EXTRA="" ;;
prod) export LLAMA_TARGET_GREEDY_ARGMAX_FASTPATH=0; B=4096; UB=2048; REA=on; RFMT=deepseek; EXTRA="--chat-template-kwargs {\"preserve_thinking\":\"true\"}" ;; prod) export LLAMA_TARGET_GREEDY_ARGMAX_FASTPATH=0; B=4096; UB=4096; REA=on; RFMT=deepseek; EXTRA="--chat-template-kwargs {\"preserve_thinking\":\"true\"}"; MMAP="--no-mmap" ;;
prodnommap) export LLAMA_TARGET_GREEDY_ARGMAX_FASTPATH=0; B=4096; UB=2048; REA=on; RFMT=deepseek; EXTRA="--chat-template-kwargs {\"preserve_thinking\":\"true\"}"; MMAP="--no-mmap" ;; prodnommap) export LLAMA_TARGET_GREEDY_ARGMAX_FASTPATH=0; B=4096; UB=2048; REA=on; RFMT=deepseek; EXTRA="--chat-template-kwargs {\"preserve_thinking\":\"true\"}"; MMAP="--no-mmap" ;;
esac esac
+3 -2
View File
@@ -51,9 +51,10 @@ LISTEN_PORT=52625
CTX_SIZE=262144 CTX_SIZE=262144
PARALLEL=1 PARALLEL=1
# Local override: the pinned runner uses 2048/512; production chat uses a # Local override: the pinned runner uses 2048/512; production chat uses a
# larger 4096/2048 batch (unified memory has headroom). # larger 4096/4096 batch (unified memory has headroom; matches the retired
# v3 profile).
BATCH_SIZE=4096 BATCH_SIZE=4096
UBATCH_SIZE=2048 UBATCH_SIZE=4096
TARGET_THREADS=16 TARGET_THREADS=16
BATCH_THREADS=32 BATCH_THREADS=32
CACHE_TYPE_K=f16 CACHE_TYPE_K=f16
+12 -7
View File
@@ -12,11 +12,13 @@
# See docs/kairic-edge.md for the build runbook and binary identity. # See docs/kairic-edge.md for the build runbook and binary identity.
# #
# This script replicates the release pinned runner # This script replicates the release pinned runner
# (ROCmFPX-kairic/scripts/run-kairic-edge-gfx1151.sh) verbatim. The only # (ROCmFPX-kairic/scripts/run-kairic-edge-gfx1151.sh). The deliberate
# deliberate differences: --host/--port come from BIND_ADDR/LISTEN_PORT # differences: --host/--port come from BIND_ADDR/LISTEN_PORT (safe names;
# (safe names; the ROCm HIP runtime reads a bare HOST env var, see the # the ROCm HIP runtime reads a bare HOST env var, see the 2026-08-13 TEMP
# 2026-08-13 TEMP incident) and LD_LIBRARY_PATH uses this machine's split # incident), LD_LIBRARY_PATH uses this machine's split 7.15 SDK layout,
# 7.15 SDK layout. # the profile runs --no-mmap, batch 4096 / ubatch 4096, reasoning on with
# deepseek format and preserve_thinking, and the target greedy argmax fast
# path is disabled for non-greedy chat sampling.
# #
# Tunables come from frank-default-large.env in this directory unless # Tunables come from frank-default-large.env in this directory unless
# FRANK_LARGE_ENV_FILE overrides that path. # FRANK_LARGE_ENV_FILE overrides that path.
@@ -48,7 +50,7 @@ fi
: "${CTX_SIZE:=262144}" : "${CTX_SIZE:=262144}"
: "${PARALLEL:=1}" : "${PARALLEL:=1}"
: "${BATCH_SIZE:=4096}" : "${BATCH_SIZE:=4096}"
: "${UBATCH_SIZE:=2048}" : "${UBATCH_SIZE:=4096}"
: "${TARGET_THREADS:=16}" : "${TARGET_THREADS:=16}"
: "${BATCH_THREADS:=32}" : "${BATCH_THREADS:=32}"
: "${CACHE_TYPE_K:=f16}" : "${CACHE_TYPE_K:=f16}"
@@ -126,7 +128,9 @@ export PROMPTFORGE_OUTPUT_K8_STRICT_GREEDY=0
# 4, deterministic sampler). Local overrides for production chat: # 4, deterministic sampler). Local overrides for production chat:
# - reasoning on + deepseek format (thinking goes to message.reasoning_content, # - reasoning on + deepseek format (thinking goes to message.reasoning_content,
# matching the retired v3 service so the fleet's reasoning clients work) # matching the retired v3 service so the fleet's reasoning clients work)
# - batch 4096 / ubatch 2048 (larger than the runner's 2048/512) # - batch 4096 / ubatch 4096 (larger than the runner's 2048/512; matches v3)
# - --no-mmap (local choice; benchmarks show no perf difference versus
# mmap, kept per operator preference)
# - chat_template_kwargs preserve_thinking=true (preserve thinking tokens # - chat_template_kwargs preserve_thinking=true (preserve thinking tokens
# in the conversation history across turns) # in the conversation history across turns)
# Sampling is overridden per request by clients; the card recommends Qwen # Sampling is overridden per request by clients; the card recommends Qwen
@@ -152,6 +156,7 @@ exec "$LLAMA_BIN" \
--cache-ram "$CACHE_RAM" \ --cache-ram "$CACHE_RAM" \
--cache-prompt \ --cache-prompt \
--cache-idle-slots \ --cache-idle-slots \
--no-mmap \
--metrics \ --metrics \
--kairic-edge \ --kairic-edge \
--spec-type draft-mtp \ --spec-type draft-mtp \