Skip to content

Latest commit

 

History

History
26 lines (18 loc) · 4.82 KB

optimization-reference-map.md

File metadata and controls

26 lines (18 loc) · 4.82 KB

优化参考与修改归属

按当前瓶颈选择参考工程,不按厂商或引擎名称排固定优先级。先读实际 HCU 调用链和配方,再比较官方实现、教程与 PR;同名机制必须核对版本、数值和运行条件。无需为一次优化读完全部工程。

从问题选择参考

问题 Wiki 入口 重点比较
并行切分、流水调度、MoE 与训练状态 Megatron LM/Core、Megatron Bridge TP/PP/CP/EP/DP、microbatch、模型配方、参数/优化器/激活生命周期
PyTorch 原生训练、FSDP、compile 与组合优化 TorchTitan、DeepSpeed 分片/预取、ZeRO/offload、编译边界、通信重叠及实际内存峰值;DeepSpeed 同样是跨引擎机制参考
激进的 Megatron 优化和新模型实现 AMD Primus、MindSpeed / MindSpeed-LLM、百度 LoongForge 调度、overlap、重算/offload、融合及补丁基线;借鉴算法,核对设备专用实现
SFT 数据、packing、LoRA、并行与吞吐 MS-SWIFT、LLaMA-Factory、DeepSpeed mask/有效 token、batch 语义、训练后端与显存;速度比较保留学习目标
RL 训练生成协同、权重传输与恢复 verl、slime、AReaL、TitanRL policy version、采样/更新比、角色共卡/分离、同步窗口及恢复语义
rollout 吞吐、KV cache 与训推切换 SGLang、vLLM 实际 backend 的 sleep、权重加载、调度和显存;不能将推理吞吐直接当训练收益
低精度、融合、Attention 与图执行 Transformer Engine、cuDNN Frontend 前后向、累积精度、缓存失效、workspace 与 dispatch,结合 HCU 已有算子/通信库
数据供给、确定性与恢复 Energon及目标引擎数据层 worker 分片、混合、packing、预取及 checkpoint 后的数据连续性

每个入口包含官方仓库、教程、固定源码与 PR 路线。共同内容标准见 Wiki 组织规范,全库入口见 Wiki。未收录的机制继续用 wiki-search-pr 查官方 PR,再读最终实现、调用者与测试;发布、roadmap、open PR 与当前可用能力分别判断。

HCU 修改并不限于算子

根因在哪层,就在对应 HCU 工程修复:训练引擎负责配方、调度和接入;TE / Flash-Train / Primus Turbo 负责各自算子;RCCL、rocSHMEM、DeepEP 等负责通信实现;必要时还可涉及数学库、PyTorch、编译器和 Runtime。具体仓、分支选择和验证要求见 HCU 库联动与修改归属。

例如 HCU RCCL 的已登记源码在 DTK GitLab / dcutoolkit/deeplearing/rccl,使用前通过 HCU-Knowledge 核对当前适用分支;rccl-test 是测试工程,AMD/ROCm RCCL 是外部参考,三者不能混用。内部链接需要对应账号权限,不将内部源码复制进公共 Wiki。

修改前确认部署版本、实际加载库与调用关系;采用任务独立 checkout 和构建目录,保留基线与回退。完成库级、模型集成及受影响的阶段数值验证后,向真实归属仓交付 PR。参考工程中的机制不是新环境默认开关,候选以端到端收益和训练质量验收。