GLM-5 与 GLM-5.1¶
1 引言¶
本文档同时适用于 GLM-5 和 GLM-5.1。除非另有说明,本文档中所有关于 GLM-5 的描述、配置和部署流程同样适用于 GLM-5.1。为简洁起见,下文统一使用 GLM-5 指代 GLM-5 和 GLM-5.1。
GLM-5 采用混合专家(MoE)架构,面向复杂的系统工程和长周期智能体任务。
GLM-5模型首次在vllm-ascend:v0.17.0rc1中支持(对于Ascend950DT,该模型从vllm-ascend:v0.23.0rc1开始支持),所有v0.17.0rc1及更高版本均可稳定运行。要使用最新功能(如PD分离、MTP),建议使用最新的候选版本或正式版本。transformers的版本需要升级到5.2.0或更高版本。
本文档将展示该模型的主要验证步骤,包括支持的特性、特性配置、环境准备、单节点和多节点部署、精度及性能评估。
2 支持的特性¶
请参阅支持的功能以获取该模型支持的功能矩阵。
请参阅功能指南以获取该功能的配置。
3 前提条件¶
3.1 模型权重¶
GLM-5-w4a8(量化版本):下载模型权重。GLM-5-w8a8(量化版本):下载模型权重。GLM-5.1-w4a8(量化版本):下载模型权重。GLM-5.1-w8a8(量化版本):下载模型权重。GLM-5.1-w8a8c8(量化版本):下载模型权重。该权重已在Atlas 800 A3上验证,推荐使用。GLM-5.1-w4a4(Ascend950DT mxfp4量化):下载模型权重。该权重已在Ascend 950DT上验证,推荐使用。
建议将模型权重下载到多节点的共享目录中,例如 /root/.cache/
3.2 验证多节点通信(可选)¶
如果需要多节点部署,请按照验证多节点通信环境指南进行通信验证。
4 安装¶
4.1 Docker 镜像安装¶
您可以直接使用我们的官方 Docker 镜像来运行 GLM-5/5.1。
在每个节点上启动docker镜像。
export IMAGE=quay.io/ascend/vllm-ascend:v0.23.0-a5
export NAME=vllm-ascend
docker run --rm \
--name $NAME \
--net=host \
--shm-size=1g \
--device /dev/davinci0 \
--device /dev/davinci1 \
--device /dev/davinci2 \
--device /dev/davinci3 \
--device /dev/davinci4 \
--device /dev/davinci5 \
--device /dev/davinci6 \
--device /dev/davinci7 \
--device /dev/davinci_manager \
--device /dev/hisi_hdc \
--device /dev/ummu \
--device /dev/uburma \
-v /usr/local/Ascend/driver:/usr/local/Ascend/driver \
-v /etc/ascend_install.info:/etc/ascend_install.info \
-v /etc/hccl_rootinfo.json:/etc/hccl_rootinfo.json \
-v /etc/hixlep/:/etc/hixlep/ \
-v /root/.cache:/root/.cache \
-v /usr/local/sbin:/usr/local/sbin \
-v /usr/local/dcmi:/usr/local/dcmi \
-v /usr/local/bin/npu-smi:/usr/local/bin/npu-smi \
-v /usr/local/sbin/npu-smi:/usr/local/sbin/npu-smi \
-v /usr/bin/urma_admin:/usr/bin/urma_admin \
-v /lib/route.conf:/lib/route.conf \
-v /usr/lib64:/usr/lib64 \
-itd $IMAGE bash
在每个节点上启动docker镜像。
export IMAGE=quay.io/ascend/vllm-ascend:v0.23.0-a3
export NAME=vllm-ascend
# Run the container using the defined variables
# Note: If you are running bridge network with docker, please expose available ports for multiple nodes communication in advance
docker run --rm \
--name $NAME \
--net=host \
--shm-size=1g \
--device /dev/davinci0 \
--device /dev/davinci1 \
--device /dev/davinci2 \
--device /dev/davinci3 \
--device /dev/davinci4 \
--device /dev/davinci5 \
--device /dev/davinci6 \
--device /dev/davinci7 \
--device /dev/davinci8 \
--device /dev/davinci9 \
--device /dev/davinci10 \
--device /dev/davinci11 \
--device /dev/davinci12 \
--device /dev/davinci13 \
--device /dev/davinci14 \
--device /dev/davinci15 \
--device /dev/davinci_manager \
--device /dev/devmm_svm \
--device /dev/hisi_hdc \
-v /usr/local/dcmi:/usr/local/dcmi \
-v /usr/local/Ascend/driver/tools/hccn_tool:/usr/local/Ascend/driver/tools/hccn_tool \
-v /usr/local/bin/npu-smi:/usr/local/bin/npu-smi \
-v /usr/local/Ascend/driver/lib64/:/usr/local/Ascend/driver/lib64/ \
-v /usr/local/Ascend/driver/version.info:/usr/local/Ascend/driver/version.info \
-v /etc/ascend_install.info:/etc/ascend_install.info \
-v /root/.cache:/root/.cache \
-it $IMAGE bash
在每个节点上启动docker镜像。
export IMAGE=quay.io/ascend/vllm-ascend:v0.23.0
docker run --rm \
--name vllm-ascend \
--shm-size=1g \
--net=host \
--device /dev/davinci0 \
--device /dev/davinci1 \
--device /dev/davinci2 \
--device /dev/davinci3 \
--device /dev/davinci4 \
--device /dev/davinci5 \
--device /dev/davinci6 \
--device /dev/davinci7 \
--device /dev/davinci_manager \
--device /dev/devmm_svm \
--device /dev/hisi_hdc \
-v /usr/local/dcmi:/usr/local/dcmi \
-v /usr/local/Ascend/driver/tools/hccn_tool:/usr/local/Ascend/driver/tools/hccn_tool \
-v /usr/local/bin/npu-smi:/usr/local/bin/npu-smi \
-v /usr/local/Ascend/driver/lib64/:/usr/local/Ascend/driver/lib64/ \
-v /usr/local/Ascend/driver/version.info:/usr/local/Ascend/driver/version.info \
-v /etc/ascend_install.info:/etc/ascend_install.info \
-v /root/.cache:/root/.cache \
-it $IMAGE bash
如果要部署多节点环境,需要在每个节点上进行环境设置。
要验证环境是否安装成功,请参阅安装。
4.2 源码安装¶
此外,如果您不想使用上述 Docker 镜像,也可以从源码构建所有内容:
- 从源码安装
vllm-ascend,请参阅安装。
如果要部署多节点环境,需要在每个节点上进行环境设置。
5 在线服务部署¶
5.1 单节点在线部署¶
- 量化模型
glm-5-w4a4可以部署在1台Ascend950DT(96GB × 8)上。
运行以下脚本执行在线推理。
常见问题提示:如果遇到问题,请参阅常见问题。
#!/usr/bin/env bash
source /root/.bashrc
# this obtained through ifconfig
# nic_name is the network interface name corresponding to local_ip of the current node
nic_name="xxx"
local_ip="xxx"
export HCCL_BUFFSIZE=400
export HCCL_IF_IP=$local_ip
export HCCL_INTRA_ROCE_ENABLE=0
export HCCL_OP_EXPANSION_MODE="AIV"
export HCCL_SOCKET_IFNAME=$nic_name
export ASCEND_RT_VISIBLE_DEVICES=0,1,2,3,4,5,6,7
export DYNAMIC_EPLB="true"
export GLOO_SOCKET_IFNAME=$nic_name
export PYTORCH_NPU_ALLOC_CONF=expandable_segments:True
export PROMETHEUS_MULTIPROC_DIR=/dev/shm/vllm_metrics && mkdir -p $PROMETHEUS_MULTIPROC_DIR
export HCCL_DFS_CONFIG="task_exception:off,inconsistent_check:off"
export VLLM_ASCEND_ENABLE_PREFETCH_MLP=1
vllm serve /root/.cache/modelscope/hub/models/vllm-ascend/GLM5-w4a4 \
--host 0.0.0.0 \
--port 8077 \
--data-parallel-size 1 \
--tensor-parallel-size 8 \
--seed 1024 \
--served-model-name glm-5 \
--enable-expert-parallel \
--max-num-seqs 128 \
--max-model-len 202752 \
--max-num-batched-tokens 8192 \
--trust-remote-code \
--enable-prefix-caching \
--gpu-memory-utilization 0.95 \
--quantization ascend \
--enable-auto-tool-choice \
--tool-call-parser glm47 \
--reasoning-parser glm45 \
--compilation-config '{"cudagraph_mode": "FULL_DECODE_ONLY"}' \
--hf-overrides '{"use_index_cache": true, "index_topk_freq": 4}' \
--speculative-config '{"num_speculative_tokens": 3, "method": "deepseek_mtp", "enforce_eager": true}' \
--additional-config '{"enable_cpu_binding": "True", "multistream_overlap_shared_expert": "True", "enable_sparse_c8": "True", "enable_dsa_cp": true, "eplb_config": {"dynamic_eplb": true, "expert_heat_collection_interval": 50, "algorithm_execution_interval": 5, "eplb_policy_type": 2, "num_redundant_experts": 0}, "enable_flashcomm1": true}'
- 量化模型
glm-5-w4a8和glm-5.1-w4a8可以部署在1台Atlas 800 A3(128GB × 8)上。
运行以下脚本执行在线推理。
常见问题提示:如果遇到问题,请参阅常见问题。
# The version of transformers needs to be upgraded to 5.2.0.
# pip install transformers==5.2.0 --upgrade
export HCCL_BUFFSIZE=200
export HCCL_OP_EXPANSION_MODE="AIV"
export PYTORCH_NPU_ALLOC_CONF=expandable_segments:True
vllm serve /root/.cache/modelscope/hub/models/vllm-ascend/GLM5-w4a8 \
--host 0.0.0.0 \
--port 8077 \
--data-parallel-size 1 \
--tensor-parallel-size 16 \
--enable-expert-parallel \
--seed 1024 \
--served-model-name glm-5 \
--max-num-seqs 16 \
--max-model-len 200000 \
--max-num-batched-tokens 4096 \
--trust-remote-code \
--gpu-memory-utilization 0.95 \
--quantization ascend \
--enable-chunked-prefill \
--enable-prefix-caching \
--additional-config '{"multistream_overlap_shared_expert": true, "enable_balance_scheduling": true, "enable_flashcomm1": true}' \
--compilation-config '{"cudagraph_mode": "FULL_DECODE_ONLY"}' \
--hf-overrides '{"use_index_cache": true, "index_topk_freq": 4}' \
--speculative-config '{"num_speculative_tokens": 3, "method": "deepseek_mtp", "enforce_eager": true}'
- 量化模型
glm-5-w4a8可部署在1台Atlas 800 A2(64GB × 8)上。
运行以下脚本执行在线推理。
常见问题提示:如果遇到问题,请参阅常见问题。
export HCCL_BUFFSIZE=200
export HCCL_OP_EXPANSION_MODE="AIV"
export PYTORCH_NPU_ALLOC_CONF=expandable_segments:True
vllm serve /root/.cache/modelscope/hub/models/vllm-ascend/GLM5-w4a8 \
--host 0.0.0.0 \
--port 8077 \
--data-parallel-size 1 \
--tensor-parallel-size 8 \
--enable-expert-parallel \
--seed 1024 \
--served-model-name glm-5 \
--max-num-seqs 8 \
--max-model-len 32768 \
--max-num-batched-tokens 4096 \
--trust-remote-code \
--gpu-memory-utilization 0.95 \
--quantization ascend \
--enable-chunked-prefill \
--enable-prefix-caching \
--compilation-config '{"cudagraph_mode": "FULL_DECODE_ONLY"}' \
--additional-config '{"multistream_overlap_shared_expert": true, "enable_balance_scheduling": true, "enable_flashcomm1": true}' \
--hf-overrides '{"use_index_cache": true, "index_topk_freq": 4}' \
--speculative-config '{"num_speculative_tokens": 3, "method": "deepseek_mtp", "enforce_eager": true}'
关键参数说明:
以下仅描述该模型/场景特有的关键参数。max-model-len 和 max-num-seqs 需要根据实际使用场景进行设置。
模型特有参数:
--enable-expert-parallel:对于GLM-5的MoE架构必须启用。--tensor-parallel-size 16/--tensor-parallel-size 8:每个DP rank内的张量并行。对于A3(16个NPU),使用tp16;对于A2(8个NPU),使用tp8。--quantization ascend:为w4a8/w8a8量化权重启用Ascend量化。--speculative-config '{"num_speculative_tokens": 3, "method": "deepseek_mtp", "enforce_eager": true}':使用GLM-5的DeepSeek风格MTP草稿模型启用多令牌预测(MTP)投机解码。num_speculative_tokens(3-5)控制每步推测的令牌数量;enforce_eager: true是必需的,因为GLM-5不支持图模式投机解码。--enable-chunked-prefill/--enable-prefix-caching:推荐用于长上下文和多用户场景——chunked prefill将长提示拆分以改善TTFT,prefix caching为共享前缀(如系统提示)重用KV缓存。--compilation-config '{"cudagraph_mode": "FULL_DECODE_ONLY"}':仅对解码阶段启用图捕获,通过减少内核启动开销来提升解码性能。--additional-config '{"multistream_overlap_shared_expert": true}':在额外的流上重叠共享专家计算。注意:当"enable_fused_mc2": true时自动禁用,因为这两种优化存在冲突。
关键的--additional-config字段:
"enable_flashcomm1": true:启用FlashComm优化以减少通信开销(主要有利于prefill路径)。启用FlashComm后,layer_sharding不能包含o_proj。"enable_mlapo": true:启用MLA预处理融合算子(MlaPreprocessOperation)。对于w8a8模型默认启用——显著提升Decode性能但消耗更多NPU内存;如果内存优先,请设置"enable_mlapo": false。推荐用于w8a8;w4a8可能无法受益。"enable_balance_scheduling": true:启用平衡调度以提升v1调度器中的输出吞吐量并降低TPOT。
单节点性能调优说明:
- 对于低延迟场景,使用
dp1tp16(data-parallel-size 1,tensor-parallel-size 16),并考虑减少--max-num-seqs和--max-num-batched-tokens。 - 对于高吞吐量场景,增加
--max-num-seqs并启用--enable-prefix-caching。 - 对于长上下文场景(如200K),使用w4a8权重(为KV缓存留出更多内存)并将
--max-model-len设置为所需的上下文长度。考虑启用--enable-chunked-prefill。 - 如果遇到OOM,请减少
--gpu-memory-utilization、--max-num-seqs或--max-model-len。禁用"enable_mlapo"也可以减少内存使用(以性能为代价)。
5.2 多节点部署¶
如果要部署多节点环境,需要按照验证多节点通信环境进行多节点通信验证。
常见问题提示:如果遇到问题,请参阅常见问题。
高吞吐量场景(DP8 TP4)
glm-5.1-w8a8c8:可以部署在2台Atlas 800 A3(128GB × 8)上用于高吞吐量场景。
分别在两个节点上运行以下脚本。
节点 0
# this obtained through ifconfig
# nic_name is the network interface name corresponding to local_ip of the current node
nic_name="xxx"
local_ip="xxx"
export HCCL_BUFFSIZE=400
export HCCL_IF_IP=$local_ip
export HCCL_OP_EXPANSION_MODE="AIV"
export HCCL_SOCKET_IFNAME=$nic_name
export GLOO_SOCKET_IFNAME=$nic_name
export PYTORCH_NPU_ALLOC_CONF=expandable_segments:True
vllm serve /root/.cache/modelscope/hub/models/vllm-ascend/GLM-5.1-W8A8C8-MTP \
--host 0.0.0.0 \
--port 8077 \
--data-parallel-size 8 \
--data-parallel-size-local 4 \
--data-parallel-address $local_ip \
--enable-expert-parallel \
--data-parallel-rpc-port 12980 \
--tensor-parallel-size 4 \
--hf-overrides '{"use_index_cache": true, "index_topk_freq": 4}' \
--seed 1024 \
--served-model-name glm-5 \
--tool-call-parser glm47 \
--reasoning-parser glm45 \
--enable-auto-tool-choice \
--trust-remote-code \
--gpu-memory-utilization 0.92 \
--quantization ascend \
--enable-chunked-prefill \
--enable-prefix-caching \
--async-scheduling \
--additional-config '{"enable_dsa_cp": true, "enable_sparse_sfa_c8": true, "enable_sparse_li_c8": true, "enable_balance_scheduling": true, "fuse_muls_add": true, "enable_flashcomm1": true, "enable_fused_mc2": true, "enable_mlapo": true}' \
--compilation-config '{"cudagraph_mode": "FULL_DECODE_ONLY"}' \
--speculative-config '{"num_speculative_tokens": 3, "method": "deepseek_mtp","enforce_eager":true}'
节点 1
# this obtained through ifconfig
# nic_name is the network interface name corresponding to local_ip of the current node
nic_name="xxx"
local_ip="xxx"
# IP of node 0 (the data parallel master node), must be consistent with the local_ip of node 0
node0_ip="xxxx"
export HCCL_BUFFSIZE=400
export HCCL_IF_IP=$local_ip
export HCCL_OP_EXPANSION_MODE="AIV"
export HCCL_SOCKET_IFNAME=$nic_name
export GLOO_SOCKET_IFNAME=$nic_name
export PYTORCH_NPU_ALLOC_CONF=expandable_segments:True
vllm serve /root/.cache/modelscope/hub/models/vllm-ascend/GLM-5.1-W8A8C8-MTP \
--host 0.0.0.0 \
--port 8077 \
--headless \
--data-parallel-size 8 \
--data-parallel-size-local 4 \
--data-parallel-start-rank 4 \
--data-parallel-address $node0_ip \
--enable-expert-parallel \
--data-parallel-rpc-port 12980 \
--tensor-parallel-size 4 \
--hf-overrides '{"use_index_cache": true, "index_topk_freq": 4}' \
--seed 1024 \
--served-model-name glm-5 \
--tool-call-parser glm47 \
--reasoning-parser glm45 \
--enable-auto-tool-choice \
--trust-remote-code \
--gpu-memory-utilization 0.92 \
--quantization ascend \
--enable-chunked-prefill \
--enable-prefix-caching \
--async-scheduling \
--additional-config '{"enable_dsa_cp": true, "enable_sparse_sfa_c8": true, "enable_sparse_li_c8": true, "enable_balance_scheduling": true, "fuse_muls_add": true, "enable_flashcomm1": true, "enable_fused_mc2": true, "enable_mlapo": true}' \
--compilation-config '{"cudagraph_mode": "FULL_DECODE_ONLY"}' \
--speculative-config '{"num_speculative_tokens": 3, "method": "deepseek_mtp","enforce_eager":true}'
注意:
- 当测试的前缀缓存命中率 > 0 时,保留
--enable-prefix-caching(如上述脚本所示);当命中率为 0 时,将其替换为--no-enable-prefix-caching。 "enable_fused_mc2": true与"multistream_overlap_shared_expert": true冲突——当启用 fused MC2 时,运行时自动禁用multistream_overlap_shared_expert。
分别在两个节点上运行以下脚本。
节点 0
# this obtained through ifconfig
# nic_name is the network interface name corresponding to local_ip of the current node
nic_name="xxx"
local_ip="xxx"
# The value of node0_ip must be consistent with the value of local_ip set in node0 (master node)
node0_ip="xxx"
export HCCL_BUFFSIZE=200
export HCCL_IF_IP=$local_ip
export HCCL_OP_EXPANSION_MODE="AIV"
export HCCL_SOCKET_IFNAME=$nic_name
export GLOO_SOCKET_IFNAME=$nic_name
export PYTORCH_NPU_ALLOC_CONF=expandable_segments:True
vllm serve /root/.cache/modelscope/hub/models/vllm-ascend/GLM5-w4a8 \
--host 0.0.0.0 \
--port 8077 \
--data-parallel-size 2 \
--data-parallel-size-local 1 \
--data-parallel-address $node0_ip \
--data-parallel-rpc-port 13389 \
--tensor-parallel-size 8 \
--quantization ascend \
--seed 1024 \
--served-model-name glm-5 \
--enable-expert-parallel \
--max-num-seqs 2 \
--max-model-len 131072 \
--max-num-batched-tokens 4096 \
--trust-remote-code \
--gpu-memory-utilization 0.95 \
--compilation-config '{"cudagraph_mode": "FULL_DECODE_ONLY"}' \
--additional-config '{"multistream_overlap_shared_expert": true, "enable_balance_scheduling": true, "enable_flashcomm1": true}' \
--hf-overrides '{"use_index_cache": true, "index_topk_freq": 4}' \
--speculative-config '{"num_speculative_tokens": 3, "method": "deepseek_mtp", "enforce_eager": true}'
节点 1
# this obtained through ifconfig
# nic_name is the network interface name corresponding to local_ip of the current node
nic_name="xxx"
local_ip="xxx"
# The value of node0_ip must be consistent with the value of local_ip set in node0 (master node)
node0_ip="xxx"
export HCCL_BUFFSIZE=200
export HCCL_IF_IP=$local_ip
export HCCL_OP_EXPANSION_MODE="AIV"
export HCCL_SOCKET_IFNAME=$nic_name
export GLOO_SOCKET_IFNAME=$nic_name
export PYTORCH_NPU_ALLOC_CONF=expandable_segments:True
vllm serve /root/.cache/modelscope/hub/models/vllm-ascend/GLM5-w4a8 \
--host 0.0.0.0 \
--port 8077 \
--headless \
--data-parallel-size 2 \
--data-parallel-size-local 1 \
--data-parallel-start-rank 1 \
--data-parallel-address $node0_ip \
--data-parallel-rpc-port 13389 \
--tensor-parallel-size 8 \
--quantization ascend \
--seed 1024 \
--served-model-name glm-5 \
--enable-expert-parallel \
--max-num-seqs 2 \
--max-model-len 131072 \
--max-num-batched-tokens 4096 \
--trust-remote-code \
--gpu-memory-utilization 0.95 \
--compilation-config '{"cudagraph_mode": "FULL_DECODE_ONLY"}' \
--additional-config '{"multistream_overlap_shared_expert": true, "enable_balance_scheduling": true, "enable_flashcomm1": true}' \
--hf-overrides '{"use_index_cache": true, "index_topk_freq": 4}' \
--speculative-config '{"num_speculative_tokens": 3, "method": "deepseek_mtp", "enforce_eager": true}'
多节点部署的关键参数说明:
除了单节点在线部署中描述的所有单节点参数外,以下参数专门用于多节点部署:
网络和数据并行配置:
HCCL_IF_IP、GLOO_SOCKET_IFNAME、HCCL_SOCKET_IFNAME:多节点通信的网络接口配置。将nic_name设置为网络接口名称(通过ifconfig获取),将local_ip设置为当前节点的 IP 地址。每个节点上都必须正确配置这些参数,才能实现成功的多节点通信。--data-parallel-size:所有节点上的数据并行 rank 总数。对于 2 节点部署,通常设置为2。--data-parallel-size-local:当前节点上的数据并行 rank 数量。通常设置为1(每个节点一个 DP rank)。--data-parallel-address:数据并行主节点(节点 0)的 IP 地址。必须与主节点的local_ip匹配。--data-parallel-rpc-port:数据并行主节点通信的 RPC 端口。所有节点上必须相同。--headless:表示这是非主节点。不要在节点 0 上使用。--data-parallel-start-rank:此节点上数据并行 rank 的起始 rank 偏移量。节点 0 使用0;节点 1 使用节点 0 上的 DP rank 数量(例如,DP2 为1,DP8 为4)。
多节点性能调优:
- 对于低延迟多节点场景,保持
--data-parallel-size-local 1以最小化跨节点通信。 --max-num-seqs应根据模型加载后的可用 KV 缓存内存进行调整。对于 A3 上的 w8a8c8 198K 高吞吐量场景,建议设置为6;对于 198K 低延迟场景,建议设置为16。对于 A2 上长上下文的 w4a8 多节点场景,从2开始,如果内存允许则增加。- 多节点部署中的所有节点必须使用相同的
--tensor-parallel-size、--enable-expert-parallel和模型权重路径配置。
w8a8c8 特有的 --additional-config 字段:
"enable_dsa_cp": true:启用 DSA 上下文并行以加速长上下文预填充。"enable_sparse_sfa_c8": true/"enable_sparse_li_c8": true:C8 量化模型的稀疏注意力优化。"enable_balance_scheduling": true:提高 v1 调度器中的输出吞吐量并减少 TPOT。当预填充-解码分离时不建议使用。"fuse_muls_add": true:融合乘加运算。"multistream_overlap_shared_expert": true:在额外的流上重叠共享专家计算。当"enable_fused_mc2": true时自动禁用。
5.3 预填充-解码分离¶
我们希望在多节点环境中展示 GLM-5 的部署指南,采用预填充-解码(PD)分离以获得更好的性能。预填充-解码分离 是指将预填充阶段和解码阶段分布到不同节点上,以提高吞吐量和降低延迟。
在 PD 分离场景中,Mooncake 用作预填充节点和解码节点之间的 KV 缓存传输连接器。有关 Mooncake 配置,请参阅 KV 缓存池(Ascend Store)部署指南。
5.3.1 预填充-解码分离(Ascend950DT 系列)¶
开始之前,请
在每个节点上准备脚本 launch_online_dp.py:
import argparse
import multiprocessing
import os
import subprocess
import sys
def parse_args():
parser = argparse.ArgumentParser()
parser.add_argument(
"--dp-size",
type=int,
required=True,
help="Data parallel size."
)
parser.add_argument(
"--tp-size",
type=int,
default=1,
help="Tensor parallel size."
)
parser.add_argument(
"--dp-size-local",
type=int,
default=-1,
help="Local data parallel size."
)
parser.add_argument(
"--dp-rank-start",
type=int,
default=0,
help="Starting rank for data parallel."
)
parser.add_argument(
"--dp-address",
type=str,
required=True,
help="IP address for data parallel master node."
)
parser.add_argument(
"--dp-rpc-port",
type=str,
default=12345,
help="Port for data parallel master node."
)
parser.add_argument(
"--vllm-start-port",
type=int,
default=9000,
help="Starting port for the engine."
)
return parser.parse_args()
args = parse_args()
dp_size = args.dp_size
tp_size = args.tp_size
dp_size_local = args.dp_size_local
if dp_size_local == -1:
dp_size_local = dp_size
dp_rank_start = args.dp_rank_start
dp_address = args.dp_address
dp_rpc_port = args.dp_rpc_port
vllm_start_port = args.vllm_start_port
def run_command(visible_devices, dp_rank, vllm_engine_port):
command = [
"bash",
"./run_dp_template.sh",
visible_devices,
str(vllm_engine_port),
str(dp_size),
str(dp_rank),
dp_address,
dp_rpc_port,
str(tp_size),
]
subprocess.run(command, check=True)
if __name__ == "__main__":
template_path = "./run_dp_template.sh"
if not os.path.exists(template_path):
print(f"Template file {template_path} does not exist.")
sys.exit(1)
processes = []
num_cards = dp_size_local * tp_size
for i in range(dp_size_local):
dp_rank = dp_rank_start + i
vllm_engine_port = vllm_start_port + i
visible_devices = ",".join(str(x) for x in range(i * tp_size, (i + 1) * tp_size))
process = multiprocessing.Process(target=run_command,
args=(visible_devices, dp_rank,
vllm_engine_port))
processes.append(process)
process.start()
for process in processes:
process.join()
-
在每个节点上准备脚本
run_dp_template.sh。-
预填充节点 0
#!/usr/bin/env bash source /root/.bashrc # this obtained through ifconfig # nic_name is the network interface name corresponding to local_ip of the current node nic_name="xxx" local_ip="xxx" export VLLM_EXECUTE_MODEL_TIMEOUT_SECONDS=30000 export HCCL_BUFFSIZE=300 export HCCL_IF_IP=$local_ip export HCCL_SOCKET_IFNAME=$nic_name export ASCEND_RT_VISIBLE_DEVICES=$1 export GLOO_SOCKET_IFNAME=$nic_name export PYTORCH_NPU_ALLOC_CONF=expandable_segments:True export PROMETHEUS_MULTIPROC_DIR=/dev/shm/vllm_metrics && mkdir -p $PROMETHEUS_MULTIPROC_DIR export HCCL_DFS_CONFIG="task_exception:off,inconsistent_check:off" export HCCL_ALGO=level0:fullmesh export ASCEND_LOCAL_COMM_RES='{"version":"1.3"}' vllm serve /root/.cache/modelscope/hub/models/vllm-ascend/GLM5-w4a4 \ --host 0.0.0.0 \ --port $2 \ --data-parallel-size $3 \ --tensor-parallel-size $7 \ --max-model-len 135000 \ --max-num-batched-tokens 8192 \ --served-model-name glm-5 \ --gpu-memory-utilization 0.95 \ --enable-expert-parallel \ --max-num-seqs 8 \ --enable-prefix-caching \ --trust-remote-code \ --enforce-eager \ --quantization ascend \ --enable-auto-tool-choice \ --tool-call-parser glm47 \ --reasoning-parser glm45 \ --speculative-config '{"num_speculative_tokens": 1, "method": "deepseek_mtp", "enforce_eager": true}' \ --hf-overrides '{"use_index_cache": true, "index_topk_freq": 4}' \ --kv-transfer-config \ '{"kv_connector": "MooncakeConnectorV1", "kv_role": "kv_producer", "kv_port": "30100", "engine_id": "1", "kv_connector_extra_config": { "prefill": { "dp_size": 1, "tp_size": 8 }, "decode": { "dp_size": 16, "tp_size": 1 }, "ascend_local_comm_res_path": "/etc/hixlep" } }' \ --additional-config '{"enable_cpu_binding": "True", "multistream_overlap_shared_expert": "True", "recompute_scheduler_enable": "True", "enable_sparse_c8": "True", "enable_dsa_cp": true, "enable_flashcomm1": true}' -
预填充节点 1
#!/usr/bin/env bash source /root/.bashrc # this obtained through ifconfig # nic_name is the network interface name corresponding to local_ip of the current node nic_name="xxx" local_ip="xxx" export VLLM_EXECUTE_MODEL_TIMEOUT_SECONDS=30000 export HCCL_BUFFSIZE=300 export HCCL_IF_IP=$local_ip export HCCL_SOCKET_IFNAME=$nic_name export ASCEND_RT_VISIBLE_DEVICES=$1 export GLOO_SOCKET_IFNAME=$nic_name export PYTORCH_NPU_ALLOC_CONF=expandable_segments:True export PROMETHEUS_MULTIPROC_DIR=/dev/shm/vllm_metrics && mkdir -p $PROMETHEUS_MULTIPROC_DIR export HCCL_DFS_CONFIG="task_exception:off,inconsistent_check:off" export HCCL_ALGO=level0:fullmesh export ASCEND_LOCAL_COMM_RES='{"version":"1.3"}' vllm serve /root/.cache/modelscope/hub/models/vllm-ascend/GLM5-w4a4 \ --host 0.0.0.0 \ --port $2 \ --data-parallel-size $3 \ --tensor-parallel-size $7 \ --max-model-len 135000 \ --max-num-batched-tokens 8192 \ --served-model-name glm-5 \ --gpu-memory-utilization 0.95 \ --enable-expert-parallel \ --max-num-seqs 8 \ --enable-prefix-caching \ --trust-remote-code \ --enforce-eager \ --quantization ascend \ --enable-auto-tool-choice \ --tool-call-parser glm47 \ --reasoning-parser glm45 \ --speculative-config '{"num_speculative_tokens": 1, "method": "deepseek_mtp", "enforce_eager": true}' \ --hf-overrides '{"use_index_cache": true, "index_topk_freq": 4}' \ --kv-transfer-config \ '{"kv_connector": "MooncakeConnectorV1", "kv_role": "kv_producer", "kv_port": "30100", "engine_id": "1", "kv_connector_extra_config": { "prefill": { "dp_size": 1, "tp_size": 8 }, "decode": { "dp_size": 16, "tp_size": 1 }, "ascend_local_comm_res_path": "/etc/hixlep" } }' \ --additional-config '{"enable_cpu_binding": "True", "multistream_overlap_shared_expert": "True", "recompute_scheduler_enable": "True", "enable_sparse_c8": "True", "enable_dsa_cp": true, "enable_flashcomm1": true}' -
解码节点 0
#!/usr/bin/env bash source /root/.bashrc # this obtained through ifconfig # nic_name is the network interface name corresponding to local_ip of the current node nic_name="xxx" local_ip="xxx" export VLLM_EXECUTE_MODEL_TIMEOUT_SECONDS=30000 export HCCL_BUFFSIZE=1200 export HCCL_IF_IP=$local_ip export HCCL_SOCKET_IFNAME=$nic_name export ASCEND_RT_VISIBLE_DEVICES=$1 export GLOO_SOCKET_IFNAME=$nic_name export PYTORCH_NPU_ALLOC_CONF=expandable_segments:True export PROMETHEUS_MULTIPROC_DIR=/dev/shm/vllm_metrics && mkdir -p $PROMETHEUS_MULTIPROC_DIR export HCCL_DFS_CONFIG="task_exception:off,inconsistent_check:off" export HCCL_ALGO=level0:fullmesh export ASCEND_LOCAL_COMM_RES='{"version":"1.3"}' vllm serve /root/.cache/modelscope/hub/models/vllm-ascend/GLM5-w4a4 \ --host 0.0.0.0 \ --port $2 \ --data-parallel-size $3 \ --data-parallel-rank $4 \ --data-parallel-address $5 \ --data-parallel-rpc-port $6 \ --tensor-parallel-size $7 \ --max-model-len 135000 \ --max-num-batched-tokens 240 \ --served-model-name glm-5 \ --gpu-memory-utilization 0.95 \ --enable-expert-parallel \ --max-num-seqs 60 \ --enable-prefix-caching \ --trust-remote-code \ --quantization ascend \ --enable-auto-tool-choice \ --tool-call-parser glm47 \ --reasoning-parser glm45 \ --compilation-config '{"cudagraph_mode":"FULL_DECODE_ONLY"}' \ --speculative-config '{"num_speculative_tokens": 3, "method": "deepseek_mtp", "enforce_eager": true}' \ --hf-overrides '{"use_index_cache": true, "index_topk_freq": 4}' \ --kv-transfer-config \ '{"kv_connector": "MooncakeConnectorV1", "kv_role": "kv_consumer", "kv_port": "30300", "engine_id": "3", "kv_connector_extra_config": { "prefill": { "dp_size": 1, "tp_size": 8 }, "decode": { "dp_size": 16, "tp_size": 1 }, "ascend_local_comm_res_path": "/etc/hixlep" } }' \ --additional-config '{"enable_cpu_binding": "True", "multistream_overlap_shared_expert": "True", "recompute_scheduler_enable": "True", "enable_sparse_c8": "True", "finegrained_tp_config": {"lmhead_tensor_parallel_size":8}}' -
解码节点 1
#!/usr/bin/env bash source /root/.bashrc # this obtained through ifconfig # nic_name is the network interface name corresponding to local_ip of the current node nic_name="xxx" local_ip="xxx" export VLLM_EXECUTE_MODEL_TIMEOUT_SECONDS=30000 export HCCL_BUFFSIZE=1200 export HCCL_IF_IP=$local_ip export HCCL_SOCKET_IFNAME=$nic_name export ASCEND_RT_VISIBLE_DEVICES=$1 export GLOO_SOCKET_IFNAME=$nic_name export PYTORCH_NPU_ALLOC_CONF=expandable_segments:True export PROMETHEUS_MULTIPROC_DIR=/dev/shm/vllm_metrics && mkdir -p $PROMETHEUS_MULTIPROC_DIR export HCCL_DFS_CONFIG="task_exception:off,inconsistent_check:off" export HCCL_ALGO=level0:fullmesh export ASCEND_LOCAL_COMM_RES='{"version":"1.3"}' vllm serve /root/.cache/modelscope/hub/models/vllm-ascend/GLM5-w4a4 \ --host 0.0.0.0 \ --port $2 \ --data-parallel-size $3 \ --data-parallel-rank $4 \ --data-parallel-address $5 \ --data-parallel-rpc-port $6 \ --tensor-parallel-size $7 \ --max-model-len 202752 \ --max-num-batched-tokens 240 \ --served-model-name glm-5 \ --gpu-memory-utilization 0.95 \ --enable-expert-parallel \ --max-num-seqs 60 \ --enable-prefix-caching \ --trust-remote-code \ --quantization ascend \ --enable-auto-tool-choice \ --tool-call-parser glm47 \ --reasoning-parser glm45 \ --compilation-config '{"cudagraph_mode":"FULL_DECODE_ONLY"}' \ --speculative-config '{"num_speculative_tokens": 3, "method": "deepseek_mtp", "enforce_eager": true}' \ --hf-overrides '{"use_index_cache": true, "index_topk_freq": 4}' \ --kv-transfer-config \ '{"kv_connector": "MooncakeConnectorV1", "kv_role": "kv_consumer", "kv_port": "30300", "engine_id": "3", "kv_connector_extra_config": { "prefill": { "dp_size": 1, "tp_size": 8 }, "decode": { "dp_size": 16, "tp_size": 1 }, "ascend_local_comm_res_path": "/etc/hixlep" } }' \ --additional-config '{"enable_cpu_binding": "True", "multistream_overlap_shared_expert": "True", "recompute_scheduler_enable": "True", "enable_sparse_c8": "True", "finegrained_tp_config": {"lmhead_tensor_parallel_size":8}}'
-
准备工作完成后,可以在每个节点上使用以下命令启动服务器:
-
预填充节点 0
-
预填充节点 1
-
解码节点 0
-
解码节点 1
5.3.2 预填充-解码分离(A3 系列)¶
高吞吐量(198K 上下文)场景已在 4 个 Atlas 800 A3(128GB × 8)上验证:2 个预填充节点(PP2 TP16,78 层划分为 41/37,每个节点一个 PP rank)和 2 个解码节点(DP8 TP4,每个节点 4 个 DP rank)。相同的脚本同时适用于高吞吐量和低延迟场景。
开始之前,请
在每个节点上准备脚本 launch_online_dp.py:
import argparse
import multiprocessing
import os
import subprocess
import sys
def parse_args():
parser = argparse.ArgumentParser()
parser.add_argument("--dp-size", type=int, required=True, help="Data parallel size.")
parser.add_argument("--tp-size", type=int, default=1, help="Tensor parallel size.")
parser.add_argument("--pp-size", type=int, default=1, help="Pipeline parallel size.")
parser.add_argument("--dp-size-local", type=int, default=-1, help="Local data parallel size.")
parser.add_argument("--dp-rank-start", type=int, default=0, help="Starting rank for data parallel.")
parser.add_argument("--dp-address", type=str, required=True, help="IP address for data parallel master node.")
parser.add_argument("--dp-rpc-port", type=str, default="12321", help="Port for data parallel master node.")
parser.add_argument("--vllm-start-port", type=int, default=8000, help="Starting port for the engine.")
return parser.parse_args()
args = parse_args()
dp_size = args.dp_size
tp_size = args.tp_size
pp_size = args.pp_size
dp_size_local = args.dp_size_local
if dp_size_local == -1:
dp_size_local = dp_size
dp_rank_start = args.dp_rank_start
dp_address = args.dp_address
dp_rpc_port = args.dp_rpc_port
vllm_start_port = args.vllm_start_port
gpus_per_dp_rank = tp_size * pp_size
def run_command(visible_devices, dp_rank, vllm_engine_port):
command = [
"bash",
"./run_dp_template.sh",
visible_devices,
str(vllm_engine_port),
str(dp_size),
str(dp_rank),
dp_address,
dp_rpc_port,
str(tp_size),
str(pp_size),
]
subprocess.run(command, check=True)
if __name__ == "__main__":
template_path = "./run_dp_template.sh"
if not os.path.exists(template_path):
print(f"Template file {template_path} does not exist.")
sys.exit(1)
processes = []
num_cards = dp_size_local * gpus_per_dp_rank
for i in range(dp_size_local):
dp_rank = dp_rank_start + i
vllm_engine_port = vllm_start_port + i
visible_devices = ",".join(str(x) for x in range(i * gpus_per_dp_rank, (i + 1) * gpus_per_dp_rank))
process = multiprocessing.Process(
target=run_command,
args=(visible_devices, dp_rank, vllm_engine_port)
)
processes.append(process)
process.start()
for process in processes:
process.join()
-
在每个节点上准备脚本
run_dp_template.sh。-
预填充节点 0
预填充脚本通过
node_rank选择节点:在预填充节点 0(PP 主节点,引擎端口9081)上设置node_rank=0,在预填充节点 1(非主节点,--headless,无 API 服务器)上设置node_rank=1。nic_name="xxxx" # change to your own nic name local_ip="xxxx" # change to your own ip # pp=2 # prefill node 0: node_rank=0, prefill node 1: node_rank=1 node_rank=0 export VLLM_PP_LAYER_PARTITION="41,37" export HCCL_BUFFSIZE=400 export HCCL_IF_IP=$local_ip export HCCL_OP_EXPANSION_MODE="AIV" export HCCL_SOCKET_IFNAME=$nic_name export LD_LIBRARY_PATH=$LD_LIBRARY_PATH:/usr/local/lib export GLOO_SOCKET_IFNAME=$nic_name export PYTORCH_NPU_ALLOC_CONF=expandable_segments:True vllm serve /root/.cache/modelscope/hub/models/vllm-ascend/GLM-5.1-W8A8C8-MTP \ --host 0.0.0.0 \ --port 9081 \ --pipeline-parallel-size 2 \ --distributed-executor-backend mp \ --master-addr $local_ip \ --master-port 7060 \ --nnodes 2 \ --node-rank $node_rank \ --tensor-parallel-size 16 \ --enable-expert-parallel \ --speculative-config '{"num_speculative_tokens": 1, "method":"deepseek_mtp","enforce_eager":true}' \ --seed 1024 \ --served-model-name glm-5 \ --max-model-len 202752 \ --additional-config '{"fuse_muls_add": true, "recompute_scheduler_enable": false, "multistream_overlap_shared_expert": true, "enable_dsa_cp": true, "enable_sparse_sfa_c8": true, "enable_sparse_li_c8": true, "c8_enable_reshape_optim": true, "enable_flashcomm1": true, "enable_fused_mc2": true}' \ --max-num-batched-tokens 16384 \ --trust-remote-code \ --enable-prefix-caching \ --max-num-seqs 64 \ --quantization ascend \ --gpu-memory-utilization 0.92 \ --hf-overrides '{"use_index_cache": true, "index_topk_freq": 4}' \ --enforce-eager \ --enable-auto-tool-choice \ --tool-call-parser glm47 \ --reasoning-parser glm45 \ --kv-transfer-config \ '{"kv_connector": "MooncakeConnectorV1", "kv_role": "kv_producer", "kv_port": "30000", "engine_id": "0", "kv_connector_extra_config": { "use_ascend_direct": true, "prefill": {"dp_size": 1, "pp_size": 2, "tp_size": 16, "pp_layer_partition": "41,37"}, "decode": {"dp_size": 8, "tp_size": 4} } }' -
预填充节点 1
nic_name="xxxx" # change to your own nic name local_ip="xxxx" # change to your own ip # IP of prefill node 0 (the PP master node), must be consistent with the local_ip of prefill node 0 node_p0_ip="xxxx" # pp=2 # prefill node 0: node_rank=0, prefill node 1: node_rank=1 node_rank=1 export VLLM_PP_LAYER_PARTITION="41,37" export HCCL_BUFFSIZE=400 export HCCL_IF_IP=$local_ip export HCCL_OP_EXPANSION_MODE="AIV" export HCCL_SOCKET_IFNAME=$nic_name export LD_LIBRARY_PATH=$LD_LIBRARY_PATH:/usr/local/lib export GLOO_SOCKET_IFNAME=$nic_name export PYTORCH_NPU_ALLOC_CONF=expandable_segments:True vllm serve /root/.cache/modelscope/hub/models/vllm-ascend/GLM-5.1-W8A8C8-MTP \ --host 0.0.0.0 \ --pipeline-parallel-size 2 \ --distributed-executor-backend mp \ --master-addr $node_p0_ip \ --master-port 7060 \ --nnodes 2 \ --node-rank $node_rank \ --headless \ --tensor-parallel-size 16 \ --enable-expert-parallel \ --speculative-config '{"num_speculative_tokens": 1, "method":"deepseek_mtp","enforce_eager":true}' \ --seed 1024 \ --served-model-name glm-5 \ --max-model-len 202752 \ --additional-config '{"fuse_muls_add": true, "recompute_scheduler_enable": false, "multistream_overlap_shared_expert": true, "enable_dsa_cp": true, "enable_sparse_sfa_c8": true, "enable_sparse_li_c8": true, "c8_enable_reshape_optim": true, "enable_flashcomm1": true, "enable_fused_mc2": true}' \ --max-num-batched-tokens 16384 \ --trust-remote-code \ --enable-prefix-caching \ --max-num-seqs 64 \ --quantization ascend \ --gpu-memory-utilization 0.92 \ --hf-overrides '{"use_index_cache": true, "index_topk_freq": 4}' \ --enforce-eager \ --enable-auto-tool-choice \ --tool-call-parser glm47 \ --reasoning-parser glm45 \ --kv-transfer-config \ '{"kv_connector": "MooncakeConnectorV1", "kv_role": "kv_producer", "kv_port": "30000", "engine_id": "0", "kv_connector_extra_config": { "use_ascend_direct": true, "prefill": {"dp_size": 1, "pp_size": 2, "tp_size": 16, "pp_layer_partition": "41,37"}, "decode": {"dp_size": 8, "tp_size": 4} } }' -
解码节点 0(rank 0–3)
通过位置参数为每个 DP rank 启动一个实例:
$1= 可见设备,$2= 引擎端口,$3= 数据并行大小,$4= 数据并行 rank,$5= 数据并行地址,$6= 数据并行 RPC 端口,$7= 张量并行大小。在解码节点 0 上准备run_dp_template.sh,内容如下。nic_name="xxxx" # change to your own nic name local_ip="xxxx" # change to your own ip # Each DP rank uses an independent engine ID to avoid KV route confusion. # $4 = data-parallel-rank. The rank offset within a node is 0 to 3, and the corresponding engine ID is 100 to 103. ENGINE_ID=$((100 + $4)) #Mooncake export HCCL_BUFFSIZE=256 export HCCL_IF_IP=$local_ip export HCCL_OP_EXPANSION_MODE="AIV" export HCCL_SOCKET_IFNAME=$nic_name export ASCEND_RT_VISIBLE_DEVICES=$1 export LD_LIBRARY_PATH=$LD_LIBRARY_PATH:/usr/local/lib export GLOO_SOCKET_IFNAME=$nic_name export PYTORCH_NPU_ALLOC_CONF=expandable_segments:True vllm serve /root/.cache/modelscope/hub/models/vllm-ascend/GLM-5.1-W8A8C8-MTP \ --host 0.0.0.0 \ --port $2 \ --data-parallel-size $3 \ --data-parallel-rank $4 \ --data-parallel-address $5 \ --data-parallel-rpc-port $6 \ --tensor-parallel-size $7 \ --enable-expert-parallel \ --speculative-config '{"num_speculative_tokens": 3, "method":"deepseek_mtp","enforce_eager":true}' \ --seed 1024 \ --served-model-name glm-5 \ --max-model-len 202752 \ --max-num-batched-tokens 164 \ --compilation-config '{"cudagraph_mode":"FULL_DECODE_ONLY"}' \ --additional-config '{"fuse_muls_add": true, "recompute_scheduler_enable": true, "multistream_overlap_shared_expert": true, "enable_sparse_sfa_c8": true, "enable_sparse_li_c8": true, "enable_fused_mc2": true, "enable_mlapo": true}' \ --trust-remote-code \ --max-num-seqs 32 \ --gpu-memory-utilization 0.92 \ --hf-overrides '{"use_index_cache": true, "index_topk_freq": 4}' \ --async-scheduling \ --quantization ascend \ --enable-auto-tool-choice \ --tool-call-parser glm47 \ --reasoning-parser glm45 \ --kv-transfer-config \ '{"kv_connector": "MooncakeConnectorV1", "kv_role": "kv_consumer", "kv_port": "30200", "engine_id": "'"$ENGINE_ID"'", "kv_connector_extra_config": { "use_ascend_direct": true, "prefill": {"dp_size": 1, "pp_size": 2, "tp_size": 16, "pp_layer_partition": "41,37"}, "decode": {"dp_size": 8, "tp_size": 4} } }' -
解码节点 1(rank 4–7)
在解码节点 1 上准备
run_dp_template.sh,内容如下。nic_name="xxxx" # change to your own nic name local_ip="xxxx" # change to your own ip # Each DP rank uses an independent engine ID to avoid KV route confusion. # $4 = data-parallel-rank. The rank offset within a node is 0 to 3, and the corresponding engine ID is 100 to 103. ENGINE_ID=$((100 + $4)) #Mooncake export HCCL_BUFFSIZE=256 export HCCL_IF_IP=$local_ip export HCCL_OP_EXPANSION_MODE="AIV" export HCCL_SOCKET_IFNAME=$nic_name export ASCEND_RT_VISIBLE_DEVICES=$1 export LD_LIBRARY_PATH=$LD_LIBRARY_PATH:/usr/local/lib export GLOO_SOCKET_IFNAME=$nic_name export PYTORCH_NPU_ALLOC_CONF=expandable_segments:True vllm serve /root/.cache/modelscope/hub/models/vllm-ascend/GLM-5.1-W8A8C8-MTP \ --host 0.0.0.0 \ --port $2 \ --data-parallel-size $3 \ --data-parallel-rank $4 \ --data-parallel-address $5 \ --data-parallel-rpc-port $6 \ --tensor-parallel-size $7 \ --enable-expert-parallel \ --speculative-config '{"num_speculative_tokens": 3, "method":"deepseek_mtp","enforce_eager":true}' \ --seed 1024 \ --served-model-name glm-5 \ --max-model-len 202752 \ --max-num-batched-tokens 164 \ --compilation-config '{"cudagraph_mode":"FULL_DECODE_ONLY"}' \ --additional-config '{"fuse_muls_add": true, "recompute_scheduler_enable": true, "multistream_overlap_shared_expert": true, "enable_sparse_sfa_c8": true, "enable_sparse_li_c8": true, "enable_fused_mc2": true, "enable_mlapo": true}' \ --trust-remote-code \ --max-num-seqs 32 \ --gpu-memory-utilization 0.92 \ --hf-overrides '{"use_index_cache": true, "index_topk_freq": 4}' \ --async-scheduling \ --quantization ascend \ --enable-auto-tool-choice \ --tool-call-parser glm47 \ --reasoning-parser glm45 \ --kv-transfer-config \ '{"kv_connector": "MooncakeConnectorV1", "kv_role": "kv_consumer", "kv_port": "30200", "engine_id": "'"$ENGINE_ID"'", "kv_connector_extra_config": { "use_ascend_direct": true, "prefill": {"dp_size": 1, "pp_size": 2, "tp_size": 16, "pp_layer_partition": "41,37"}, "decode": {"dp_size": 8, "tp_size": 4} } }'
-
准备工作完成后,可以在每个节点上使用以下命令启动服务器:
-
预填充节点 0
-
预填充节点 1
-
解码节点 0
-
解码节点 1
注意:
- 当测试的前缀缓存命中率 > 0 时,在预填充节点上添加
--enable-prefix-caching(如上述脚本所示);当命中率为 0 时,改用--no-enable-prefix-caching。 - 在此场景中,预填充节点上的
"recompute_scheduler_enable"设置为false,解码节点上设置为true。
5.3.3 请求转发和关键参数说明¶
要设置请求转发,请在任意机器上运行以下脚本。您可以在仓库的示例中获取代理程序:load_balance_proxy_server_example.py
Ascend950DT 系列:
unset http_proxy
unset https_proxy
python load_balance_proxy_server_example.py \
--port 8000 \
--host 0.0.0.0 \
--prefiller-hosts \
$node_p0_ip \
$node_p1_ip \
--prefiller-ports \
6700 \
6700 \
--decoder-hosts \
$node_d0_ip \
$node_d0_ip \
$node_d0_ip \
$node_d0_ip \
$node_d0_ip \
$node_d0_ip \
$node_d0_ip \
$node_d0_ip \
$node_d1_ip \
$node_d1_ip \
$node_d1_ip \
$node_d1_ip \
$node_d1_ip \
$node_d1_ip \
$node_d1_ip \
$node_d1_ip \
--decoder-ports \
6721 6722 6723 6724 6725 6726 6727 6728 \
6721 6722 6723 6724 6725 6726 6727 6728
A3 系列:
unset http_proxy
unset https_proxy
python load_balance_proxy_server_example.py \
--port 9000 \
--host 0.0.0.0 \
--prefiller-hosts \
$node_p0_ip \
--prefiller-ports \
9081 \
--decoder-hosts \
$node_d0_ip \
$node_d0_ip \
$node_d0_ip \
$node_d0_ip \
$node_d1_ip \
$node_d1_ip \
$node_d1_ip \
$node_d1_ip \
--decoder-ports \
8000 8001 8002 8003 \
8000 8001 8002 8003
PD 分离部署的关键参数说明:
除了上述单节点和多节点参数外,以下参数专门用于 Prefill-Decode 分离:
Mooncake KV 传输配置(--kv-transfer-config):
"kv_connector": "MooncakeConnectorV1":使用 Mooncake 作为 prefill 和 decode 节点之间的 KV 缓存传输连接器。"kv_role": "kv_producer":在 prefill 节点上设置 — 生成 KV 缓存并将其发送到 decode 节点。在 decode 节点上使用"kv_consumer"。"kv_port":Mooncake KV 传输通信的端口。每个节点组应使用不同的端口范围。"use_ascend_direct": true:启用 Ascend 直接(类似 RDMA)传输 KV 缓存,降低延迟。"prefill"/"decode"部分:分别指定 prefill 和 decode 节点组的dp_size和tp_size。这些必须与实际部署拓扑匹配。
Prefill 节点特定配置:
--additional-config '{"enable_fused_mc2": true}':启用 fused MC2 算子(dispatch_ffn_combine/mega_moe)以优化 MoE 通信。约束条件:dispatch_ffn_combine仅适用于 w8a8 且 EP≤32;mega_moe适用于 w8a8/w4a8/bf16 且 EP≤64。两者均与 MTP 和动态 EPLB 不兼容。--additional-config '{"enable_dsa_cp": true}':在预填充节点上启用 DSA 上下文并行以加速长上下文预填充。处理高达 128K token 的提示词时需要此配置。
Decode 节点特定配置:
--additional-config '{"enable_mlapo": true}':在解码节点上启用MLA预处理操作融合,以显著提升解码性能。会消耗更多NPU内存。在PD场景中,仅在解码节点上启用MLAPO。- 在解码节点上,保持
--max-num-batched-tokens接近--max-num-seqs— 解码每步每序列处理一个token(A3场景中为164,Ascend950DT场景中为240,参见上述脚本)。 --additional-config '{"recompute_scheduler_enable": true}':启用重计算调度器。当解码节点KV缓存不足时,请求会被送回预填充节点进行KV缓存重计算。在此部署中:解码节点上为true;预填充节点上,Ascend950DT场景为true,A3场景为false(参见上述脚本)。
常见 PD 环境变量:
LD_LIBRARY_PATH=$LD_LIBRARY_PATH:/usr/local/lib:Mooncake库加载所需。
PD 场景中的 MTP:
- 预填充节点在两种场景下均使用
"num_speculative_tokens": 1(参见上述脚本)。 - 解码节点在两种场景下均使用
"num_speculative_tokens": 3以最大化解码吞吐量。 - 所有预填充和解码节点必须使用相同的
"method": "deepseek_mtp"和"enforce_eager": true。
有关上述环境变量的进一步说明和限制,请参阅:envs.py。
6 功能验证¶
服务器启动后,您可以使用输入提示来查询模型:
curl http://<node0_ip>:<port>/v1/completions \
-H "Content-Type: application/json" \
-d '{
"model": "glm-5",
"prompt": "The future of AI is",
"max_completion_tokens": 15,
"temperature": 0
}'
预期结果:
{"id": "chatcmlib-bc44ad093dec79a2", "object": "chat.completion", "created": "1770903266", "model": "glm-5", "choices": [{ "index": 0, "message": {"role": "assistant", "content": "The future of AI is not one thing, but a convergence of several powerful trends.", "annotations": "null", "audio": "null", "function_call": "null", "tool_calls": [], "reasoning": "null"}, "logprobs": "null", "finish_reason": "length", "stop_reason": "null", "token_ids": null}], "service_tier": "null", "system fingerprint": "null", "usage": {"prompt_tokens": 5, "total_tokens": 20, "completion_tokens": 15, "prompt_tokens_details": null}, "prompt_logprobs": "null", "prompt_token_ids": "null", "kv_transfer_params": null}
7 精度评估¶
7.1 使用 AISBench¶
-
有关详细信息,请参阅 使用 AISBench。
-
执行后,您可以获取结果。
8 性能评估¶
8.1 使用 AISBench¶
有关详细信息,请参阅 使用 AISBench 进行性能评估。
8.2 使用 vLLM Benchmark¶
更多详情请参考vllm基准测试。
9 性能调优¶
9.1 推荐配置¶
注意:以下配置在特定测试环境中验证,仅供参考。最佳配置取决于最大输入/输出长度、前缀缓存命中率、精度要求和部署机器比例等因素。建议参考调优指南根据实际情况进行调优。
下表提供了Atlas 800 A3上 GLM-5.1-w8a8c8 量化模型的推荐参数配置,涵盖三种部署场景:
- 双节点共置198K高吞吐:
DP8 TP4,参见多节点部署。 - 双节点共置198K低延迟:
DP2 TP16,参见多节点部署。 - 预填充-解码分离198K(PP2):预填充
PP2 TP16+ 解码DP8 TP4,参见预填充-解码分离(A3系列)。 - 预填充-解码分离198K(Ascend950DT):预填充
TP8(DSA CP 8) + 解码DP16 TP1,参见预填充-解码分离(Ascend950DT系列)。
测试用例使用 输入/输出 表示法,例如 128k/1k 表示128K输入token和1K输出token;@50%/90% 标记前缀缓存命中率。当前缀缓存命中率 > 0 时,添加 --enable-prefix-caching;当命中率为0时,改为添加 --no-enable-prefix-caching(对于PD场景,这适用于预填充节点)。
9.1.1 表1:详细节点配置¶
TP/DP列显示部署脚本中配置的每节点值(承载4个TP4 DP rank的共置节点使用16个NPU;承载1个TP16 DP rank的PD预填充节点使用16个NPU;承载4个TP4 DP rank的PD解码节点使用16个NPU;Ascend950DT节点承载8张卡)。198K PD场景预填充侧使用
PP2 TP16,层分区为41,37。所有A3场景使用GLM-5.1-w8a8c8权重;Ascend950DT场景使用GLM-5.1-w4a4权重。当前缀缓存命中率 > 0 进行测试时,保留
--enable-prefix-caching(如部署脚本中所示);当命中率为0时,将其替换为--no-enable-prefix-caching。
| Scenario | Weight Version | Configuration | NPUs | TP | DP | Max Num Seqs | Max Num Batched Tokens | Max Model Len | MTP Spec Num |
|---|---|---|---|---|---|---|---|---|---|
| Dual-Node Co-Located 198K High Throughput (A3) | w8a8c8 | Dual-Node Co-Located Node (0/1) | 16 | 4 | 4 | 6 | 4096 | 202752 | 3 |
| Dual-Node Co-Located 198K Low Latency (A3) | w8a8c8 | Dual-Node Co-Located Node (0/1) | 16 | 16 | 1 | 16 | 4096 | 202752 | 3 |
| PD 198K High Throughput (A3) | w8a8c8 | PD — Server-P Node (PP2) | 16 | 16 | 1 | 64 | 16384 | 202752 | 1 |
| PD 198K High Throughput (A3) | w8a8c8 | PD — Server-D Node | 16 | 4 | 4 | 32 | 164 | 202752 | 3 |
| PD 198K High Throughput (Ascend950DT) | w4a4 | PD — Server-P Node (DSA CP 8) | 8 | 8 | 1 | 20 | 8192 | 202752 | 1 |
| PD 198K High Throughput (Ascend950DT) | w4a4 | PD — Server-D Node | 8 | 1 | 8 | 60 | 240 | 202752 | 3 |
| PD 198K Low Latency (Ascend950DT) | w4a4 | PD — Server-P Node (DSA CP 8) | 8 | 8 | 1 | 20 | 8192 | 202752 | 1 |
| PD 198K Low Latency (Ascend950DT) | w4a4 | PD — Server-D Node | 8 | 1 | 8 | 60 | 240 | 202752 | 3 |
9.1.2 表2:需要显式启用的优化¶
以下优化必须显式启用才能生效。它们适用于A3系列(w8a8c8),具体如下:
| Optimization | Scenario | Enablement | Principle (Benefits) | Notes |
|---|---|---|---|---|
| FlashComm_v1 | A3预填充节点/共置节点 | --additional-config '{"enable_flashcomm1": true}' |
将AllReduce拆分为Reduce-Scatter和All-Gather,提升预填充吞吐量并降低通信延迟 | 当 layer_sharding 包含 o_proj 时不可用 |
| Fused MC2 | A3预填充节点 | --additional-config '{"enable_fused_mc2": true}' |
用 dispatch_ffn_combine/dispatch_gmm_combine_decode 算子替换ALLTOALL+MC2,降低MoE通信开销并提升MoE推理性能 |
dispatch_ffn_combine 仅适用于w8a8、EP≤32、非MTP、非动态EPLB;与 multistream_overlap_shared_expert 冲突(后者会自动禁用) |
| MLAPO | A3共置高吞吐/PD解码节点 | --additional-config '{"enable_mlapo": true}' |
融合MLA预处理操作,显著提升解码性能 | 消耗更多NPU内存;在PD场景中仅在解码节点上启用 |
| DSA CP | A3预填充节点;长上下文(≥128K) | --additional-config '{"enable_dsa_cp": true}' |
DSA上下文并行加速长上下文预填充,降低长提示的TTFT | 在参考配置中,共置节点和PD预填充节点上启用 |
| Balance Scheduling | A3单节点/共置/非PD场景 | --additional-config '{"enable_balance_scheduling": true}' |
提升v1调度器中的输出吞吐量并降低TPOT | TTFT可能劣化;预填充-解码分离时不推荐 |
| Sparse SFA C8 | A3(w8a8c8);长上下文预填充 | --additional-config '{"enable_sparse_sfa_c8": true}' |
稀疏Flash Attention跳过C8量化模型不必要的注意力计算,加速长上下文预填充 | v0.23.0中为实验性。在参考配置中,高吞吐和PD场景下启用;低延迟场景下禁用 |
| Sparse LI C8 | A3(w8a8c8) | --additional-config '{"enable_sparse_li_c8": true}' |
稀疏注意力优化减少C8量化模型的计算量,提升吞吐量 | 独立于 enable_sparse_sfa_c8;参考低延迟配置同时禁用两者 |
| 重计算调度器 | A3解码节点 | --additional-config '{"recompute_scheduler_enable": true}' |
当解码KV缓存不足时,在预填充节点上重计算KV缓存,避免解码端内存溢出并提高吞吐量 | 在预填充节点上设置为false |
| 多流重叠共享专家 | A3 | --additional-config '{"multistream_overlap_shared_expert": true}' |
在额外的流上重叠共享专家计算,隐藏其延迟并提高解码性能 | 当"enable_fused_mc2": true时自动禁用 |
有关完整的启动命令和详细的参数说明,请参阅在线服务部署中的部署示例和关键参数说明。
9.1.3 表3:性能相关参数调优指南¶
| 参数 | 低延迟 | 高吞吐 | 长上下文 | 描述 |
|---|---|---|---|---|
--max-num-seqs |
较低(16) | 较高(6–64) | 较高(32–64) | 限制并发序列数。较低的值减少调度延迟;较高的值提高吞吐量。 |
--max-model-len |
198K | 更长(128K–198K) | 最大(198K) | 最大上下文长度。必须容纳最长的输入+输出。较大的值消耗更多KV缓存内存。 |
--max-num-batched-tokens |
较低(4096) | 预填充较高(4096–16384) | 预填充较高(16384);解码节点较小(接近max-num-seqs) |
控制每步的批处理大小。较低的值减少每步延迟;较高的值提高预填充吞吐量。 |
--gpu-memory-utilization |
0.92 | 0.92 | 0.92 | NPU内存比例。本文档中的参考配置使用0.92。如果发生内存溢出,请降低该值。 |
--enable-chunked-prefill |
启用(共置) | 启用(共置) | 启用(共置) | 将长提示拆分为多个块,以防止预填充阻塞解码。PD预填充节点改用--enforce-eager。 |
num_speculative_tokens(MTP) |
3 | 3 | 3 | MTP推测数量。较高的值提高解码吞吐量,但会消耗草稿模型KV缓存的内存。在参考PD配置中,预填充节点使用1;解码节点使用3。 |
cudagraph_mode |
FULL_DECODE_ONLY | FULL_DECODE_ONLY(共置/解码节点) | FULL_DECODE_ONLY(共置/解码节点) | 仅对解码阶段进行图捕获。PD预填充节点改用--enforce-eager。 |
9.2 调优指南¶
有关常规性能调优方法,请参阅公开性能调优文档。
有关详细的功能描述和配置选项,请参阅功能指南。
有关环境变量的说明和约束,请参阅envs.py。
10 常见问题解答¶
-
常见问题提示:如果遇到问题,请参阅常见问题。
-
问:如何解决 ValueError: Tokenizer class TokenizersBackend does not exist or is not currently imported?
答:请将 transformers 的版本更新到 5.2.0
- 问:如何为 GLM-5 启用函数调用?
答:请在 vLLM 启动命令中添加以下配置