推理显存建模

Memory planner for distributed MoE inference

ESTIMATED PER DEVICE
47.77GiB

单卡总显存预估(含权重)

ACTIVE MODELMiniMax M3
Hidden size6,144
专家总数128
TopK 专家4
EP size32TP 4 × DP 8
本地专家数4128 ÷ EP 32
官方配置 ↗
权重占用
18.60 GiB38.9% of total
KV + Index Cache
22.13 GiB46.3% of total
激活占用
0.28 GiB0.6% of total
HCCL buffer
1.26 GiB2.6% of total
ACLGraph 占用
0.25 GiB0.5% of total
CANN + PTA + 算子
1.00 GiB2.1% of total
Device OS
4.25 GiB8.9% of total
MEMORY MAP

显存构成

GiB
权重占用KV + Index Cache激活占用HCCL bufferACLGraph 占用CANN + PTA + 算子Device OS
权重占用18.60 GiB
全量模型 411.65 GiB当前单卡 18.60 GiB本地专家 4
Routed Experts MXFP8 payload57 × (128 ÷ 32) × 3 × 6144 × 3072 × 1 B
12.02 GiB
Routed Experts MXFP8 scale随本地专家数 E ÷ EP 切分;每 [1, 32] block 1 B
384.8 MiB
Attention QKV(TP 4)Q/K/V 按 Attention TP 4 切分
835.3 MiB
Attention O-proj(TP 4)60 × 6144 × (64 × 128 ÷ TP 4);MXFP8 payload 720 MiB + scale 22.5 MiB
742.5 MiB
Attention Indexer Q/K(TP 4)57 层 × [Q: 4 heads ÷ TP 4;K: 1 head 复制] × head dim 128;MXFP8 + scale
88.2 MiB
Attention 辅助张量Q/K Norm、Indexer Q/K Norm 等
0.1 MiB
Dense MLP FP8 payload3 × 3 × H × (12288 ÷ 4)
162 MiB
Dense MLP MXFP8 scale按 TP 切分后的实际矩阵 shape 计算
5.1 MiB
Shared Experts MXFP8 payload(TP 1)Lm × Ns × 3 × H × I × 1 B;不切分
3.01 GiB
Shared Experts MXFP8 scaleTP 1,不随 EP 切分,每个 Device 保留完整 scale
96.2 MiB
Router(FP32)57 × 128 × (6144 + 1) × 4 B
171 MiB
NormBF16 Norm tensors
2.8 MiB
Embedding200064 ÷ 4 × 6144 × 2 B
586.1 MiB
LM Head200064 ÷ 4 × 6144 × 2 B
586.1 MiB
MiscRoPE inv_freq 等 buffer
0 MiB
KV + Index Cache22.13 GiB
K/V Cache(FP8 / INT8)131072 × 8 × 60 × 2 × (4 ÷ TP 4) × 128 × 1 B;有效 60 层,K/V 各存一份,随 TP 切分
15.00 GiB
Index Cache(FP8 / INT8)131072 × 8 × 57 × 128 × 1 B;有效 57 层,Index K 随 TP 复制,不切分
7.13 GiB
激活占用0.28 GiB
Hidden states + residual2 × 2 B × 4096 × 6144
96 MiB
4 份 MoE 激活 buffer4 × 2 B × 8 × 4096 × 4 ÷ 32 × 6144
192 MiB
HCCL buffer1.26 GiB
DP buffermax(ceil((8 + 1) × 4 ÷ 1024²), 50) × 2 MB
95.4 MiB
TP buffer200 MB × 2
381.5 MiB
EP buffer200 MB × 2
381.5 MiB
MC2 buffer(本地专家 4)2 × (128 ÷ 32 × Max BS × EP × 480Align512 + K × Max BS × Align512)
428 MiB
Dispatch 416 MiBCombine 12 MiB480Align512 = 13,312 BAlign512 = 12,288 B
其他运行时5.50 GiB
ACLGraph 占用(5 张)5 ÷ 5 × 0.27 GB
257.5 MiB
CANN + PTA + 算子1 GiB 预估值
1.00 GiB
Device OS 固定占用4.25 × 1024³ bytes = 4.25 GiB
4.25 GiB

口径说明 所有结果均为单卡估算并统一显示为 GiB。MiniMax M3 权重包含 MXFP8 scale;EP 自动等于 TP × DP。KV Cache 与 Index Cache 按缓存 token 总数估算,不包含分配器开销。