跳到正文
掘金· 用户36005557900·· 25 天前精选AI 评分78

Kestrel_VLLM 第 15 篇:decode 四连击把 30B-A3B MoE 从 19ms/tok 降到 8.4ms/tok

Kestrel_VLLM - 15 · decode 优化四连击:19ms → 8ms

AI 导读

Kestrel_VLLM 系列第 15 篇记录在 RTX 5090 上跑 Qwen3-30B-A3B q4 的 decode 优化,64 token 吞吐从 52.7 tok/s 提升到 118.8 tok/s,即 19.0 ms/tok 降到 8.4 ms/tok,总加速 2.25×。

借鉴点

能用 VLLM_CUDA_STUB_{ATTN,ROUTER,GU,LMHEAD} 逐段跳过定位真瓶颈,再套转置归约、router 并行化、M=1 线程映射、CUDA Graph 四个小改动;代价是 attention 与 router 属近似安全,换 prompt 要重新对拍 greedy。

来源:掘金 · juejin.cn