Kimi-K2.5¶
1 简介¶
Kimi K2.5 是一个开源的、原生多模态智能体模型,基于 Kimi-K2-Base 通过约 15 万亿混合视觉和文本 token 的持续预训练构建而成。它无缝集成了视觉和语言理解能力、先进的智能体能力、即时(instant)与思考(thinking)模式,以及对话式和智能体式范式。
本文档将介绍该模型的主要验证步骤,包括支持特性、特性配置、环境准备、单节点与多节点部署、精度和性能评估。
本文档基于 vLLM-Ascend v0.17.0rc1 编写并完成验证。当前模型(Kimi-K2.5)从该版本开始首次支持,v0.17.0rc1 及后续版本均可稳定运行。
2 支持特性¶
请参考 支持特性列表 获取该模型的支持特性矩阵。
请参考 特性指南 获取特性的配置方式。
3 前提条件¶
3.1 模型权重¶
| 权重版本 | 硬件要求 | 下载链接 |
|---|---|---|
Kimi-K2.5-w4a8(w4a8 量化版本) |
1 个 Atlas 800 A3(64GB × 16)节点或 2 个 Atlas 800 A2(64GB × 8)节点 | ModelScope |
kimi-k2.5-eagle3(Eagle3 MTP 草稿模型,用于加速 Kimi-K2.5 推理) |
Hugging Face |
建议将模型权重下载到多节点共享目录,例如 /root/.cache/。
路径说明:将模型权重下载到你选择的目录并记录下来。确保后续部署命令中的模型路径与该目录一致。
3.2 验证多节点通信(可选)¶
如果你希望部署多节点环境,请按照 验证多节点通信环境 验证多节点通信。
4 安装¶
4.1 Docker 镜像安装¶
请根据你的机器类型选择镜像,并在节点上启动 Docker 镜像,具体请参考 使用 Docker。
在每个节点上启动 Docker 镜像。
export IMAGE=quay.io/ascend/vllm-ascend:v0.23.0-a3
docker run --rm \
--name vllm-ascend \
--shm-size=1g \
--net=host \
--privileged=true \
--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 \
--privileged=true \
--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
Docker 运行成功后,可以通过执行 docker ps 命令验证容器服务是否正常运行。
4.2 源码安装¶
如果你不想使用上述 Docker 镜像,也可以选择从源码进行构建:
- 从源码安装
vllm-ascend,请参考 安装。
如果你希望部署多节点环境,需要在每个节点上完成环境配置。
5 在线服务部署¶
5.1 单节点在线部署¶
单节点部署会在同一节点内完成 Prefill 和 Decode。量化模型 Kimi-K2.5-w4a8 可部署在 1 个 Atlas 800 A3(64GB × 16)节点上。
运行以下脚本执行在线推理。
启动命令:
#!/bin/sh
# [Optional] jemalloc
# jemalloc is for better performance, if `libjemalloc.so` is installed on your machine, you can turn it on.
export LD_PRELOAD=/usr/lib/aarch64-linux-gnu/libjemalloc.so.2:$LD_PRELOAD
echo performance | tee /sys/devices/system/cpu/cpu*/cpufreq/scaling_governor
sysctl -w vm.swappiness=0
sysctl -w kernel.numa_balancing=0
sysctl -w kernel.sched_migration_cost_ns=50000
export HCCL_OP_EXPANSION_MODE="AIV"
export PYTORCH_NPU_ALLOC_CONF=expandable_segments:True
export OMP_PROC_BIND=false
export OMP_NUM_THREADS=1
export TASK_QUEUE_ENABLE=1
export HCCL_BUFFSIZE=800
# Ensure the model path matches the directory recorded during download
vllm serve Eco-Tech/Kimi-K2.5-w4a8 --additional-config '{"scheduler_config":{"enable_balance_scheduling":true},"enable_mlapo":true}' \
--host 0.0.0.0 \
--port 8088 \
--quantization ascend \
--served-model-name kimi_k25 \
--allowed-local-media-path / \
--trust-remote-code \
--no-enable-prefix-caching \
--seed 1024 \
--tensor-parallel-size 4 \
--data-parallel-size 4 \
--enable-expert-parallel \
--max-num-seqs 64 \
--max-model-len 32768 \
--max-num-batched-tokens 16384 \
--gpu-memory-utilization 0.9 \
--compilation-config '{"cudagraph_mode":"FULL_DECODE_ONLY"}' \
--speculative-config '{"method":"eagle3", "model":"lightseekorg/kimi-k2.5-eagle3", "num_speculative_tokens":3}' \
--mm-encoder-tp-mode data \
--enable-auto-tool-choice \
--tool-call-parser kimi_k2 \
--reasoning-parser kimi_k2
关键参数说明:
- 设置
additional_config.scheduler_config.enable_balance_scheduling=true可启用均衡调度。这有助于提高 v1 调度器中的输出吞吐量并降低 TPOT。但在某些场景下,TTFT 可能会下降。此外,在 PD 分离的场景中不建议启用此功能。 - 对于单节点部署,建议使用
dp4 tp4而不是dp2 tp8。 --max-model-len指定最大上下文长度,即单个请求的输入和输出 token 总数。对于输入长度为 3.5k、输出长度为 1.5k 的性能测试,16384的值已经足够;但对于精度测试,请至少设置为35000。--no-enable-prefix-caching表示禁用前缀缓存。若要启用,请移除该选项。--mm-encoder-tp-mode表示如何使用张量并行(TP)优化多模态编码器推理。如果要测试多模态输入,建议使用data。- 如果使用 w4a8 权重,kvcache 会分配更多内存,你可以尝试提高系统吞吐量以获得更高的吞吐。
常见问题提示:如果遇到问题,请参见公共常见问题进行排查。
服务验证:
curl http://<node_ip>:8088/v1/chat/completions \
-H "Content-Type: application/json" \
-d '{
"model": "kimi_k25",
"messages": [{
"role": "user",
"content": [
{
"type": "text",
"text": "The future of AI is"
}]
}],
"max_tokens": 1024,
"temperature": 1.0,
"top_p": 0.95
}'
预期结果:
服务返回 HTTP 200 OK,并返回包含 choices 字段的 JSON 响应。示例输出:
{
"id": "chatcmpl-xxxxxxxxxxxxx",
"object": "chat.completion",
"model": "kimi_k25",
"choices": [
{
"index": 0,
"message": {
"role": "assistant",
"content": "The future of AI is not a destination we are passively approaching...",
"finish_reason": "length"
}
}
],
"usage": {
"prompt_tokens": 13,
"total_tokens": 1037,
"completion_tokens": 1024
}
}
5.2 多节点数据并行部署¶
Kimi-K2.5-w4a8 可以使用数据并行跨多个节点部署。此部署模式至少需要 2 个 Atlas 800 A2(64GB × 8)节点。
分别在两个节点上运行以下脚本。
启动命令:
#!/bin/sh
# this obtained through ifconfig
# nic_name is the network interface name corresponding to local_ip of the current node
nic_name="xxxx"
local_ip="192.xx.xx.1"
# The value of node0_ip must be consistent with the value of local_ip set in node0 (master node)
node0_ip="xxxx"
export HCCL_IF_IP=$local_ip
export GLOO_SOCKET_IFNAME=$nic_name
export TP_SOCKET_IFNAME=$nic_name
export HCCL_SOCKET_IFNAME=$nic_name
export HCCL_INTRA_PCIE_ENABLE=1
export HCCL_INTRA_ROCE_ENABLE=0
# [Optional] jemalloc
# jemalloc is for better performance, if `libjemalloc.so` is installed on your machine, you can turn it on.
export LD_PRELOAD=/usr/lib/aarch64-linux-gnu/libjemalloc.so.2:$LD_PRELOAD
echo performance | tee /sys/devices/system/cpu/cpu*/cpufreq/scaling_governor
sysctl -w vm.swappiness=0
sysctl -w kernel.numa_balancing=0
sysctl -w kernel.sched_migration_cost_ns=50000
export HCCL_OP_EXPANSION_MODE="AIV"
export PYTORCH_NPU_ALLOC_CONF=expandable_segments:True
export OMP_PROC_BIND=false
export OMP_NUM_THREADS=1
export TASK_QUEUE_ENABLE=1
export HCCL_BUFFSIZE=1024
# Ensure the model path matches the directory recorded during download
vllm serve Eco-Tech/Kimi-K2.5-w4a8 --additional-config '{"scheduler_config":{"enable_balance_scheduling":true},"enable_mlapo":true}' \
--host 0.0.0.0 \
--port 8088 \
--quantization ascend \
--served-model-name kimi_k25 \
--allowed-local-media-path / \
--trust-remote-code \
--no-enable-prefix-caching \
--seed 1024 \
--data-parallel-size 4 \
--data-parallel-size-local 2 \
--data-parallel-address $node0_ip \
--data-parallel-rpc-port 13389 \
--tensor-parallel-size 4 \
--enable-expert-parallel \
--max-num-seqs 16 \
--max-model-len 32768 \
--max-num-batched-tokens 16384 \
--gpu-memory-utilization 0.9 \
--compilation-config '{"cudagraph_mode":"FULL_DECODE_ONLY"}' \
--speculative-config '{"method":"eagle3", "model":"lightseekorg/kimi-k2.5-eagle3", "num_speculative_tokens":3}' \
--mm-encoder-tp-mode data \
--enable-auto-tool-choice \
--tool-call-parser kimi_k2 \
--reasoning-parser kimi_k2
启动命令:
#!/bin/sh
# this obtained through ifconfig
# nic_name is the network interface name corresponding to local_ip of the current node
nic_name="xxx"
local_ip="192.xx.xx.2"
# The value of node0_ip must be consistent with the value of local_ip set in node0 (master node)
node0_ip="xxxx"
export HCCL_IF_IP=$local_ip
export GLOO_SOCKET_IFNAME=$nic_name
export TP_SOCKET_IFNAME=$nic_name
export HCCL_SOCKET_IFNAME=$nic_name
export HCCL_INTRA_PCIE_ENABLE=1
export HCCL_INTRA_ROCE_ENABLE=0
# [Optional] jemalloc
# jemalloc is for better performance, if `libjemalloc.so` is installed on your machine, you can turn it on.
export LD_PRELOAD=/usr/lib/aarch64-linux-gnu/libjemalloc.so.2:$LD_PRELOAD
echo performance | tee /sys/devices/system/cpu/cpu*/cpufreq/scaling_governor
sysctl -w vm.swappiness=0
sysctl -w kernel.numa_balancing=0
sysctl -w kernel.sched_migration_cost_ns=50000
export HCCL_OP_EXPANSION_MODE="AIV"
export PYTORCH_NPU_ALLOC_CONF=expandable_segments:True
export OMP_PROC_BIND=false
export OMP_NUM_THREADS=1
export TASK_QUEUE_ENABLE=1
export HCCL_BUFFSIZE=1024
# Ensure the model path matches the directory recorded during download
vllm serve Eco-Tech/Kimi-K2.5-w4a8 --additional-config '{"scheduler_config":{"enable_balance_scheduling":true},"enable_mlapo":true}' \
--host 0.0.0.0 \
--port 8088 \
--quantization ascend \
--served-model-name kimi_k25 \
--allowed-local-media-path / \
--trust-remote-code \
--no-enable-prefix-caching \
--seed 1024 \
--headless \
--data-parallel-size 4 \
--data-parallel-size-local 2 \
--data-parallel-start-rank 2 \
--data-parallel-address $node0_ip \
--data-parallel-rpc-port 13389 \
--tensor-parallel-size 4 \
--enable-expert-parallel \
--max-num-seqs 16 \
--max-model-len 32768 \
--max-num-batched-tokens 16384 \
--gpu-memory-utilization 0.9 \
--compilation-config '{"cudagraph_mode":"FULL_DECODE_ONLY"}' \
--speculative-config '{"method":"eagle3", "model":"lightseekorg/kimi-k2.5-eagle3", "num_speculative_tokens":3}' \
--mm-encoder-tp-mode data \
--enable-auto-tool-choice \
--tool-call-parser kimi_k2 \
--reasoning-parser kimi_k2
关键参数说明:
--data-parallel-size:所有节点上数据并行 rank 的总数。在此示例中,4表示模型总共拆分到 4 个 DP rank(每个节点 2 个)。--data-parallel-size-local:当前节点上运行的数据并行 rank 数量。在此示例中,每个节点运行 2 个 DP rank。--data-parallel-start-rank:此节点上数据并行 rank 的起始 rank 偏移。节点 0 从 rank 0 开始(默认),节点 1 从 rank 2 开始。这确保每个节点的 DP rank 在整体 rank 空间中占据不同位置。--data-parallel-address:数据并行主节点(节点 0)的 IP 地址。此值必须与节点 0 上设置的local_ip一致。--data-parallel-rpc-port:数据并行主通信的 RPC 端口。所有节点必须相同。--headless:表示此 vLLM 实例不是主服务节点。仅在非主节点(节点 1)上设置。主节点(节点 0)不应设置此标志。- 对于单节点部署,我们建议使用
dp4 tp4而不是dp2 tp8。
常见问题提示:如果遇到问题,请参阅公共 FAQ进行故障排查。
服务验证:
curl http://<node0_ip>:8088/v1/chat/completions \
-H "Content-Type: application/json" \
-d '{
"model": "kimi_k25",
"messages": [{
"role": "user",
"content": [
{
"type": "text",
"text": "The future of AI is"
}]
}],
"max_tokens": 1024,
"temperature": 1.0,
"top_p": 0.95
}'
预期结果:
服务返回 HTTP 200 OK。JSON 响应包含带有生成文本的 choices 字段。
5.3 多节点 PD 分离部署¶
我们建议使用 Mooncake 进行部署:Mooncake。
在标准单节点部署模式下,Prefill(提示词处理)和 Decode(令牌生成)任务运行在同一组 NPU 上。PD(Prefill-Decode)分离通过将 Prefill 和 Decode 运行在专用节点组上来解决此问题,每个节点组独立配置:
- Prefill 节点专注于高吞吐提示词处理,针对计算和通信进行优化(例如,启用序列并行以加速 Allreduce)。
- Decode 节点专注于低延迟令牌生成,针对内存带宽进行优化(例如,启用 MLAPO 融合算子)。
对于需要同时满足稳定延迟和高吞吐量的并发多用户工作负载,推荐在生产部署中使用此架构。
以 Atlas 800 A3(64GB × 16)为例,我们建议部署 2P1D(4 个节点)而不是 1P1D(2 个节点),因为在 1P1D 情况下 NPU 内存不足以支撑高并发。
Kimi-K2.5-w4a8 2P1D:需要 4 个 Atlas 800 A3(64GB × 16)节点。
要运行 vllm-ascend 的 Prefill-Decode Disaggregation 服务,需要在每个节点上部署 launch_online_dp.py 脚本和 run_dp_template.sh 脚本,并在 Prefill 主节点上部署 proxy.sh 脚本以转发请求。
参数说明:
| 参数 | 类型 | 必填 | 默认值 | 说明 |
|---|---|---|---|---|
--dp-size |
int | 是 | - | 数据并行大小(所有节点上 DP rank 的总数)。 |
--tp-size |
int | 否 | 1 | 每个 DP rank 内的张量并行大小。 |
--dp-size-local |
int | 否 | (与 --dp-size 相同) |
当前节点上的 DP rank 数量。如果未设置,默认为 --dp-size。 |
--dp-rank-start |
int | 否 | 0 | 此节点上数据并行 rank 的起始 rank 偏移。 |
--dp-address |
str | 是 | - | 数据并行主节点(节点 0)的 IP 地址。 |
--dp-rpc-port |
str | 否 | 12345 | 数据并行主通信的 RPC 端口。 |
--vllm-start-port |
int | 否 | 9000 | 此节点上每个 vLLM 引擎实例的起始端口。每个 DP rank 的引擎端口 = vllm_start_port + 本地 rank 索引。 |
-
run_dp_template.sh脚本# this obtained through ifconfig # nic_name is the network interface name corresponding to local_ip of the current node nic_name="xxx" local_ip="192.xx.xx.1" # The value of node0_ip must be consistent with the value of local_ip set in node0 (master node) node0_ip="xxxx" export HCCL_IF_IP=$local_ip export GLOO_SOCKET_IFNAME=$nic_name export TP_SOCKET_IFNAME=$nic_name export HCCL_SOCKET_IFNAME=$nic_name # [Optional] jemalloc # jemalloc is for better performance, if `libjemalloc.so` is installed on your machine, you can turn it on. export LD_PRELOAD=/usr/lib/aarch64-linux-gnu/libjemalloc.so.2:$LD_PRELOAD echo performance | tee /sys/devices/system/cpu/cpu*/cpufreq/scaling_governor sysctl -w vm.swappiness=0 sysctl -w kernel.numa_balancing=0 sysctl kernel.sched_migration_cost_ns=50000 export VLLM_RPC_TIMEOUT=3600000 export VLLM_EXECUTE_MODEL_TIMEOUT_SECONDS=30000 export HCCL_OP_EXPANSION_MODE="AIV" export PYTORCH_NPU_ALLOC_CONF=expandable_segments:True export OMP_PROC_BIND=false export OMP_NUM_THREADS=1 export TASK_QUEUE_ENABLE=1 export LD_LIBRARY_PATH=/usr/local/Ascend/ascend-toolkit/latest/python/site-packages/mooncake:$LD_LIBRARY_PATH export HCCL_BUFFSIZE=256 export ASCEND_RT_VISIBLE_DEVICES=$1 # Ensure the model path matches the directory recorded during download vllm serve Eco-Tech/Kimi-K2.5-w4a8 \ --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 \ --seed 1024 \ --quantization ascend \ --served-model-name kimi_k25 \ --trust-remote-code \ --max-num-seqs 8 \ --max-model-len 32768 \ --max-num-batched-tokens 16384 \ --no-enable-prefix-caching \ --gpu-memory-utilization 0.8 \ --enforce-eager \ --enable-auto-tool-choice \ --tool-call-parser kimi_k2 \ --reasoning-parser kimi_k2 \ --speculative-config '{"method": "eagle3", "model":"lightseekorg/kimi-k2.5-eagle3", "num_speculative_tokens": 1}' \ --additional-config '{"recompute_scheduler_enable":true}' \ --mm-encoder-tp-mode data \ --kv-transfer-config \ '{"kv_connector": "MooncakeConnectorV1", "kv_role": "kv_producer", "kv_port": "30000", "engine_id": "0", "kv_connector_extra_config": { "prefill": { "dp_size": 2, "tp_size": 8 }, "decode": { "dp_size": 32, "tp_size": 1 } } }'
```shell
# this obtained through ifconfig
# nic_name is the network interface name corresponding to local_ip of the current node
nic_name="xxx"
local_ip="192.xx.xx.2"
# The value of node0_ip must be consistent with the value of local_ip set in node0 (master node)
node0_ip="xxxx"
export HCCL_IF_IP=$local_ip
export GLOO_SOCKET_IFNAME=$nic_name
export TP_SOCKET_IFNAME=$nic_name
export HCCL_SOCKET_IFNAME=$nic_name
# [Optional] jemalloc
# jemalloc is for better performance, if `libjemalloc.so` is installed on your machine, you can turn it on.
export LD_PRELOAD=/usr/lib/aarch64-linux-gnu/libjemalloc.so.2:$LD_PRELOAD
echo performance | tee /sys/devices/system/cpu/cpu*/cpufreq/scaling_governor
sysctl -w vm.swappiness=0
sysctl -w kernel.numa_balancing=0
sysctl kernel.sched_migration_cost_ns=50000
export VLLM_RPC_TIMEOUT=3600000
export VLLM_EXECUTE_MODEL_TIMEOUT_SECONDS=30000
export HCCL_OP_EXPANSION_MODE="AIV"
export PYTORCH_NPU_ALLOC_CONF=expandable_segments:True
export OMP_PROC_BIND=false
export OMP_NUM_THREADS=1
export TASK_QUEUE_ENABLE=1
export LD_LIBRARY_PATH=/usr/local/Ascend/ascend-toolkit/latest/python/site-packages/mooncake:$LD_LIBRARY_PATH
export HCCL_BUFFSIZE=256
export ASCEND_RT_VISIBLE_DEVICES=$1
# Ensure the model path matches the directory recorded during download
vllm serve Eco-Tech/Kimi-K2.5-w4a8 \
--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 \
--seed 1024 \
--quantization ascend \
--served-model-name kimi_k25 \
--trust-remote-code \
--max-num-seqs 8 \
--max-model-len 32768 \
--max-num-batched-tokens 16384 \
--no-enable-prefix-caching \
--gpu-memory-utilization 0.8 \
--enforce-eager \
--enable-auto-tool-choice \
--tool-call-parser kimi_k2 \
--reasoning-parser kimi_k2 \
--speculative-config '{"method": "eagle3", "model":"lightseekorg/kimi-k2.5-eagle3", "num_speculative_tokens": 1}' \
--additional-config '{"recompute_scheduler_enable":true}' \
--mm-encoder-tp-mode data \
--kv-transfer-config \
'{"kv_connector": "MooncakeConnectorV1",
"kv_role": "kv_producer",
"kv_port": "30100",
"engine_id": "1",
"kv_connector_extra_config": {
"prefill": {
"dp_size": 2,
"tp_size": 8
},
"decode": {
"dp_size": 32,
"tp_size": 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="192.xx.xx.3"
# The value of node0_ip must be consistent with the value of local_ip set in node0 (master node)
node0_ip="xxxx"
export HCCL_IF_IP=$local_ip
export GLOO_SOCKET_IFNAME=$nic_name
export TP_SOCKET_IFNAME=$nic_name
export HCCL_SOCKET_IFNAME=$nic_name
# [Optional] jemalloc
# jemalloc is for better performance, if `libjemalloc.so` is installed on your machine, you can turn it on.
export LD_PRELOAD=/usr/lib/aarch64-linux-gnu/libjemalloc.so.2:$LD_PRELOAD
echo performance | tee /sys/devices/system/cpu/cpu*/cpufreq/scaling_governor
sysctl -w vm.swappiness=0
sysctl -w kernel.numa_balancing=0
sysctl kernel.sched_migration_cost_ns=50000
export VLLM_RPC_TIMEOUT=3600000
export VLLM_EXECUTE_MODEL_TIMEOUT_SECONDS=30000
export HCCL_OP_EXPANSION_MODE="AIV"
export PYTORCH_NPU_ALLOC_CONF=expandable_segments:True
export OMP_PROC_BIND=false
export OMP_NUM_THREADS=1
export TASK_QUEUE_ENABLE=1
export LD_LIBRARY_PATH=/usr/local/Ascend/ascend-toolkit/latest/python/site-packages/mooncake:$LD_LIBRARY_PATH
export HCCL_BUFFSIZE=1100
export ASCEND_RT_VISIBLE_DEVICES=$1
# Ensure the model path matches the directory recorded during download
vllm serve Eco-Tech/Kimi-K2.5-w4a8 \
--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 \
--seed 1024 \
--quantization ascend \
--served-model-name kimi_k25 \
--trust-remote-code \
--max-num-seqs 48 \
--max-model-len 32768 \
--max-num-batched-tokens 256 \
--no-enable-prefix-caching \
--gpu-memory-utilization 0.95 \
--enable-auto-tool-choice \
--tool-call-parser kimi_k2 \
--reasoning-parser kimi_k2 \
--compilation-config '{"cudagraph_mode": "FULL_DECODE_ONLY"}' \
--additional-config '{"recompute_scheduler_enable":true,"multistream_overlap_shared_expert":false,"enable_mlapo":true}' \
--speculative-config '{"method": "eagle3", "model":"lightseekorg/kimi-k2.5-eagle3", "num_speculative_tokens": 3}' \
--kv-transfer-config \
'{"kv_connector": "MooncakeConnectorV1",
"kv_role": "kv_consumer",
"kv_port": "30200",
"engine_id": "2",
"kv_connector_extra_config": {
"prefill": {
"dp_size": 2,
"tp_size": 8
},
"decode": {
"dp_size": 32,
"tp_size": 1
}
}
}'
```shell
# this obtained through ifconfig
# nic_name is the network interface name corresponding to local_ip of the current node
nic_name="xxx"
local_ip="192.xx.xx.4"
# The value of node0_ip must be consistent with the value of local_ip set in node0 (master node)
node0_ip="xxxx"
export HCCL_IF_IP=$local_ip
export GLOO_SOCKET_IFNAME=$nic_name
export TP_SOCKET_IFNAME=$nic_name
export HCCL_SOCKET_IFNAME=$nic_name
# [Optional] jemalloc
# jemalloc is for better performance, if `libjemalloc.so` is installed on your machine, you can turn it on.
export LD_PRELOAD=/usr/lib/aarch64-linux-gnu/libjemalloc.so.2:$LD_PRELOAD
echo performance | tee /sys/devices/system/cpu/cpu*/cpufreq/scaling_governor
sysctl -w vm.swappiness=0
sysctl -w kernel.numa_balancing=0
sysctl kernel.sched_migration_cost_ns=50000
export VLLM_RPC_TIMEOUT=3600000
export VLLM_EXECUTE_MODEL_TIMEOUT_SECONDS=30000
export HCCL_OP_EXPANSION_MODE="AIV"
export PYTORCH_NPU_ALLOC_CONF=expandable_segments:True
export OMP_PROC_BIND=false
export OMP_NUM_THREADS=1
export TASK_QUEUE_ENABLE=1
export LD_LIBRARY_PATH=/usr/local/Ascend/ascend-toolkit/latest/python/site-packages/mooncake:$LD_LIBRARY_PATH
export HCCL_BUFFSIZE=1100
export ASCEND_RT_VISIBLE_DEVICES=$1
# Ensure the model path matches the directory recorded during download
vllm serve Eco-Tech/Kimi-K2.5-w4a8 \
--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 \
--seed 1024 \
--quantization ascend \
--served-model-name kimi_k25 \
--trust-remote-code \
--max-num-seqs 48 \
--max-model-len 32768 \
--max-num-batched-tokens 256 \
--no-enable-prefix-caching \
--gpu-memory-utilization 0.95 \
--enable-auto-tool-choice \
--tool-call-parser kimi_k2 \
--reasoning-parser kimi_k2 \
--compilation-config '{"cudagraph_mode": "FULL_DECODE_ONLY"}' \
--additional-config '{"recompute_scheduler_enable":true,"multistream_overlap_shared_expert":false,"enable_mlapo":true}' \
--speculative-config '{"method": "eagle3", "model":"lightseekorg/kimi-k2.5-eagle3", "num_speculative_tokens": 3}' \
--kv-transfer-config \
'{"kv_connector": "MooncakeConnectorV1",
"kv_role": "kv_consumer",
"kv_port": "30200",
"engine_id": "2",
"kv_connector_extra_config": {
"prefill": {
"dp_size": 2,
"tp_size": 8
},
"decode": {
"dp_size": 32,
"tp_size": 1
}
}
}'
```
关键参数说明:
additional_config.enable_mlapo=true:启用融合算子,可显著提升性能,但会消耗更多 NPU 内存。在预填充-解码(PD)分离场景中,仅在解码节点上启用 MLAPO。recompute_scheduler_enable: true:启用重计算调度器。当解码节点的键值缓存(KV Cache)不足时,请求将被发送到预填充节点以重新计算 KV Cache。在 PD 分离场景中,建议在预填充节点和解码节点上同时启用该配置。multistream_overlap_shared_expert: true:当张量并行(TP)大小为 1 或enable_shared_expert_dp: true时,会启用一个额外流,以重叠共享专家的计算过程,从而提升效率。
run_dp_template.sh 脚本使用位置参数($1-$7)从 launch_online_dp.py 接收配置值:
$1(ASCEND_RT_VISIBLE_DEVICES):分配给该 DP 实例的 NPU 设备,例如0,1,2,3或4,5,6,7。$2(--port):该 DP 实例的 vLLM 服务端口,从--vllm-start-port开始自动分配(例如7100、7101)。$3(--data-parallel-size):DP rank 总数。$4(--data-parallel-rank):该 DP 实例的 rank 索引。$5(--data-parallel-address):DP 主节点的 IP 地址。$6(--data-parallel-rpc-port):用于 DP 主节点通信的 RPC 端口。$7(--tensor-parallel-size):每个 DP rank 内的 TP 大小。
-
在每个节点上运行服务:
# p0 python launch_online_dp.py --dp-size 2 --tp-size 8 --dp-size-local 2 --dp-rank-start 0 --dp-address 192.xx.xx.1 --dp-rpc-port 12321 --vllm-start-port 7100 # p1 python launch_online_dp.py --dp-size 2 --tp-size 8 --dp-size-local 2 --dp-rank-start 0 --dp-address 192.xx.xx.2 --dp-rpc-port 12321 --vllm-start-port 7100 # d0 python launch_online_dp.py --dp-size 32 --tp-size 1 --dp-size-local 16 --dp-rank-start 0 --dp-address 192.xx.xx.3 --dp-rpc-port 12321 --vllm-start-port 7100 # d1 python launch_online_dp.py --dp-size 32 --tp-size 1 --dp-size-local 16 --dp-rank-start 16 --dp-address 192.xx.xx.3 --dp-rpc-port 12321 --vllm-start-port 7100 -
在预填充主节点上运行
proxy.sh脚本在与预填充器服务实例相同的节点上运行一个代理服务器。您可以在仓库的示例中获取代理程序:load_balance_proxy_server_example.py
python load_balance_proxy_server_example.py \ --port 1999 \ --host 192.xx.xx.1 \ --prefiller-hosts \ 192.xx.xx.1 \ 192.xx.xx.1 \ 192.xx.xx.2 \ 192.xx.xx.2 \ --prefiller-ports \ 7100 7101 7100 7101 \ --decoder-hosts \ 192.xx.xx.3 \ 192.xx.xx.3 \ 192.xx.xx.3 \ 192.xx.xx.3 \ 192.xx.xx.3 \ 192.xx.xx.3 \ 192.xx.xx.3 \ 192.xx.xx.3 \ 192.xx.xx.3 \ 192.xx.xx.3 \ 192.xx.xx.3 \ 192.xx.xx.3 \ 192.xx.xx.3 \ 192.xx.xx.3 \ 192.xx.xx.3 \ 192.xx.xx.3 \ 192.xx.xx.4 \ 192.xx.xx.4 \ 192.xx.xx.4 \ 192.xx.xx.4 \ 192.xx.xx.4 \ 192.xx.xx.4 \ 192.xx.xx.4 \ 192.xx.xx.4 \ 192.xx.xx.4 \ 192.xx.xx.4 \ 192.xx.xx.4 \ 192.xx.xx.4 \ 192.xx.xx.4 \ 192.xx.xx.4 \ 192.xx.xx.4 \ 192.xx.xx.4 \ --decoder-ports \ 7100 7101 7102 7103 7104 7105 7106 7107 7108 7109 7110 7111 7112 7113 7114 7115 \ 7100 7101 7102 7103 7104 7105 7106 7107 7108 7109 7110 7111 7112 7113 7114 7115
部署验证:
PD 分离服务完全启动后,通过 Prefill 主节点上的代理端口发送请求,以验证 Prefill 节点和 Decode 节点是否协同工作正常:
curl http://192.xx.xx.1:1999/v1/chat/completions \
-H "Content-Type: application/json" \
-d '{
"model": "kimi_k25",
"messages": [{
"role": "user",
"content": [
{
"type": "text",
"text": "The future of AI is"
}]
}],
"max_tokens": 1024,
"temperature": 1.0,
"top_p": 0.95
}'
预期结果:
代理返回 HTTP 200 OK。JSON 响应包含 choices 字段,其中包含生成的文本,确认 Prefill 节点已成功处理提示词,且 Decode 节点已生成响应:
{
"id": "chatcmpl-xxxxxxxxxxxxx",
"object": "chat.completion",
"model": "kimi_k25",
"choices": [
{
"index": 0,
"message": {
"role": "assistant",
"content": "The future of AI is not a destination we are passively approaching...",
"finish_reason": "length"
}
}
],
"usage": {
"prompt_tokens": 13,
"total_tokens": 1037,
"completion_tokens": 1024
}
}
常见问题提示:如果在 PD 分离部署中遇到问题,请参阅 公共常见问题 进行故障排查。
6 功能验证¶
服务器启动后,您可以使用输入提示词查询模型:
curl http://<node0_ip>:8088/v1/chat/completions \
-H "Content-Type: application/json" \
-d '{
"model": "kimi_k25",
"messages": [{
"role": "user",
"content": [
{
"type": "text",
"text": "The future of AI is"
}]
}],
"max_tokens": 1024,
"temperature": 1.0,
"top_p": 0.95
}'
预期结果:
服务返回 HTTP 200 OK。JSON 响应包含 choices 字段,其中包含生成的文本以及用量统计:
{
"id": "chatcmpl-xxxxxxxxxxxxx",
"object": "chat.completion",
"model": "kimi_k25",
"choices": [
{
"index": 0,
"message": {
"role": "assistant",
"content": "The future of AI is not a destination we are passively approaching...",
"finish_reason": "length"
}
}
],
"usage": {
"prompt_tokens": 13,
"total_tokens": 1037,
"completion_tokens": 1024
}
}
7 精度评估¶
以下是一种精度评估方法。
7.1 使用 AISBench¶
-
有关详细信息,请参阅 使用 AISBench。
-
执行完成后,您可以获取结果。以下是
Kimi-K2.5-w4a8在vllm-ascend:v0.17.0rc1中的结果,仅供参考。
| 数据集 | 版本 | 指标 | 模式 | vllm-api-general-chat | 备注 |
|---|---|---|---|---|---|
| GSM8K | - | accuracy | gen | 96.07 | 1 台 Atlas 800 A3(64GB × 16) |
| AIME2025 | - | accuracy | gen | 90.00 | 1 台 Atlas 800 A3(64GB × 16) |
| GPQA | - | accuracy | gen | 84.85 | 1 台 Atlas 800 A3(64GB × 16) |
| TextVQA | - | accuracy | gen | 80.29 | 1 台 Atlas 800 A3(64GB × 16) |
8 性能评估¶
8.1 使用 AISBench¶
有关详细信息,请参阅 使用 AISBench 进行性能评估。
8.2 使用 vLLM Benchmark¶
以运行 Kimi-K2.5-w4a8 的性能评估为例。
有关更多详细信息,请参阅 vllm benchmark。
vllm bench 有三个子命令:
latency:基准测试单个请求批次的延迟。serve:基准测试在线服务吞吐量。throughput:基准测试离线推理吞吐量。
以 serve 为例,按如下方式运行代码。
vllm bench serve --model Eco-Tech/Kimi-K2.5-w4a8 --dataset-name random --random-input 1024 --num-prompts 200 --request-rate 1 --save-result --result-dir ./
大约几分钟后,您可以获取性能评估结果。
9 性能调优¶
9.1 推荐配置¶
注意:以下配置已在特定测试环境中验证,仅供参考。最优配置取决于最大输入/输出长度、前缀缓存命中率、精度要求以及部署机器比例等因素。建议参考第 9.2 节,根据实际条件进行调优。
表 1:场景概览¶
*Total NPUs表示所有节点使用的 NPU 总数。1 个节点 = 1 台 Atlas 800 A3 服务器(64GB × 16 个 NPU)。
| 场景 | 部署模式 | *Total NPUs | 权重版本 | 关键考量 |
|---|---|---|---|---|
| 高吞吐 / 低延迟 (16k 上下文) |
单节点混合 | 16 (A3) | kimi-k2.5-w4a8 | 使用 dp4 tp4 以获得最佳吞吐量和低延迟 |
| 高吞吐 / 低延迟 (16k 上下文) |
2 节点数据并行 | 16 (A2) | kimi-k2.5-w4a8 | 跨 2 个节点使用 dp4 tp4;延迟和吞吐量均衡 |
| 高吞吐 / 低延迟 (16k 上下文) |
2P1D 部署 | 64 (A3) | kimi-k2.5-w4a8 | Prefill:dp2 tp8;Decode:dp32 tp1 以支持高并发 |
| 长上下文 (128k,低并发 ≤4) |
单节点混合 | 16 (A3) | kimi-k2.5-w4a8 | 使用 dp1 tp16 以最大化 TP,适应极端上下文长度 |
| 长上下文 (128k,高并发 >4) |
单节点混合 | 16 (A3) | kimi-k2.5-w4a8 | 使用 dp2 tp8 以优化内存带宽并支持更高并发 |
表 2:详细节点配置¶
|场景|配置|NPUs|TP|DP|最大模型长度|MTP 推测数|
| 场景 | 配置 | NPU 数量 | TP | DP | 最大模型长度 | MTP 推测数量 |
|---|---|---|---|---|---|---|
| -------- | ------------- | ----- | -- | -- | ------------- | -------------------- |
| 高吞吐 / 低延迟(16k) | 服务器 / 单机 | 16 | 4 | 4 | ~16k | 3 |
| 高吞吐 / 低延迟(16k) | 服务器 / 2 节点 DP | 8 | 4 | 2 | ~16k | 3 |
| 高吞吐 / 低延迟(16k) | Server-P 节点 | 16 | 8 | 2 | ~16k | 3 |
| 高吞吐 / 低延迟(16k) | Server-D 节点 | 16 | 1 | 32 | ~16k | 3 |
| 长上下文(128k,低并发 ≤4) | 服务器 / 单机 | 16 | 16 | 1 | 128k | 3 |
| 长上下文(128k,高并发 >4) | 服务器 / 单机 | 16 | 8 | 2 | 128k | 3 |
完整的启动命令和参数说明,请参见第 5 章中的部署示例。
注意:
max-model-len 和 max-num-seqs 需要根据实际使用场景进行设置。其他设置请参见 部署 章节。
9.2 调优指南¶
调优方法请参见公共性能调优文档。
详细功能说明请参见功能矩阵。
10 常见问题¶
对于常见的环境、安装和通用参数问题,请参见公共常见问题解答;本章仅涵盖模型特定问题。
- 问:单节点部署推荐的 TP/DP 配置是什么?
答:对于单节点部署,我们建议使用 dp4 tp4 而不是 dp2 tp8。
本页原文 Markdown:在 AtomGit 查看·内容源自开源项目 vl/vllm-ascend