Thanks for sending an requirement! Please fill in the following template to help quickly solve your problem.
GNTrainingReduce 负责求 GroupNorm 训练阶段所需的分组一阶/二阶统计量(Σx 与 Σx²): 输入 x 为 rank-4(NCHW 或 NHWC)、dtype 为 fp16/fp32,属性 num_groups = G; 按组归约后输出两个 fp32 张量。
该算子此前的支持范围不包含 A5,需要在 A5(Ascend950 / arch35) 上补齐实现, 使其可在新硬件上正常编译、执行。
CANN
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 (价值/作用)
Design(设计方案)