已关闭
[Requirement|需求建议]: L2Normalize Ascend 950 长归约性能与独立精度基准加固 #6229
raoliang_sac创建于  20 天前关闭于  19 天前
raoliang_sac成员
20 天前 创建

Backgroud(背景信息)

L2Normalize Ascend 950 实现的长归约存在串行累加和 Group 输出阶段并行度不足的问题;补偿求和中 Inf-Inf 产生的 NaN 残差还会污染本应归一化为零的有限值位置。独立 Golden 也需要全计算链 FP64 参考值,避免真值提前舍入。

Origin(信息来源)

本算子质量加固、源码审计、性能 profiling 及特殊值定向测试。

Benefit / Necessity (价值/作用)

改善长归约吞吐量,保留 Inf/NaN 传播语义,提升精度基准的独立性;不改变公开接口与产品支持范围。

Design(设计方案)

  • Base FP16 长归约采用 64-lane 补偿累加,对无效 NaN 补偿残差清零,主和保留特殊值传播。
  • Group 第三阶段以分母槽位和归约块分配核间工作,保留跨阶段同步。
  • NumPy Golden 平方、求和、max、sqrt、除法全程 FP64;保留独立 PyTorch 对照,移除未交付的 ACLNN TestSpec。
  • 变更限于 Base/Group arch35 kernel 和本算子 Golden,不修改 Host/Tiling、REG_OP 或公共文件。

关联 PR:https://gitcode.com/cann/ops-nn/pull/11237

likedislike
Rraoliang_sac成员
20 天前 添加了label:requirement
raoliang_sac成员
20 天前 评论:

/assign @raoliang_sac

likedislike
CANN-robotCANN-robot成员
20 天前 将 raoliang_sac 设为负责人
CANN-robotCANN-robot成员
19 天前 关闭了 issue
CANN-robotCANN-robot成员
19 天前 添加了label:resolved