recipe.toml 7.8 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180
  1. # kindlingai/glm-5.3-flash-gx10 — integration mode B (operator 2026-10-01:
  2. # "image + patches first, their full stack later"). Their IMAGE (15 anchored
  3. # patches + FlashKDA rebuild, baked experimental/ overlay stack via
  4. # docker-images/glm53-kindling/build.sh) launched with sm121's mp rendezvous.
  5. # Every arg/env below is a 1:1 transcription of upstream `image/entrypoint.sh`
  6. # (`exec vllm serve …` and its `case "$TP"` block) plus the
  7. # `experimental/compose/*.yaml` env blocks (arx, snapshot, adaptive-k, fp8,
  8. # megamoe, fixes, sp, recoverssm, tp3-pads) — the stack the upstream README
  9. # headline numbers were measured with. Deviation ledger:
  10. # docker-images/glm53-kindling/README.md.
  11. #
  12. # NOT used (mode A territory): their entrypoint (mentatd roles/election,
  13. # preflight, fabric-check, status server, self-test, GID derivation) — sm121
  14. # owns orchestration, NCCL env comes from the cluster [nccl] block (auto GID
  15. # detect included). Weights: NAS HF cache (upstream demands local disks per
  16. # box — operator call, documented). KV pin: upstream per-TP value on GB10 UMA
  17. # (profiled-unpinned pools starve FlashInfer autotune at boot — upstream).
  18. # Build: `sm121 build` in docker-images/glm53-kindling (tag spark-glm53:v9).
  19. # Bench: seed 42 hardmode + perf c1,c2 + depth suite, as everywhere.
  20. engine = "vllm"
  21. name = "GLM-5.3-Flash NVFP4 (kindlingai gx10 stack, 4x, DFlash2)"
  22. docker_image = "spark-glm53:v9"
  23. # shared image build dir (all three TP variants build the same image)
  24. docker_image_dir = "glm53-kindling"
  25. entrypoint = ["vllm", "serve"]
  26. model = "/root/.cache/huggingface/hub/models--nvidia--GLM-5.3-Flash-NVFP4/snapshots/da920bb0b9f4a06727223a349e55468e38352348"
  27. parent_model = "nvidia/GLM-5.3-Flash-NVFP4"
  28. port = 30000
  29. required_nodes = 4
  30. tensor_parallel_size = 4
  31. pipeline_parallel_size = 1
  32. distributed_backend = "mp"
  33. shm_size = "32gb"
  34. # GB10 UMA gate (vLLM #48140): reclaim the page cache before the engine startup check reads MemFree — leftover weight/NFS page cache (~10-15 GiB)
  35. # tipped random nodes below the 0.88 GMU threshold at boot.
  36. before_serve = [
  37. { command = "sudo sh -c 'echo 3 > /proc/sys/vm/drop_caches'", on_failure = "warn" },
  38. ]
  39. args = [
  40. # --- upstream entrypoint.sh `exec vllm serve …`, TP=4 case block -----
  41. "--served-model-name", "glm53",
  42. "--distributed-executor-backend", "mp",
  43. "--load-format", "auto",
  44. "--safetensors-load-strategy", "eager",
  45. "--gpu-memory-utilization", "0.88",
  46. "--max-model-len", "524288",
  47. "--kv-cache-dtype", "fp8_e4m3",
  48. "--block-size", "2304",
  49. "--kv-cache-memory", "27917287424",
  50. "--trust-remote-code",
  51. "--enable-prefix-caching",
  52. "--max-num-batched-tokens", "16384",
  53. "--max-num-seqs", "64",
  54. "--long-prefill-token-threshold", "2304",
  55. # flashinfer_cutlass — native NVFP4 MoE on the checkpoint's per-projection
  56. # input scales (marlin branch of their entrypoint not used). NO explicit
  57. # cudagraph cap: adaptive-k's num_speculative_tokens_per_batch_size makes
  58. # vLLM pick its own ceiling (upstream entrypoint deliberately leaves it).
  59. "--moe-backend", "flashinfer_cutlass",
  60. "--limit-mm-per-prompt", '{"image":16,"video":0}',
  61. # upstream's chat template: thinking:false -> low reasoning effort (the
  62. # model has no real non-thinking mode; an open breaks long output)
  63. "--chat-template", "/usr/local/share/glm53-chat-template.jinja",
  64. # entrypoint TOOL_ARGS / TOPK_ARGS (vendor parser-name notes apply)
  65. "--enable-auto-tool-choice",
  66. "--tool-call-parser", "glm47",
  67. "--sparse-indexer-topk-backend", "per_row",
  68. "--reasoning-parser", "glm45",
  69. # adaptive-k overlay EXTRA_ARGS — upstream appends it AFTER the entrypoint's
  70. # own spec line, so this JSON is the effective one (same race resolved).
  71. "--scheduler-cls", "vllm.v1.core.sched.adaptive_k.AdaptiveKScheduler",
  72. "--speculative-config", """
  73. '{"method":"dflash","disable_eagle_block_drop":true,"model":"{{dflash}}","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]]}'""",
  74. ]
  75. # DFlash2 drafter: {{dflash}} resolves per node from the NAS HF
  76. # cache (upstream mounts DFLASH_HOST_DIR on every box instead).
  77. [[extra_mounts]]
  78. repo = "incoai/GLM-5.3-Flash-DFlash2"
  79. container = "dflash"
  80. # Node-local persistence, upstream CACHE_HOME equivalent (processed weight
  81. # snapshots ~48 GB/rank for their ~3-min restarts + JIT tiers + adaptive-k/
  82. # draft-trunc control state). One dir per cache root — NOT /root/.cache whole
  83. # (that would nest over the HF mount).
  84. [[extra_mounts]]
  85. host = "~/sm121-cache/glm53-kindling/snapshots"
  86. container = "/root/.cache/weight-snapshots"
  87. [[extra_mounts]]
  88. host = "~/sm121-cache/glm53-kindling/triton"
  89. container = "/root/.cache/triton"
  90. [[extra_mounts]]
  91. host = "~/sm121-cache/glm53-kindling/tilelang"
  92. container = "/root/.cache/tilelang"
  93. [[extra_mounts]]
  94. host = "~/sm121-cache/glm53-kindling/flashinfer"
  95. container = "/root/.cache/flashinfer"
  96. [[extra_mounts]]
  97. host = "~/sm121-cache/glm53-kindling/vllm"
  98. container = "/root/.cache/vllm"
  99. [[extra_mounts]]
  100. host = "~/sm121-cache/glm53-kindling/adaptive-k"
  101. container = "/root/.cache/adaptive-k"
  102. [[extra_mounts]]
  103. host = "~/sm121-cache/glm53-kindling/draft-trunc"
  104. container = "/root/.cache/draft-trunc"
  105. [env]
  106. HF_HUB_OFFLINE = "1"
  107. VLLM_ENGINE_READY_TIMEOUT_S = "3600"
  108. PYTORCH_CUDA_ALLOC_CONF = "expandable_segments:True"
  109. TORCH_MEM_FRACTION = "0.92"
  110. MAX_JOBS = "4"
  111. # entrypoint cache layout (dirs above): autotune per-boot and EPHEMERAL — a
  112. # persisted autotune cache deadlocks the next TP>1 boot (upstream measurement)
  113. TRITON_CACHE_DIR = "/root/.cache/triton"
  114. TILELANG_CACHE_DIR = "/root/.cache/tilelang"
  115. VLLM_CACHE_ROOT = "/root/.cache/vllm"
  116. VLLM_FLASHINFER_AUTOTUNE_CACHE_DIR = "/tmp/flashinfer_autotune"
  117. VLLM_WEIGHT_SNAPSHOT_DIR = "/root/.cache/weight-snapshots"
  118. # TP-scoped tag (OURS — upstream snapshots are per TP rank; its single shared
  119. # tag is safe only on a fixed-topology box set; TP2/TP4 must not collide)
  120. VLLM_WEIGHT_SNAPSHOT_TAG = "fp8w4c-tp4-2026-09-28"
  121. # arx.yaml — RDMA collectives: decode two-shot, prefill RS gather
  122. VLLM_ARX_ALLREDUCE = "1"
  123. VLLM_ARXBIG = "1"
  124. VLLM_ARXBIG_RS = "1"
  125. VLLM_ARXBIG_AG = "0"
  126. VLLM_ARX_TWO_SHOT_MAX_KB = "2048"
  127. # adaptive-k.yaml — online step-cost learner + draft truncation
  128. VLLM_ADAPTIVE_K_CONTROL = "/root/.cache/adaptive-k"
  129. VLLM_ADAPTIVE_K_MODEL = "26.4,0.660,0.484"
  130. VLLM_ADAPTIVE_K_PER_REQUEST = "0"
  131. VLLM_ADAPTIVE_K_CODE_MODE = "gated"
  132. VLLM_ADAPTIVE_K_ONLINE = "1"
  133. 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"
  134. VLLM_DRAFT_TRUNC_TAU = "0.3"
  135. VLLM_DRAFT_TRUNC_EXPLORE = "4"
  136. VLLM_DRAFT_TRUNC_CONTROL = "/root/.cache/draft-trunc"
  137. VLLM_DRAFT_TRUNC_LOG = ""
  138. # fp8.yaml — dense linears the checkpoint kept in bf16 -> FP8/NVFP4 (the
  139. # measured ~1% NLL-on-prose cost is upstream's documented tuned default)
  140. VLLM_DENSE_FP8 = "1"
  141. VLLM_DENSE_FP8_LM_HEAD = "1"
  142. VLLM_DENSE_W4 = 'in_proj_qkvbfg_a$|^model\.|self_attn\.o_proj$|shared_experts\.(gate_up_proj|down_proj)$' # literal string (TOML backslash escapes)
  143. # megamoe.yaml — W4A16 decode MoE (batch<=8) + fused prefill >=1024 tokens
  144. VLLM_MEGAMOE = "1"
  145. VLLM_MEGAMOE_MAX_TOKENS = "8"
  146. VLLM_MEGAMOE_VARIANT = "20"
  147. VLLM_MOE_PREFILL = "1"
  148. VLLM_MOE_PREFILL_MIN_TOKENS = "1024"
  149. VLLM_MOE_PREFILL_CHECK = "0"
  150. VLLM_MOE_PREFILL_Y8 = "1"
  151. # fixes.yaml — triton sparse MLA (3.3x vs FA2), drafter KV pool, fused mHC
  152. # cap, KDA/conv1d copy elision, KV coordinator files
  153. VLLM_MHC_FUSED_MAX_TOKENS = "32"
  154. VLLM_TRITON_SPARSE_MLA = "1"
  155. VLLM_GLM5NEXT_DRAFT_POOL = "1"
  156. VLLM_GLM5NEXT_DRAFT_BLOCK_SIZE = ""
  157. # sp.yaml — sequence-parallel prefill over TP, quantized gathers, arx prefetch
  158. VLLM_GLM_SP_TP = "1"
  159. VLLM_GLM_SP_MIN_TOKENS = "1024"
  160. VLLM_GLM_SP_FP8_GATHER = "1"
  161. VLLM_GLM_ARX_PREFETCH = "1"
  162. VLLM_GLM_SP_MOE_FUSED = "1"
  163. VLLM_GLM_SP_MOE_QUANT_GATHER = "1"
  164. VLLM_GLM_SP_MOE_CHECK = "0"
  165. # recoverssm.yaml — KDA verify from one state per request (seqs raised above)
  166. VLLM_GLM5NEXT_RECOVERSSM = "1"