MoE 批量推理免费提速:生成前先按领域排序 prompt,最高 +17%

yb2698 · x · 2026-10-20

一个零成本的 MoE 批量推理优化技巧:在调用 llm.generate 前先对 prompt 排序。

原理:MoE 模型每个 decode 步都要读取批内所有请求选中的专家;混合领域的批次会触及更多专家,而同领域批次专家高度重叠,缓存命中更高。

实测(原生 vLLM、同样的请求,只改顺序):

只需一个 argsort,无需改任何模型或框架代码。

原文链接 →

「Infra」频道最新

更多「Infra」频道 AI 资讯 →