TriMoE: Augmenting GPU with AMX-Enabled CPU and DIMM-NDP for High-Throughput MoE Inference via Offloading
TriMoE 提出 GPU-CPU-NDP 三级异构推理架构,将 MoE 专家按热度精确映射到最优计算单元:热专家留 GPU、暖专家用 AMX CPU、冷专家卸载至 DIMM-NDP。配合瓶颈感知调度和预测驱动的动态重布局,实现 2.83x 加速。
INSIGHTMoE 推理中存在大量暖专家既不够热(GPU)也不够冷(NDP),AMX CPU 恰好填补这一计算空白。三级分层卸载比二级方案效率高 2.83 倍,为搜推 MoE 模型的低成本部署提供新思路。
业务关联搜推线上服务部署 MoE 模型时 GPU 成本高昂,TriMoE 的异构卸载方案可以用更少的 GPU 实现相同吞吐,直接降低 serving 成本。
证据相比最优 GPU-CPU 和 GPU-NDP 方案,TriMoE 实现最高 2.83x 加速;暖专家分配至 AMX CPU 可充分利用其计算能力同时避免 NDP 的计算瓶颈。