#3256 e2e: run the DSv4 megatron impl CI on the cudnn DSA backend
原始 PR · 作者 yueming-yuan · 合并时间 2026-09-15 11:28
镜像补 FlashMLA/cudnn-frontend,DSv4 e2e 切到 cuDNN DSA 后端
建议精读,但重点不在测试文件的一行改动,而在 docker/Dockerfile 中新增的安装块:`--no-deps` 与 `ENABLE_CUDA_13` 门控体现了「镜像依赖决定 CI 能力」的现实约束,是理解本仓库 GPU CI 体系的一个好样本。阅读时重点关注三点:一是 flash_mla wheel 的来源与 torch 版本绑定关系(升级 torch 时需要同步重出 wheel);二是 5 个 `register_ci_gate` 指标在切换内核后端后是否需要重新标定基线;三是建议后续给 flash_mla 安装补上与其他 wheel 一致的 `ls | grep -q .` 存在性守卫,避免 wheel 缺失时镜像构建直接硬失败。对于只关心业务逻辑的读者,本 PR 可以略过。