执行摘要
新增 AMD ROCm e2e PPO 训练 CI 工作流
扩大 CI 覆盖至 AMD ROCm 硬件,确保 veril 在 MI300 上的兼容性,与现有 NVIDIA 和 Ascend 工作流对齐。
值得合并以扩展硬件支持;建议确认 runner 就绪并定期更新镜像版本。
无 review 讨论;仅 Gemini 自动评论表示无法生成 review。
扩大 CI 覆盖至 AMD ROCm 硬件,确保 veril 在 MI300 上的兼容性,与现有 NVIDIA 和 Ascend 工作流对齐。
值得合并以扩展硬件支持;建议确认 runner 就绪并定期更新镜像版本。
无 review 讨论;仅 Gemini 自动评论表示无法生成 review。
.github/workflows/ 下新建 e2e_ppo_trainer_megatron_vllm_rocm.yml,定义触发条件(push/PR 到 main/v0.* 分支,指定路径过滤)。e2e_ppo_trainer_fsdp_vllm_rocm(FSDP/FSDP2/DDP,GSM8K 函数奖励测试 + checkpoint 合并)和 e2e_ppo_trainer_megatron-moe-expert-parallel_rocm(Megatron-Bridge 3D 并行 MoE 测试,含 FP8 rollout)。cleanup_rocm 在 if: always() 下运行,移除测试遗留的 checkpoint 文件。verlai/verl:rocm702-vllm020-te210-20260518 镜像,挂载 /dev/kfd、/dev/dri 等 ROCm 设备,以及宿主模型/数据目录。| 文件 | 模块 | 状态 | 重要度 |
|---|---|---|---|
.github/workflows/e2e_ppo_trainer_megatron_vllm_rocm.yml |
CI 工作流 | added | 6.26 |
分析完成后,这里会展示 LLM 生成的相对完整源码片段和详细注释。
Gemini Code Assist 无法为此 PR 生成 review(因文件类型不支持)。
结论:无实际讨论。 · 已解决
仅涉及 CI 基础设施,无代码逻辑变更。风险主要在于 self-hosted ROCm runner 的可用性和维护成本;若 runner 不可用,工作流会挂起或失败。另外,工作流中硬编码的镜像路径和依赖可能需要随版本更新。
对用户无直接影响;对开发流程有正面影响,确保 ROCm 回归测试覆盖;对 CI 团队增加工作流维护负担。
当前没有检测到明确关联的 Issue 链接,后续同步到相关引用后会出现在这里。
参与讨论