已关闭
[Requirement|需求建议]: 新增swiglu_group_grad算子 #4424
dong-yanrong创建于  7月29日关闭于  7月30日
dong-yanrong成员
7月29日 创建

Thanks for sending an requirement! Please fill in the following template to help quickly solve your problem.

Backgroud(背景信息)

swiglu_group_grad 用于 MoE FFN 前向训练时,反向需要计算 ∂L/∂x,并在使用 TopK 权重融合时同时回传 ∂L/∂weight。前向涉及 SwiGLU、Clamp(gate 单侧 / up 双侧)、TopK 权重乘等多种非线性操作,反向穿过这些操作的链式法则,无法采用通用 elemwise 算子拼接得到等价梯度(Clamp 的 mask 与 SiLU 的导数耦合度高),需要一个专用融合反向算子。

Origin(信息来源)

Ascend950

Benefit / Necessity (价值/作用)

Design(设计方案)

likedislike
Ddong-yanrong成员
7月29日 添加了label:requirement
dong-yanrong成员
7月29日 评论:

assign

likedislike
dong-yanrong成员
7月29日 评论:

/assign

likedislike
CANN-robotCANN-robot成员
7月29日 将 dong-yanrong 设为负责人
CANN-robotCANN-robot成员
7月30日 关闭了 issue
CANN-robotCANN-robot成员
7月30日 添加了label:resolved
sinobrsinobr
7月30日 关联了pull request:修复 SwigluGroupGrad 算子 Torch Extension 目录结构