首页
学习
活动
专区
圈层
工具
发布
精选内容/技术社群/优惠产品,尽在小程序
立即前往

DeepSeek开源第四弹:并行优化策略3项齐开

【TechWeb】2月27日消息,DeepSeek开源周第四日,DeepSeek开源了并行优化策略(Optimized Parallelism Strategies),一次开源了3项:

DualPipe:一种用于V3/R1模型训练中实现计算与通信重叠的双向流水线并行算法

EPLB:一个针对V3/R1的专家并行负载均衡工具

Profile-data:训练和推理框架的分析数据

GitHub地址:

https://github.com/deepseek-ai/DualPipe

https://github.com/deepseek-ai/eplb

https://github.com/deepseek-ai/profile-data

一、DualPipe

DualPipe是DeepSeek-V3技术报告中提出的一种创新双向流水线并行算法。

它能够实现前向与后向计算和通信阶段的完全重叠,同时有效减少流水线气泡(空闲时间)。

DeepSeek展示了在8个流水线并行阶段和20个micro-batches情况下,DualPipe在两个方向上的调度示例。(来自DeepSeek-V3技术报告)

反向方向的微批次与前向方向对称,为了简化图示,这里省略了反向方向的批次 ID。图中由同一个黑色边框包围的两个单元格具有相互重叠的计算和通信。

流水线气泡与内存使用比较

表中,

  • 发表于:
  • 原文链接https://page.om.qq.com/page/O-fooCiMNR6upF9nqGTOfLLw0
  • 腾讯「腾讯云开发者社区」是腾讯内容开放平台帐号(企鹅号)传播渠道之一,根据《腾讯内容开放平台服务协议》转载发布内容。
  • 如有侵权,请联系 cloudcommunity@tencent.com 删除。

扫码

添加站长 进交流群

领取专属 10元无门槛券

私享最新 技术干货

扫码加入开发者社群
领券