研究人员开发了ThunderEP,这是一种新的通信设计,用于在通过PCIe连接的消费级GPU上高效运行大型专家混合(MoE)模型。该系统解决了这些设置中固有的通信瓶颈,数据传输通常涉及CPU。ThunderEP旨在绕过CPU中继跳跃并利用DMA引擎,避免与计算资源的争用,从而减少同步延迟。在配备RTX 4090和RTX 5090 GPU的系统上进行的评估表明,与NCCL等现有方法相比,速度显著提升,MoE推理的端到端改进高达1.66倍。 AI
影响 优化消费级硬件上大型MoE模型的推理性能,可能降低研究人员和开发人员的门槛。
排序理由 该项目是一篇研究论文,详细介绍了一种在特定硬件上优化AI模型推理的新技术方法。[lever_c_demoted from research: ic=1 ai=1.0]
- Chris Jaehwan Lee
- graphics processing unit
- Mixture of Experts (MoE)
- NCCL
- PCI Express
- RTX 4090
- RTX 5090
- ThunderEP
- vLLM
AI 生成摘要 · Google Gemini · 来自 1 个来源。 我们如何撰写摘要 →