Prhub

#6657 [veomni] fix: add veomni backend docs for ascend

原始 PR 作者 phdddd 合并时间 2026-06-08 19:28 文件变更 1 提交数 1 评论 1 代码增减 +74 / -1

执行摘要

新增 Ascend 上 VeOmni 后端文档

为 Ascend 平台用户提供 VeOmni 后端的官方文档说明,降低使用门槛。PR Body 明确说明目标是 "add veomni backend docs for ascend"。

建议作者根据 Review 建议修正 Router Replay 参数配置路径的说明,确保文档准确无误。

讨论亮点

Gemini Code Assist 机器人在评论中指出:Router Replay 小节中 record_filereplay_file 的配置路径有误,它们实际上属于 actor_rollout_ref.actor.router_replay(使用 RouterReplayConfig),而不是 actor_rollout_ref.actor.veomni.router_replay(仅 mode 属于 veomni)。建议在表格中明确标注完整配置路径以避免混淆。该评论在 PR 合并前未得到人工回复或修改。

实现拆解

  1. docs/ascend_tutorial/feature_support/ascend_backend_features.md### 3. VeOmni 小节下新增了 VeOmni 后端特性文档。
  2. 详细介绍了 VeOmni 是什么(基于 FSDP 的统一 RL 训练后端,专为大模型高效训练设计),并列出 19 个 VeOmni 配置参数的表格,包括 param_offloadoptimizer_offloadfsdp_sizeulysses_parallel_sizeexpert_parallel_sizemixed_precisionenable_full_shardforward_prefetchattn_implementationmoe_implementationcross_entropy_loss_implementationrms_norm_implementation 等。
  3. 增加了 Router Replay 功能的配置说明,包括 record_filereplay_filemode 等参数,并提供了完整的配置示例。
  4. 给出了一个从 run_qwen3_8b_fsdp.sh 脚本启动的完整训练命令示例。
  5. 补充了 VeOmni 的关键特性说明:ZeRO-3 完全分片、序列并行、专家并行、混合精度训练、CPU-offloading、Recompute、FA2/3 SDPA 集成、fused kernel 优化等。
文件 模块 状态 重要度
docs/ascend_tutorial/feature_support/ascend_backend_features.md Ascend 教程 modified 3.41

分析完成后,这里会展示 LLM 生成的相对完整源码片段和详细注释。

评论区精华

Router Replay 参数配置路径错误 正确性

Gemini Code Assist 机器人指出文档中 `record_file` 和 `replay_file` 的配置路径写为 `actor_rollout_ref.actor.veomni.router_replay`,但实际应属于 `actor_rollout_ref.actor.router_replay`(使用 `RouterReplayConfig`),仅 `mode` 属于 `veomni`。建议在表格中标注完整配置路径。

结论:未在 PR 中修复,评论未被采纳或回复。 · unresolved

风险与影响

风险很低。纯文档变更,不涉及代码逻辑修改。主要风险是文档中可能存在的配置路径错误(如 review 指出的 Router Replay 参数归属问题)可能误导用户。

正面影响:为 Ascend 平台用户提供了 VeOmni 后端的官方文档,降低使用门槛,提升用户体验。影响范围仅限于阅读该文档的用户。

文档可能包含配置路径错误

关联 Issue

未识别关联 Issue

当前没有检测到明确关联的 Issue 链接,后续同步到相关引用后会出现在这里。

完整报告

参与讨论