recipe.toml 5.0 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130
  1. # SparkRing shared image — TP4 SWITCHED profile
  2. # (`/opt/sparkring/profiles/glm53-flash-spark-tp4-switched/profile.json`,
  3. # "glm53-flash-spark-tp4-switched-mtp3", status: research-only — "not validated
  4. # on switched hardware"). Plain NCCL (RoCEnante OFF), NCCL_ALGO=Ring,
  5. # KV 24 GiB fp8, max-len 1M, MTP3 with draft TP4. Structural flags
  6. # (--tp/--pp/--nnodes/--node-rank/--master-addr/--master-port/--host/--port)
  7. # come from the launcher (master-port: 25000 vs their 29795 — cosmetic).
  8. # Model: local-inference-lab/GLM-5.3-Flash-NVFP4-Spark @ df116c4f from the
  9. # shared NFS cache, mounted read-only at /models/target as the profile expects.
  10. engine = "vllm"
  11. name = "GLM-5.3-Flash NVFP4-Spark (SparkRing shared img, TP4 switched profile)"
  12. docker_image = "sparkring-shared:local"
  13. docker_image_dir = "sparkring-shared"
  14. model = "local-inference-lab/GLM-5.3-Flash-NVFP4-Spark"
  15. port = 8000
  16. required_nodes = 4
  17. tensor_parallel_size = 4
  18. pipeline_parallel_size = 1
  19. distributed_backend = "mp"
  20. args = [
  21. "--decode-context-parallel-size", "1",
  22. "--cp-kv-cache-interleave-size", "1",
  23. "--disable-custom-all-reduce",
  24. "--mamba-cache-mode", "align",
  25. "--limit-mm-per-prompt", "{\"image\":4,\"video\":1}",
  26. "--enable-chunked-prefill",
  27. "--dtype", "bfloat16",
  28. "--kv-cache-dtype", "fp8",
  29. "--quantization", "modelopt_mixed",
  30. "--attention-backend", "B12X",
  31. "--block-size", "512",
  32. "--mamba-block-size", "512",
  33. "--recurrent-checkpoint-policy", "aligned",
  34. "--prefix-cache-retention-interval", "0",
  35. "--moe-backend", "b12x",
  36. "--linear-backend", "b12x",
  37. "--no-enable-flashinfer-autotune",
  38. "--load-format", "fastsafetensors",
  39. "--enable-auto-tool-choice",
  40. "--tool-call-parser", "glm47",
  41. "--reasoning-parser", "glm45",
  42. "--kda-prefill-backend", "b12x",
  43. "--gpu-memory-utilization", "0.80",
  44. "--kv-cache-memory-bytes", "25769803776",
  45. "--max-model-len", "1048576",
  46. "--max-num-seqs", "16",
  47. "--max-num-batched-tokens", "8192",
  48. "--prefill-schedule-interval", "2",
  49. "--speculative-config", "{\"method\":\"mtp\",\"num_speculative_tokens\":3,\"draft_tensor_parallel_size\":4,\"kv_cache_dtype\":\"auto\",\"draft_sample_method\":\"probabilistic\",\"rejection_sample_method\":\"standard\",\"draft_load_config\":{\"load_format\":\"safetensors\"},\"attention_backend\":\"B12X\"}",
  50. "--compilation-config", "{\"cudagraph_mode\":\"FULL_AND_PIECEWISE\",\"cudagraph_capture_sizes\":[4,8,12,16,20,24,28,32,36,40,44,48,52,56,60,64],\"custom_ops\":[\"all\"],\"pass_config\":{\"fuse_allreduce_rms\":false}}",
  51. "--max-cudagraph-capture-size", "64",
  52. "--async-scheduling",
  53. "--enable-prefix-caching",
  54. "--cudagraph-metrics",
  55. "--enable-prompt-tokens-details",
  56. "--served-model-name", "glm-5.3-flash-spark",
  57. ]
  58. [env]
  59. PYTHONDONTWRITEBYTECODE = "1"
  60. VLLM_SPARK_SHARED_CAPTURE_STREAM = "0"
  61. SPARK_TP4_HEALTH_GATE = "0"
  62. SPARK_CUDAGRAPH_REPLAY_TIMING = "0"
  63. SPARK_TP4_DCP_COLLECTIVE_AUDIT = "0"
  64. SPARK_GDN_PREFILL_CHECKPOINTS = "0"
  65. SPARK_GDN_CONTINUATION_CHECKPOINTS = "0"
  66. SPARK_MHC_PREFILL_SHARD = "0"
  67. SPARK_MTP_CAPTURE_TAIL_MAX_REQS = "0"
  68. VLLM_ENABLE_PCIE_ALLREDUCE = "0"
  69. VLLM_ENABLE_ROCE_ALLREDUCE = "0"
  70. VLLM_ALLREDUCE_USE_FLASHINFER = "0"
  71. VLLM_ALLREDUCE_USE_SYMM_MEM = "0"
  72. VLLM_USE_NCCL_SYMM_MEM = "0"
  73. NCCL_SWITCHLESS_RING_ONLY = "0"
  74. NCCL_IB_SUBNET_AWARE_ROUTING = "0"
  75. NCCL_IB_EXTENDED_IPV4_GIDS = "1"
  76. NCCL_IB_PRESERVE_PCI_DOMAIN = "1"
  77. NCCL_IB_ROUTE_DIAGNOSTICS = "0"
  78. NCCL_NET = "IB"
  79. NCCL_NET_PLUGIN = "none"
  80. NCCL_IB_DISABLE = "0"
  81. NCCL_ALGO = "Ring"
  82. NCCL_PROTO = "LL,LL128,Simple"
  83. NCCL_DEBUG = "WARN"
  84. NCCL_CUMEM_ENABLE = "0"
  85. NCCL_IB_MERGE_NICS = "0"
  86. NCCL_CROSS_NIC = "1"
  87. VLLM_NCCL_SO_PATH = "/opt/sparkring/nccl-pci/libnccl.so.2.30.7"
  88. LD_PRELOAD = "/opt/sparkring/nccl-pci/libnccl.so.2.30.7"
  89. VLLM_USE_V2_MODEL_RUNNER = "1"
  90. VLLM_B12X_KDA_PREFILL_COALESCING = "1"
  91. VLLM_GLM53_MHC_PREFILL_SHARD = "1"
  92. VLLM_GLM53_MHC_PREFILL_DIAGNOSTICS = "1"
  93. VLLM_GLM53_KDA_GATE_SIDE_STREAM = "1"
  94. VLLM_DCP_TOPK_OWNER_MERGE = "0"
  95. VLLM_DCP_OWNER_FUSED_ENDPOINTS = "0"
  96. VLLM_DCP_COMPACT_INDEX_CACHE_OWNER = "0"
  97. VLLM_DCP_COMPACT_INDEX_TENSOR_VOTE = "0"
  98. VLLM_DCP_COMPACT_INDEX_LOCAL_WIDTHS = "0"
  99. VLLM_DCP_COMPACT_INDEX_PROFILE = "0"
  100. SPARKCACHE_ENABLED = "0"
  101. SPARKCACHE_ASYNC_PAGE_CAPTURE = "0"
  102. VLLM_GLM53_SPLIT_TARGET_BLOCK_SIZE = "512"
  103. VLLM_GLM53_SPLIT_MAMBA_BLOCK_SIZE = "512"
  104. VLLM_B12X_MLA_CKV_GATHER = "1"
  105. VLLM_MTP_NVFP4_LM_HEAD = "1"
  106. VLLM_LM_HEAD_A16 = "1"
  107. VLLM_MXFP8_LM_HEAD = "0"
  108. VLLM_B12X_MOE_FP4_FORCE_A16 = "0"
  109. VLLM_GDN_SPEC_DECODE_METADATA_FASTPATH = "1"
  110. VLLM_FASTSAFETENSORS_QUEUE_SIZE = "1"
  111. VLLM_WORKER_MULTIPROC_METHOD = "spawn"
  112. VLLM_NO_USAGE_STATS = "1"
  113. HF_HUB_OFFLINE = "1"
  114. TRANSFORMERS_OFFLINE = "1"
  115. # The serve-with-warmup wrapper polls the API at 127.0.0.1:$PORT — the value
  116. # resolves from the recipe's `port` field, so the two can't drift apart.
  117. PORT = { from = "port" }
  118. SERVED_MODEL_NAME = "glm-5.3-flash-spark"
  119. DFLASH_WARMUP = "1"
  120. DFLASH_WARMUP_CONCURRENCIES = "1"
  121. DFLASH_WARMUP_SHAPE_WORDS = "8"
  122. DFLASH_WARMUP_MAX_TOKENS = "16"
  123. DFLASH_WARMUP_TIMEOUT_SECONDS = "1800"
  124. SPARKRING_WARMUP_TEMPERATURE = "1"
  125. SPARKRING_LIVENESS_ENABLED = "1"
  126. SPARKRING_LIVENESS_PORT = "8001"