Open
[Requirement|需求建议]: 新增TextGatedResidual算子 #7217
kdy18482276080Add on  1 day ago
kdy18482276080Collaborators
1 day ago Create

Background(背景信息)

Hyper-Connection 的前向读融合包含分组 RMSNorm、低秩 down/SiLU/up 投影、sigmoid 混合及可选 inject。使用小算子链会产生中间张量读写和多次 kernel 下发,需要在 ops-transformer 中新增 TextGatedResidual 融合算子。

Benefit / Necessity(价值/作用)

减少中间数据搬运和算子调度开销,提供统一 ACLNN 与 Torch 调用接口,并可保存独立反向计算需要的缓存。不在本需求中承诺尚未验证的整网收益。

Design(设计方案)

  • 算子分类:mhc,目录 mhc/text_gated_residual。
  • 融合分组 RMSNorm、down/SiLU/up、sigmoid 混合与可选 inject;gamma 内部使用 1+gamma。
  • 支持 FP16/BF16,适配 A2/A5;needBackward 控制 invRms、downPreAct、upPreAct 缓存输出。
  • 分别交付 kernel/host/ACLNN、Torch 封装和接口文档。
  • 网络代表形状:R=320、C=4、每通道 H=2560、D=10240,覆盖小 token 数及大 token 数场景。
  • 通过设备构建、代表用例精度、host/op_api UT 及模型性能对比验证;完整反向算子不属于以下三个前向交付 PR 的范围。

关联 PR

likedislike
Kkdy18482276080Collaborators
1 day ago Added label: requirement
Kkdy18482276080Collaborators
1 day ago Assign kdy18482276080 as responsible person
周云飞
周云飞Collaborators
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 场景的性能对比,同时衡量融合资源开销和缓存成本。

likedislike