DeepSeek第二炸!开源首个用于MoE模型训练通信库

4个月前 16观看

DeepSeek 的“开源周”活动今日已经来到第二天,今天发布的是首个开源的用于 MoE 模型训练和推理的 EP 通信库 ——DeepEP。

YYo喜好网-记录每日喜好的科技时尚娱乐生活

YYo喜好网-记录每日喜好的科技时尚娱乐生活

官方表示其具备如下特征:YYo喜好网-记录每日喜好的科技时尚娱乐生活

高效优化的全到全通信方式YYo喜好网-记录每日喜好的科技时尚娱乐生活

支持节点内外通信,兼容 NVLink 和 RDMA 技术YYo喜好网-记录每日喜好的科技时尚娱乐生活

提供高吞吐量的内核,提升训练和推理前期填充效率YYo喜好网-记录每日喜好的科技时尚娱乐生活

提供低延迟内核,优化推理解码速度YYo喜好网-记录每日喜好的科技时尚娱乐生活

完全支持 FP8 数据格式调度YYo喜好网-记录每日喜好的科技时尚娱乐生活

提供灵活的 GPU 资源管理,支持计算与通信的重叠执行

YYo喜好网-记录每日喜好的科技时尚娱乐生活

YYo喜好网-记录每日喜好的科技时尚娱乐生活

据介绍,DeepEP 是一款专为混合专家(MoE)和专家并行(EP)设计的通信库,提供了高吞吐量和低延迟的 all-to-all GPU 内核,常用于 MoE 派发和合并操作。该库还支持低精度计算,包括 FP8。YYo喜好网-记录每日喜好的科技时尚娱乐生活

为了与 DeepSeek-V3 论文中提出的组限制门控算法兼容,DeepEP 提供了一些针对不对称带宽转发优化的内核,比如将数据从 NVLink 域转发到 RDMA 域。这些优化的内核能够提供高吞吐量,适合用于训练和推理的预填充任务,同时支持 SM(流式多处理器)数量控制。YYo喜好网-记录每日喜好的科技时尚娱乐生活

对于延迟敏感型的推理解码任务,DeepEP 提供了一套低延迟内核,采用纯 RDMA 技术以最大程度减少延迟。此外,该库还采用了一种基于 Hook 的通信与计算重叠方法,不会占用任何 SM 资源。YYo喜好网-记录每日喜好的科技时尚娱乐生活

本文链接:http://www.xihao.site/showinfo-3-86631.htmlDeepSeek第二炸!开源首个用于MoE模型训练通信库

声明:本网页内容旨在传播知识,若有侵权等问题请及时与本网联系,我们将在第一时间删除处理。邮件:2376512515@qq.com。天上从来不会掉馅饼,请大家时刻谨防诈骗