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

业界 来源:游民星空 2025-02-26 08:21:41

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

官方表示其具备如下特征:

高效优化的全到全通信方式

支持节点内外通信,兼容 NVLink 和 RDMA 技术

提供高吞吐量的内核,提升训练和推理前期填充效率

提供低延迟内核,优化推理解码速度

完全支持 FP8 数据格式调度

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

据介绍,DeepEP 是一款专为混合专家(MoE)和专家并行(EP)设计的通信库,提供了高吞吐量和低延迟的 all-to-all GPU 内核,常用于 MoE 派发和合并操作。该库还支持低精度计算,包括 FP8。

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

对于延迟敏感型的推理解码任务,DeepEP 提供了一套低延迟内核,采用纯 RDMA 技术以最大程度减少延迟。此外,该库还采用了一种基于 Hook 的通信与计算重叠方法,不会占用任何 SM 资源。

延伸阅读
  • 雷军回应REDMI K90扬声器容易进灰:做了两层防护!

    根据微博@新浪科技的消息称,今天上午,REDMI红米手机官方在微博发布《答网友问》,小米创办人、董事长兼CEO雷军回应了REDMI K90系列的几个常见问题。对于粉丝们关心的“背部的大尺寸扬声器会不会

  • 黄翠如萧正楠谈许绍雄离世:约定成永别 很舍不得

    据港媒,28日,黄翠如、萧正楠接受采访时谈及许绍雄去世,表示很舍不得。黄翠如夫妇与许绍雄是挚友邻居,对方常邀他们吃火锅,疫情时还让女儿给生病的黄翠如送面包。“我们和Benz哥是邻居,经常我们没饭吃的时

  • 徐冬冬尹子维吵出结果了:翻篇 婚礼分两地办

    近日,貌似尹子维与徐冬冬因婚礼筹备问题产生争执。10月28日凌晨,尹子维连发多条微博,宣布将于2026年2月26日在哈尔滨举办婚礼,但是一个人的婚礼。“大家2月26哈尔滨!参加我一个的婚礼!”网友纷纷

关注公众号:拾黑(shiheibook)了解更多

友情链接:

关注数据与安全,洞悉企业级服务市场:https://www.ijiandao.com/
安全、绿色软件下载就上极速下载站:https://www.yaorank.com/

公众号 关注网络尖刀微信公众号
随时掌握互联网精彩
赞助链接