Prhub

#6269 [ci] chore: bump trtllm CI image to 1.3.0rc14

原始 PR 作者 Superjomn 合并时间 2026-05-12 09:37 文件变更 3 提交数 7 评论 6 代码增减 +36 / -13

执行摘要

升级 TRT-LLM CI 镜像至 rc14 并清理临时修复

PR 关联 #6262 升级了 TRT-LLM 基础镜像至 rc14,新镜像已修复 mbridge 问题,因此需要同步更新 CI 工作流中的镜像引用并清理临时 workaround。同时,在 #6275 合并后 trtllm_async_unit_tests 被移除,需要恢复以保持测试覆盖。

值得阅读以了解 CI 维护的最佳实践:升级基础依赖时同步清理临时修复,并在 reviewer 反馈下恢复测试覆盖。commit 历史展示了如何处理 merge 冲突。

讨论亮点

Reviewer hchings 提出(1)恢复被 #6275 删除的 fully_async unittest job;(2)由于新镜像已修复 mbridge 问题,移除所有对应的 WAR pip 安装行。作者在后续 commit 中全部采纳并解决。

实现拆解

  1. 更新 docker/Dockerfile.stable.trtllm:将 cccl 包含路径从硬编码 x86_64-linux 改为通过 uname -m 动态获取架构,使 Dockerfile 同时支持 x86_64 和 aarch64 构建。

  2. 更新两个 CI 工作流文件中的 IMAGE 环境变量:e2e_fully_async_policy_trtllm.ymle2e_ppo_grpo_trainer_trtllm.yml 中的镜像引用从 trtllm1.3.0rc13-20250504 改为 trtllm1.3.0rc14,采用浮动标签。

  3. 清理临时修复:在 e2e_ppo_grpo_trainer_trtllm.yml 中删除三处 pip install mbridge 的 WAR 步骤;在 e2e_fully_async_policy_trtllm.yml 中删除一处 WAR 步骤。

  4. 恢复测试覆盖:在 e2e_fully_async_policy_trtllm.yml 中重新添加 trtllm_async_unit_tests job(从 #6275 中移除),运行 test_trtllm_abort.py 等异步单元测试;同时在 e2e_ppo_grpo_trainer_trtllm.yml 中移除重复的 abort 测试运行。

文件 模块 状态 重要度
.github/workflows/e2e_fully_async_policy_trtllm.yml CI 配置 modified 4.64
.github/workflows/e2e_ppo_grpo_trainer_trtllm.yml CI 配置 modified 3.77
docker/Dockerfile.stable.trtllm Docker modified 2.78

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

评论区精华

恢复 fully_async 单元测试 job 测试

hchings 要求除了镜像升级外,恢复 #6275 中删除的 fully_async unittest job。

结论:作者同意并在后续 commit 中恢复了该 job。 · 已解决

清理 mbridge 临时修复 infra

hchings 指出新镜像已修复 mbridge 问题,应移除所有 WAR pip 安装行。

结论:作者在更新中移除了三个位置的 mbridge 安装。 · 已解决

风险与影响

  1. CI 镜像更新可能导致环境不兼容,但基础镜像已在 #6262 中验证。
  2. 测试 job 的移动和恢复需要确保没有遗漏测试用例,commit 历史显示已正确处理。
  3. Dockerfile 的架构感知路径修改在 x86_64 和 aarch64 上均需验证,但仅影响 cccl 路径,风险可控。

影响范围限于 TRT-LLM 相关的 CI 流程(e2e_ppo_grpo_trainer_trtllm 和 e2e_fully_async_policy_trtllm),包括 3 个工作流文件和 Dockerfile。对用户无直接影响,团队需确保新镜像通过 CI 测试。

CI 镜像升级 测试覆盖恢复 临时修复清理

关联 Issue

未识别关联 Issue

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

完整报告

参与讨论