launcher/recipes/glm53-kindling-6x/recipe.toml

# kindlingai/glm-5.3-flash-gx10 — integration mode B (operator 2026-10-01:
# "image + patches first, their full stack later"). Their IMAGE (15 anchored
# patches + FlashKDA rebuild, baked experimental/ overlay stack via
# docker-images/glm53-kindling/build.sh) launched with sm121's mp rendezvous.
# Every arg/env below is a 1:1 transcription of upstream `image/entrypoint.sh`
# (`exec vllm serve …` and its `case "$TP"` block) plus the
# `experimental/compose/*.yaml` env blocks (arx, snapshot, adaptive-k, fp8,
# megamoe, fixes, sp, recoverssm, tp3-pads) — the stack the upstream README
# headline numbers were measured with. Deviation ledger:
# docker-images/glm53-kindling/README.md.
#
# NOT used (mode A territory): their entrypoint (mentatd roles/election,
# preflight, fabric-check, status server, self-test, GID derivation) — sm121
# owns orchestration, NCCL env comes from the cluster [nccl] block (auto GID
# detect included). Weights: NAS HF cache (upstream demands local disks per
# box — operator call, documented). KV pin: upstream per-TP value on GB10 UMA
# (profiled-unpinned pools starve FlashInfer autotune at boot — upstream).
# TP=6 ONLY, on the ZERO-PADDED checkpoint — first run on a node
# (bootstrap task):
#   python3 experimental/tp3/make_padded.py \
#     /mnt/nas/ai/huggingface/hub/models--nvidia--GLM-5.3-Flash-NVFP4/snapshots/da920bb0b9f... \
#     /mnt/nas/ai/huggingface/glm-5.3-flash-nvfp4-tp6pad \
#     /mnt/nas/ai/huggingface/hub/models--incoai--GLM-5.3-Flash-DFlash2/snapshots/bf582e4eacc... \
#     /mnt/nas/ai/huggingface/glm-5.3-flash-dflash2-tp6pad \
#     --tp 6 --draft-heads 48 --draft-kv-heads 12
# 66 MLA / 66 KDA heads and 2304 MoE columns do not split by 6;
# tp3pad.py zeroes the extra rows at load (baked into the image).
# Build: `sm121 build` in docker-images/glm53-kindling (tag spark-glm53:v9).
# Bench: seed 42 hardmode + perf c1,c2 + depth suite, as everywhere.

engine = "vllm"
name = "GLM-5.3-Flash NVFP4 (kindlingai gx10 stack, 6x, DFlash2)"
docker_image = "spark-glm53:v9"
# shared image build dir (all three TP variants build the same image)
docker_image_dir = "glm53-kindling"
entrypoint = ["vllm", "serve"]
model = "/root/.cache/huggingface/glm-5.3-flash-nvfp4-tp6pad"
parent_model = "nvidia/GLM-5.3-Flash-NVFP4"
port = 30000
required_nodes = 6
tensor_parallel_size = 6
pipeline_parallel_size = 1
distributed_backend = "mp"
shm_size = "32gb"

# GB10 UMA gate (vLLM #48140): reclaim the page cache before the engine startup check reads MemFree — leftover weight/NFS page cache (~10-15 GiB)
# tipped random nodes below the 0.88 GMU threshold at boot.
before_serve = [
  { command = "sudo sh -c 'echo 3 > /proc/sys/vm/drop_caches'", on_failure = "warn" },
]

args = [
    # --- upstream entrypoint.sh `exec vllm serve …`, TP=6 case block -----
    "--served-model-name", "glm53",
    "--distributed-executor-backend", "mp",
    "--load-format", "auto",
    "--safetensors-load-strategy", "eager",
    "--gpu-memory-utilization", "0.88",
    "--max-model-len", "524288",
    "--kv-cache-dtype", "fp8_e4m3",
    "--block-size", "1792",
    "--kv-cache-memory", "27917287424",
    "--trust-remote-code",
    "--enable-prefix-caching",
    "--max-num-batched-tokens", "16384",
    "--max-num-seqs", "64",
    "--long-prefill-token-threshold", "2304",
    # flashinfer_cutlass — native NVFP4 MoE on the checkpoint's per-projection
    # input scales (marlin branch of their entrypoint not used). NO explicit
    # cudagraph cap: adaptive-k's num_speculative_tokens_per_batch_size makes
    # vLLM pick its own ceiling (upstream entrypoint deliberately leaves it).
    "--moe-backend", "flashinfer_cutlass",
    "--limit-mm-per-prompt", '{"image":16,"video":0}',
    # upstream's chat template: thinking:false -> low reasoning effort (the
    # model has no real non-thinking mode; an open  breaks long output)
    "--chat-template", "/usr/local/share/glm53-chat-template.jinja",
    # entrypoint TOOL_ARGS / TOPK_ARGS (vendor parser-name notes apply)
    "--enable-auto-tool-choice",
    "--tool-call-parser", "glm47",
    "--sparse-indexer-topk-backend", "per_row",
    "--reasoning-parser", "glm45",
    # adaptive-k overlay EXTRA_ARGS — upstream appends it AFTER the entrypoint's
    # own spec line, so this JSON is the effective one (same race resolved).
    "--scheduler-cls", "vllm.v1.core.sched.adaptive_k.AdaptiveKScheduler",
    "--speculative-config", """
'{"method":"dflash","disable_eagle_block_drop":true,"model":"/root/.cache/huggingface/glm-5.3-flash-dflash2-tp6pad","num_speculative_tokens":7,"attention_backend":"TRITON_ATTN","num_speculative_tokens_per_batch_size":[[1,1,7],[2,2,5],[3,3,4],[4,4,3],[5,5,2],[6,32,7]]}'""",
    # tp6.yaml: the vision tower does not split by 6
    "--mm-encoder-tp-mode", "data",
]

# Node-local persistence, upstream CACHE_HOME equivalent (processed weight
# snapshots ~48 GB/rank for their ~3-min restarts + JIT tiers + adaptive-k/
# draft-trunc control state). One dir per cache root — NOT /root/.cache whole
# (that would nest over the HF mount).
[[extra_mounts]]
host = "~/sm121-cache/glm53-kindling-tp6/snapshots"
container = "/root/.cache/weight-snapshots"

[[extra_mounts]]
host = "~/sm121-cache/glm53-kindling-tp6/triton"
container = "/root/.cache/triton"

[[extra_mounts]]
host = "~/sm121-cache/glm53-kindling-tp6/tilelang"
container = "/root/.cache/tilelang"

[[extra_mounts]]
host = "~/sm121-cache/glm53-kindling-tp6/flashinfer"
container = "/root/.cache/flashinfer"

[[extra_mounts]]
host = "~/sm121-cache/glm53-kindling-tp6/vllm"
container = "/root/.cache/vllm"

[[extra_mounts]]
host = "~/sm121-cache/glm53-kindling-tp6/adaptive-k"
container = "/root/.cache/adaptive-k"

[[extra_mounts]]
host = "~/sm121-cache/glm53-kindling-tp6/draft-trunc"
container = "/root/.cache/draft-trunc"

[env]
HF_HUB_OFFLINE = "1"
VLLM_ENGINE_READY_TIMEOUT_S = "3600"
PYTORCH_CUDA_ALLOC_CONF = "expandable_segments:True"
TORCH_MEM_FRACTION = "0.92"
MAX_JOBS = "4"
# entrypoint cache layout (dirs above): autotune per-boot and EPHEMERAL — a
# persisted autotune cache deadlocks the next TP>1 boot (upstream measurement)
TRITON_CACHE_DIR = "/root/.cache/triton"
TILELANG_CACHE_DIR = "/root/.cache/tilelang"
VLLM_CACHE_ROOT = "/root/.cache/vllm"
VLLM_FLASHINFER_AUTOTUNE_CACHE_DIR = "/tmp/flashinfer_autotune"
VLLM_WEIGHT_SNAPSHOT_DIR = "/root/.cache/weight-snapshots"
# TP-scoped tag (OURS — upstream snapshots are per TP rank; its single shared
# tag is safe only on a fixed-topology box set; TP2/TP4 must not collide)
VLLM_WEIGHT_SNAPSHOT_TAG = "tp6pad-2026-09-29"
# arx.yaml — RDMA collectives: decode two-shot, prefill RS gather
VLLM_ARX_ALLREDUCE = "1"
VLLM_ARXBIG = "1"
VLLM_ARXBIG_RS = "1"
VLLM_ARXBIG_AG = "0"
VLLM_ARX_TWO_SHOT_MAX_KB = "2048"
# adaptive-k.yaml — online step-cost learner + draft truncation
VLLM_ADAPTIVE_K_CONTROL = "/root/.cache/adaptive-k"
VLLM_ADAPTIVE_K_MODEL = "19.3,0.50,0.40"
VLLM_ADAPTIVE_K_PER_REQUEST = "0"
VLLM_ADAPTIVE_K_CODE_MODE = "gated"
VLLM_ADAPTIVE_K_ONLINE = "1"
VLLM_ADAPTIVE_K_COST_MS = "2:29.7,3:33.1,4:36.5,5:39.8,6:43.1,7:46.0,8:48.9"
VLLM_DRAFT_TRUNC_TAU = "0.3"
VLLM_DRAFT_TRUNC_EXPLORE = "4"
VLLM_DRAFT_TRUNC_CONTROL = "/root/.cache/draft-trunc"
VLLM_DRAFT_TRUNC_LOG = ""
# fp8.yaml — dense linears the checkpoint kept in bf16 -> FP8/NVFP4 (the
# measured ~1% NLL-on-prose cost is upstream's documented tuned default)
VLLM_DENSE_FP8 = "1"
VLLM_DENSE_FP8_LM_HEAD = "1"
VLLM_DENSE_W4 = 'in_proj_qkvbfg_a$|^model\.|self_attn\.o_proj$|shared_experts\.(gate_up_proj|down_proj)$'  # literal string (TOML backslash escapes)
# megamoe.yaml — W4A16 decode MoE (batch<=8) + fused prefill >=1024 tokens
VLLM_MEGAMOE = "1"
VLLM_MEGAMOE_MAX_TOKENS = "8"
VLLM_MEGAMOE_VARIANT = "20"
VLLM_MOE_PREFILL = "1"
VLLM_MOE_PREFILL_MIN_TOKENS = "1024"
VLLM_MOE_PREFILL_CHECK = "0"
VLLM_MOE_PREFILL_Y8 = "1"
# fixes.yaml — triton sparse MLA (3.3x vs FA2), drafter KV pool, fused mHC
# cap, KDA/conv1d copy elision, KV coordinator files
VLLM_MHC_FUSED_MAX_TOKENS = "32"
VLLM_TRITON_SPARSE_MLA = "1"
VLLM_GLM5NEXT_DRAFT_POOL = "1"
VLLM_GLM5NEXT_DRAFT_BLOCK_SIZE = ""
# sp.yaml — sequence-parallel prefill over TP, quantized gathers, arx prefetch
VLLM_GLM_SP_TP = "1"
VLLM_GLM_SP_MIN_TOKENS = "1024"
VLLM_GLM_SP_FP8_GATHER = "1"
VLLM_GLM_ARX_PREFETCH = "1"
VLLM_GLM_SP_MOE_FUSED = "1"
VLLM_GLM_SP_MOE_QUANT_GATHER = "1"
VLLM_GLM_SP_MOE_CHECK = "0"
# recoverssm.yaml — KDA verify from one state per request (seqs raised above)
VLLM_GLM5NEXT_RECOVERSSM = "1"