尝试预测下一个token会用到哪些MoE专家来加速CPU/GPU offload,得到了一些实际数据,这真的可实现吗还是我在浪费时间(30tg/s -> 150-200tg/s)

Reddit r/LocalLLaMA 论文

摘要

探索通过预测下一个token将使用的MoE专家来改进CPU/GPU offload,实现了30->150-200 tg/s的加速,并质疑实现可行性。

暂无内容
查看原文

相似文章

通过自适应专家分配实现高效的CPU-GPU协同MoE推理

arXiv cs.AI

本文提出了Apex,一个CPU-GPU协同的MoE推理系统,该系统根据token负载自适应地将GPU未命中的专家分配到CPU或GPU上,相比最强基线,预填充速度提升最高达51.9%,解码吞吐量提升最高达39.5%。