网易首页 > 网易号 > 正文 申请入驻

DeepSpeed v0.19.9发布:Triton算子、AutoEP、AutoTP、ZeRO与CPU优化迎来集中修复与增强

0
分享至




DeepSpeed v0.19.9 已于 2026 年 10 月 30 日发布。本次版本是在 v0.19.8 之后的补丁更新,覆盖训练、推理、并行、专家模型、通信、优化器、CPU 后端、CI 测试以及文档维护等多个方向。

从发布记录看,本次更新共有 19 位贡献者参与,包含 28 次提交,涉及 105 个文件变更。虽然这是一个补丁版本,但改动范围并不局限于单点问题修复,而是对 Triton 内核、词表并行计算、AutoTP、AutoEP、ZeRO、AffineIR、HybridEngine、通信日志、CPU Adam、测试体系和文档链接进行了集中完善。

对于关注大规模训练、混合并行、专家模型训练、CPU 加速器适配,以及训练稳定性的用户来说,v0.19.9 值得重点关注。

一、版本与发布流程调整

本次更新首先完成了 v0.19.8 发布后的版本号更新,确保项目版本状态进入新的补丁版本周期。

同时,PR 门禁流程新增了对version.txt的过滤处理。该调整意味着版本文件不会再对相关 PR 门禁产生不必要影响,使发布后的版本维护流程更加清晰。

与版本管理相关的改动虽然不直接影响模型训练结果,但对于持续集成、自动化验证和发布稳定性具有基础意义。尤其是在频繁迭代的项目中,版本文件变化往往会带来无关的验证触发,本次处理正是对此类流程问题的针对性修复。

二、Triton 计算内核更新:词表并行与数值稳定性同步推进

v0.19.9 中一个重要方向是 Triton 内核能力的扩展与稳定性修复。

本次新增了用于词表并行场景的 Triton 内核,覆盖以下计算:

  • • Logprobs

  • • Entropy

  • • KL

词表并行是大词表模型训练与推理中的关键能力。对于大规模语言模型而言,输出词表往往会带来显著的计算和存储开销。此次针对词表并行的 logprobs、熵和 KL 计算新增 Triton 内核,意味着相关计算路径获得了专门的内核支持。

其中,logprobs 常用于输出概率计算与结果评估;熵用于描述概率分布的不确定性;KL 则用于衡量两个概率分布之间的差异。这些计算在训练、强化学习、蒸馏、策略优化和概率分布约束等场景中都具有重要位置。

除了新增内核外,本次还修复了 Triton 掩码 Softmax 的数值稳定性问题。

掩码 Softmax 是注意力计算和序列建模中的基础操作。在实际模型运行中,掩码位置往往涉及极小值、无效位置或特殊填充值。数值稳定性问题一旦出现,可能影响后续概率分布计算。v0.19.9 对 Triton masked softmax 的数值稳定性进行了修复,使该路径在处理掩码场景时更加可靠。

从这两项改动可以看到,本版本对 Triton 的维护同时覆盖了“新增高价值计算能力”和“修复基础算子稳定性”两个方面。

三、AutoTP 更新:输出头绑定与分片写入校验

AutoTP 是本次更新的重要区域之一,相关改动涉及模型结构兼容性与分片数据写入检查。

首先,v0.19.9 调整了 AutoTP 对行并行输出头与嵌入层绑定关系的预期。

具体来说,当一个行并行的输出头与嵌入层保持权重绑定时,AutoTP 需要维持这一关系。本次更新明确了这一预期,确保 AutoTP 在相关模型结构下能够正确处理输出头与嵌入层之间的绑定关系。

权重绑定通常意味着输入嵌入与输出投影使用相同或关联的参数。对于并行化框架而言,这类关系需要在模型切分、参数布局和运行时计算过程中保持一致。此次更新聚焦于行并行输出头与嵌入绑定的兼容处理。

其次,本次还新增了对列切分片段写入位置的检查。

当参数或张量按列方向进行切分时,各个切分片段需要写入正确的分片位置。v0.19.9 增加了对列切分片段在 shard 中写入位置的验证,进一步加强了分片布局正确性的检查能力。

此外,针对某个模型结构中的输出投影偏置,本版本要求该偏置在每一个 rank 上保持完整,并将其描述为复制状态。

也就是说,该输出投影偏置不再按照局部切分方式处理,而是在各个 rank 上完整保留。这一变化对应的是模型参数布局与并行策略之间的兼容性处理,特别是涉及输出投影层偏置时,需要确保各个 rank 上的参数状态与并行语义一致。

四、AutoEP 更新:融合内核、分组顺序与竞态修复

AutoEP 在 v0.19.9 中获得了多项更新,覆盖专家激活、ZeRO-3 场景下的通信顺序,以及测试中的竞态问题。

首先,本次为被钳制的专家激活新增了融合 Triton 内核。

专家模型场景中,激活值可能需要执行钳制操作。此次新增的融合 Triton 内核面向这一计算过程,使被钳制的专家激活具备专用的融合实现。

其次,v0.19.9 修复了 AutoEP 在 ZeRO-3 条件下进行 all-gather 时,分区组顺序可能不一致的问题。

在分布式训练中,all-gather 是常见通信操作。当不同参与方对于分区组的顺序理解不一致时,可能影响数据聚合过程。本次更新确保在 ZeRO-3 场景下,AutoEP 的 all-gather 能够使用一致的分区组顺序。

这项改动同时涉及 AutoEP 与 ZeRO-3 两个关键机制,重点在于保持通信分组的确定性和一致性。

此外,本次还修复了 AutoEP 与 AutoTP 联合测试中的竞态问题。

该问题出现在折叠式检查点加载与元数据删除之间。当检查点加载与元数据清理操作发生时序竞争时,测试过程可能出现不稳定。本次修复处理了折叠检查点加载和元数据删除之间的竞态关系,使 AutoEP 与 AutoTP 的联合测试更加可靠。

五、AffineIR 更新:基于运行时布局规划专家传输

AffineIR 在本次版本中包含两项核心更新,均与参数、优化器状态和专家传输的运行时布局有关。

第一项更新是:根据实时运行时放置状态规划专家传输。

这意味着,专家传输不再仅依赖静态信息,而是基于当前运行时的实际放置状态进行规划。该变更面向专家模型的数据传输和布局管理过程,重点是让传输规划与当前运行时 placement 保持对应。

第二项更新是:为存活的 ZeRO 参数和优化器状态提供仿射存储映射。

本次增加的仿射存储映射,面向运行中的 ZeRO 参数及其优化器状态。这里的重点在于,参数和优化器状态的存储布局能够使用 AffineIR 的映射方式进行表达和管理。

两项 AffineIR 更新共同指向一个方向:让运行时参数、优化器状态、专家位置与传输规划之间的关系更加明确,并能够基于实时状态进行处理。

六、ZeRO 优化器与恢复训练行为修复

ZeRO 相关更新主要覆盖 fp16 优化器行为以及恢复训练时的主参数处理。

首先,v0.19.9 修复了 ZeRO Stage 0 的 fp16 优化器未正确遵循 hysteresis 配置的问题。

在 fp16 优化器中,hysteresis 是相关的配置行为之一。本次更新确保 ZeRO Stage 0 的 fp16 优化器能够遵循该设置。

其次,本次更新改进了 ZeRO-1 和 ZeRO-2 在恢复训练过程中的 fp32 master 参数保留逻辑。

具体而言,当恢复训练跳过优化器状态时,ZeRO-1 和 ZeRO-2 需要继续保留 fp32 master 参数。本次修复确保在这种恢复路径下,fp32 master 不会因跳过优化器状态而被错误处理。

这一改动直接面向恢复训练场景。当恢复流程不加载或跳过优化器状态时,模型参数、主参数和优化器数据之间的关系需要得到正确维护。v0.19.9 对这一情形进行了针对性处理。

七、HybridEngine 与 rollout prefill 调整

在 HybridEngine 的 rollout prefill 路径中,本次更新改为仅请求最后一个位置的 logits。

该调整发生在 rollout prefill 阶段,核心变化是 logits 请求范围缩小为最后位置,而不是请求更多位置的 logits。

发布记录中明确指出,这项更新属于相关 rollout prefill 路径的调整。对于使用 HybridEngine 进行 rollout 流程的场景,该变化是 v0.19.9 的一项明确行为更新。

八、通信操作与通信日志修复

通信相关功能在本版本中也获得了多项修复。

首先,对于未被性能分析的通信操作,v0.19.9 会跳过 accelerator 同步。

这一改动针对通信操作与性能分析状态之间的关系:如果某个通信操作不在性能分析范围内,则不再进行对应的 accelerator 同步。

其次,本次修复了通信日志带来的性能回退问题,同时修复了comms_logger.enabled的问题。

通信日志通常用于记录和观察通信行为,但日志机制本身也需要避免对运行造成额外影响。本次更新同时处理了通信日志导致的性能回退,以及启用状态判断相关的问题。

此外,v0.19.9 改为使用 NVTX 的模块级 API,以支持旧版本 NVTX 中的 domain range。

这项变更面向 NVTX 兼容性问题。通过采用模块级 API,NVTX domain range 可以在较旧的 NVTX 版本上正常工作。

从通信与分析工具链角度看,本次更新覆盖了同步行为、日志机制和 NVTX 兼容性三个层面。

九、CPU 后端更新:CPU Adam 增加 RVV 向量化 FP32 内核

CPU 优化器方面,v0.19.9 为 CPU Adam 新增了 RVV 向量化 FP32 内核。

该改动明确针对 CPU Adam 的 FP32 计算路径,并引入 RVV 向量化实现。

与此同时,本次还调整了 CPU accelerator 上的测试策略:跳过仅适用于 accelerator 的 CPU 优化器测试。

这里的调整重点在于测试适配。对于 CPU accelerator 环境,某些仅面向 accelerator 的 CPU 优化器测试将不再执行。

此外,v0.19.9 新增了一套 v1 多 rank CPU 测试套件,并且初始阶段采用手动分发方式。

该测试套件面向多 rank CPU 场景,发布记录特别说明其首先采用 manual dispatch 的方式。这意味着 CPU 多 rank 测试覆盖在本次版本中得到了扩展。

综合来看,CPU 方向的更新包括:

  • • CPU Adam 增加 RVV 向量化 FP32 内核

  • • CPU accelerator 跳过特定优化器测试

  • • 新增 v1 多 rank CPU 测试套件

  • • 初始采用手动分发测试方式

十、标量参数广播修复

本次更新修复了标量参数的 byte-view 广播问题。

该问题与标量参数以 byte-view 形式进行广播有关。v0.19.9 对这一广播路径进行了修正,确保标量参数在对应广播场景下得到正确处理。

虽然该项修改在发布记录中描述较为简洁,但它属于参数通信和数据视图处理链路中的明确修复内容。

十一、文档更新:修复英文与中文资料中的失效链接

文档方面,v0.19.9 完成了多处失效链接修复。

首先,README、文档页面、教程和文章中的 30 个失效链接已被修复。

这批修复覆盖范围包括:

  • • README

  • • 文档页面

  • • 教程

  • • 文章内容

其次,中文 README、中文教程和中文博客中的其余失效链接也已修复。

这意味着本次文档维护不仅覆盖英文资料,也同步处理了中文资料中的链接问题。对于查阅中文 README、教程和博客内容的用户来说,相关访问路径得到了整理。

十二、CI 环境与测试稳定性调整

v0.19.9 还包含多项 CI 和测试基础设施改动。

首先,modal-torch-latest CI 的 Python 版本从 3.10 升级至 3.12。

其次,AWS capacity-fallback 环境中的 Python 版本从 3.10.13 升级至 3.12.14。

这两项变更均是 CI 环境中的 Python 版本更新,分别覆盖 modal-torch-latest 与 AWS capacity-fallback 路径。

此外,为了避免 modal CI 沙箱中出现 300 秒测试超时,本次设置了:

TORCHINDUCTOR_COMPILE_THREADS=1

发布记录明确指出,这一设置用于停止 modal CI sandbox 中出现的 300 秒测试超时问题。通过将 TorchInductor 编译线程设置为 1,相关测试超时问题得到针对性处理。

在 CPU 多测试失败信息解析方面,本次还增加了对 ANSI 颜色代码的清理。

具体来说,在解析失败行之前,会先移除 ANSI 颜色代码。这样可以避免终端颜色控制字符影响失败信息的解析过程。

这些 CI 与测试改动分别涉及:

  • • Python 版本升级

  • • TorchInductor 编译线程设置

  • • 测试超时处理

  • • 失败日志解析

  • • CPU 多 rank 测试覆盖

  • • CPU accelerator 测试适配

十三、按提交时间梳理 v0.19.9 更新节奏

从提交时间看,本次版本的工作在 2026 年 10 月下旬持续推进。

10 月 21 日

当天完成了 v0.19.8 发布后的版本更新,以及对version.txt的 PR 门禁过滤。

同日还包括 CPU accelerator 跳过特定 CPU 优化器测试、输出投影偏置在各 rank 保持完整、AutoTP 处理行并行输出头与嵌入层绑定关系,以及词表并行 logprobs、entropy、KL Triton 内核的加入。

10 月 22 日

当天更新包括 Triton masked softmax 数值稳定性修复、CPU Adam 的 RVV 向量化 FP32 内核,以及 HybridEngine rollout prefill 中仅请求最后位置 logits 的调整。

10 月 23 日

当天主要聚焦文档链接维护,修复 README、文档页面、教程和文章中的 30 个失效链接,同时处理中文 README、中文教程和中文博客中的其余失效链接。

10 月 25 日

当天完成 modal-torch-latest CI 的 Python 版本升级,从 3.10 更新至 3.12。

10 月 26 日

当天增加了基于实时运行时 placement 规划专家传输的 AffineIR 能力。

10 月 27 日

当天完成多项稳定性与环境调整,包括:

  • • 设置TORCHINDUCTOR_COMPILE_THREADS=1处理 modal CI 沙箱测试超时

  • • 修复标量参数 byte-view 广播

  • • 修复 ZeRO Stage 0 fp16 优化器 hysteresis 行为

  • • 清理 ANSI 颜色代码后再解析 CPU 多测试失败行

  • • 升级 AWS capacity-fallback 的 Python 版本

  • • 为被钳制专家激活增加融合 Triton 内核

  • • 修复恢复训练跳过优化器状态时的 ZeRO-1、ZeRO-2 fp32 master 参数保留问题

  • • 未分析通信操作跳过 accelerator 同步

10 月 28 日

当天主要集中于通信与存储映射能力,包括通信日志性能回退修复、comms_logger.enabled修复、NVTX 模块级 API 适配旧版本 NVTX domain range,以及为运行中的 ZeRO 参数和优化器状态增加 AffineIR 仿射存储映射。

10 月 29 日

当天增加 v1 多 rank CPU 测试套件,并采用手动分发方式起步。同时修复 AutoEP 在 ZeRO-3 all-gather 中分区组顺序不一致的问题,修复 AutoEP 与 AutoTP 测试中折叠检查点加载和元数据删除之间的竞态问题,并增加对列切分片段写入 shard 位置的检查。

十四、v0.19.9 更新内容完整汇总

为了便于快速查看,以下汇总本次版本全部变更方向:

  • • 完成 v0.19.8 发布后的版本更新

  • • PR 门禁过滤version.txt

  • • CPU accelerator 跳过仅适用于 accelerator 的 CPU 优化器测试

  • • 输出投影偏置在每个 rank 上保持完整,并作为复制状态处理

  • • AutoTP 保持行并行输出头与嵌入层之间的绑定关系

  • • 新增词表并行 logprobs、entropy、KL 的 Triton 内核

  • • 修复 Triton masked softmax 数值稳定性问题

  • • CPU Adam 新增 RVV 向量化 FP32 内核

  • • HybridEngine rollout prefill 仅请求最后位置 logits

  • • 修复 README、文档页面、教程、文章中的 30 个失效链接

  • • 修复中文 README、教程、博客中的其余失效链接

  • • modal-torch-latest CI 的 Python 从 3.10 升级至 3.12

  • • AffineIR 根据实时运行时 placement 规划专家传输

  • • 设置TORCHINDUCTOR_COMPILE_THREADS=1处理 modal CI 沙箱 300 秒测试超时

  • • 修复标量参数 byte-view 广播

  • • ZeRO Stage 0 fp16 优化器正确遵循 hysteresis

  • • 解析 CPU 多测试失败行前移除 ANSI 颜色代码

  • • AWS capacity-fallback 的 Python 从 3.10.13 升级至 3.12.14

  • • AutoEP 为被钳制专家激活增加融合 Triton 内核

  • • ZeRO-1、ZeRO-2 在恢复跳过优化器状态时保留 fp32 master 参数

  • • 未被性能分析的通信操作跳过 accelerator 同步

  • • 修复通信日志性能回退和comms_logger.enabled问题

  • • 使用 NVTX 模块级 API 支持旧版本 NVTX domain range

  • • AffineIR 为运行中的 ZeRO 参数与优化器状态增加仿射存储映射

  • • 新增 v1 多 rank CPU 测试套件,初始采用手动分发

  • • AutoEP 在 ZeRO-3 all-gather 中保持一致的分区组顺序

  • • 修复 AutoEP 与 AutoTP 测试中的折叠检查点加载和元数据删除竞态

  • • 检查列切分片段在 shard 中的写入位置

十五、结语

代码地址:github.com/deepspeedai/DeepSpeed

DeepSpeed v0.19.9 是一次覆盖面较广的补丁更新。

本次版本一方面补齐了词表并行 logprobs、entropy 和 KL 的 Triton 内核,并修复 masked softmax 数值稳定性问题;另一方面持续完善 AutoTP、AutoEP、ZeRO、AffineIR 等并行训练与专家模型相关能力。

同时,CPU Adam 的 RVV 向量化 FP32 内核、多 rank CPU 测试套件、通信日志修复、NVTX 兼容性处理、恢复训练主参数保留逻辑,以及 CI Python 版本升级和测试超时处理,也使 v0.19.9 的更新不止于单一功能点。

对于正在使用 AutoTP、AutoEP、ZeRO、HybridEngine、Triton、CPU Adam 或多 rank CPU 测试环境的用户,v0.19.9 中的对应改动尤其值得关注。

我们相信人工智能为普通人提供了一种“增强工具”,并致力于分享全方位的AI知识。在这里,您可以找到最新的AI科普文章、工具评测、提升效率的秘籍以及行业洞察。 欢迎关注“福大大架构师每日一题”,发消息可获得面试资料,让AI助力您的未来发展。

特别声明:以上内容(如有图片或视频亦包括在内)为自媒体平台“网易号”用户上传并发布,本平台仅提供信息存储服务。

Notice: The content above (including the pictures and videos if any) is uploaded and posted by a user of NetEase Hao, which is a social media platform and only provides information storage services.

相关推荐
热点推荐
武契奇和欧尔班共同出镜:我们虽然选举失利,但没有丢掉尊严

武契奇和欧尔班共同出镜:我们虽然选举失利,但没有丢掉尊严

澎湃新闻
2026-11-01 16:32:24
北京现代艾尼氪V月销仅135台 总经理李凤刚离职

北京现代艾尼氪V月销仅135台 总经理李凤刚离职

爱卡汽车
2026-11-01 13:41:29
刚刚,胡歌公布病情!冲击很大,很痛苦

刚刚,胡歌公布病情!冲击很大,很痛苦

小椰的奶奶
2026-11-02 14:32:00
挖到宝了!北京丽泽商务区地下7米,藏着三艘1800年前的官方大船

挖到宝了!北京丽泽商务区地下7米,藏着三艘1800年前的官方大船

别人都叫我阿腈
2026-11-02 06:02:08
WTT冠军赛:男单签位公布!王楚钦再战劲敌,梁靖崑阻击世界冠军

WTT冠军赛:男单签位公布!王楚钦再战劲敌,梁靖崑阻击世界冠军

全言作品
2026-11-02 03:42:43
现场画面:高市早苗屡次引导外国领导人向日本国旗鞠躬,英法德意领导人均被套路,孟加拉国总理未中招,特朗普直接无视

现场画面:高市早苗屡次引导外国领导人向日本国旗鞠躬,英法德意领导人均被套路,孟加拉国总理未中招,特朗普直接无视

极目新闻
2026-11-01 13:48:10
心都凉了!查尔斯国王遭遇“窘迫”,卡米拉王后头也不回的离场

心都凉了!查尔斯国王遭遇“窘迫”,卡米拉王后头也不回的离场

蹲坑看世界
2026-11-02 16:19:11
日媒反思:为什么中国不断诞生DeepSeek、Kimi,日本却越来越难?

日媒反思:为什么中国不断诞生DeepSeek、Kimi,日本却越来越难?

互联鱼
2026-11-01 06:30:42
弟弟举报哥哥“高考冒名顶替案”最新进展:经湖南高院批准,案件审理延期三个月;湘潭中院:涉及全国多个省市及多个部门,社会影响重大

弟弟举报哥哥“高考冒名顶替案”最新进展:经湖南高院批准,案件审理延期三个月;湘潭中院:涉及全国多个省市及多个部门,社会影响重大

犀利强哥
2026-11-01 19:43:15
6战4胜状态拉满!多队想抢欧文!

6战4胜状态拉满!多队想抢欧文!

柚子说球
2026-11-02 11:32:14
场均仅8.1分!球队收获7连胜,他却高兴不起来!天赋在CBA顶级,却白白浪费4年时间

场均仅8.1分!球队收获7连胜,他却高兴不起来!天赋在CBA顶级,却白白浪费4年时间

后仰跳投绝杀
2026-11-02 19:06:30
随着日本1-0绝杀!美国队0-3惨败,U17女足世界杯四强诞生2席!

随着日本1-0绝杀!美国队0-3惨败,U17女足世界杯四强诞生2席!

大秦壁虎白话体育
2026-11-02 01:28:17
西方思路失灵!法国在暴乱之后调转方向,开始借鉴东方治理逻辑,差别太大

西方思路失灵!法国在暴乱之后调转方向,开始借鉴东方治理逻辑,差别太大

步论天下事
2026-10-31 10:45:11
杨尚昆担任国家主席时,为什么没有入选政治局常委?

杨尚昆担任国家主席时,为什么没有入选政治局常委?

【历史客栈】
2026-10-29 16:01:29
广州白云山索道将于11月3日起停运

广州白云山索道将于11月3日起停运

南方都市报
2026-11-02 15:28:35
人情薄如纸,河南女子在浙江被朋友做局欠下220万,债务逾期后,亲戚朋友全部远离,如今无家可归

人情薄如纸,河南女子在浙江被朋友做局欠下220万,债务逾期后,亲戚朋友全部远离,如今无家可归

捣蛋窝
2026-10-20 08:56:07
宅基地大局已定:若不出意外,2027年农村宅基地,或将迎3大变化

宅基地大局已定:若不出意外,2027年农村宅基地,或将迎3大变化

兴趣知识
2026-11-02 13:39:54
银行员工说漏嘴:存款过万办纸质存单的人,都在交一种隐形税

银行员工说漏嘴:存款过万办纸质存单的人,都在交一种隐形税

时尚的弄潮
2026-10-18 05:10:07
春晚台词揪出潜伏间谍,离世26年赵丽蓉,为何年年被央媒缅怀

春晚台词揪出潜伏间谍,离世26年赵丽蓉,为何年年被央媒缅怀

墨策史
2026-11-01 16:51:36
菲律宾传来消息,杜特尔特案出现了意想不到的变数!

菲律宾传来消息,杜特尔特案出现了意想不到的变数!

故事终将光明磊落
2026-11-02 11:34:29
2026-11-02 19:48:49
moonfdd incentive-icons
moonfdd
福大大架构师每日一题
1550文章数 83关注度
往期回顾 全部

科技要闻

连AI都有领导了!DeepSeek让它们自己抢活干

头条要闻

中共中央批准开除郭永航党籍:曾任广州市委书记等职

头条要闻

中共中央批准开除郭永航党籍:曾任广州市委书记等职

体育要闻

24岁老秀:说最狂的话,打最勇的球

娱乐要闻

宋慧乔首次公开离婚真相,原因心酸

财经要闻

在座各位AI,都是“垃圾”

汽车要闻

百万超跑同款悬架? 第二代海豹07野心不小

态度原创

健康
家居
旅游
游戏
本地

查不出猝死风险?可能是项目没做对

家居要闻

2026建博会(广州) 公装联探展交流活动

旅游要闻

秋假5天慢遛娃|赴台儿庄古城,把秋日浪漫与成长装进旅途!

武侠游戏请玩家上恋综?天龙归来这操作,看完我想问:还有名额吗

本地新闻

十月奔赴迪庆,解锁梅里日照金山

无障碍浏览 进入关怀版