已关闭
[Requirement|需求建议]:下一代支持GNTrainingReduce #6170
Sun创建于  21 天前关闭于  17 天前
Sun
Sun成员
21 天前 创建

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

Backgroud(背景信息)

GNTrainingReduce 负责求 GroupNorm 训练阶段所需的分组一阶/二阶统计量(Σx 与 Σx²):
输入 x 为 rank-4(NCHW 或 NHWC)、dtype 为 fp16/fp32,属性 num_groups = G;
按组归约后输出两个 fp32 张量。

该算子此前的支持范围不包含 A5,需要在 A5(Ascend950 / arch35) 上补齐实现,
使其可在新硬件上正常编译、执行。

Origin(信息来源)

CANN

Benefit / Necessity (价值/作用)

  • 在 A5 上获得可用的 GNTrainingReduce 算子,支撑 GroupNorm 相关训练场景

Design(设计方案)

likedislike
SunSun成员
21 天前 添加了label:requirement
CANN-robotCANN-robot成员
17 天前 关闭了 issue
CANN-robotCANN-robot成员
17 天前 添加了label:resolved