Open
[Requirement|需求建议]: 新增TextGatedResidual算子 #7217
kdy18482276080Add on 1 day ago
1 day ago Added label: requirement
1 day ago Assign kdy18482276080 as responsible person
周云飞
1 day ago Comments:
1 day ago Comments:
问题说明
TextGatedResidual 属于新增融合算子的工程需求。按描述,融合 RMSNorm、低秩投影和门控混合有望减少中间读写与 kernel 下发,但缓存输出会增加显存占用,收益需实测,不能直接推断整网加速。正文关联的三个 PR 分别面向算子、Torch 接口和文档,其实际覆盖及合入状态尚未确认。
处理建议
建议围绕 mhc/text_gated_residual 统一 ACLNN、Torch 与文档中的计算语义,明确 1+gamma、可选 inject,以及 needBackward 对三个缓存输出的控制;完整反向算子保持在本次范围外。
验证重点是 A2/A5 上 FP16/BF16 的构建与精度、host/op_api 单测,以及代表形状 R=320、C=4、H=2560、D=10240 下大小 token 场景的性能对比,同时衡量融合资源开销和缓存成本。


Background(背景信息)
Hyper-Connection 的前向读融合包含分组 RMSNorm、低秩 down/SiLU/up 投影、sigmoid 混合及可选 inject。使用小算子链会产生中间张量读写和多次 kernel 下发,需要在 ops-transformer 中新增 TextGatedResidual 融合算子。
Benefit / Necessity(价值/作用)
减少中间数据搬运和算子调度开销,提供统一 ACLNN 与 Torch 调用接口,并可保存独立反向计算需要的缓存。不在本需求中承诺尚未验证的整网收益。
Design(设计方案)
关联 PR