已关闭
[Requirement|需求建议]: 升级 INTrainingReduceV2、SGD、NormalizeBBox golden TestSpec #4679
raoliang_sac创建于  8月10日关闭于  8月10日
raoliang_sac成员
8月10日 创建

Backgroud(背景信息)

INTrainingReduceV2、SGD 和 NormalizeBBox 是当前准备合入的三个新算子。为满足新的 golden 规范,需要将三个算子的 tests/assets/golden.py 升级为 TestSpec 多通路形式,并提供可用于 kernel/GEIR 验证及第三方交叉检查的统一参考实现。

本需求仅覆盖本次三个 golden 文件的规范升级。三个算子均明确不支持 ACLNN 接口,也不新增 ACLNN 或 e2e 通路。

Origin(信息来源)

ops-nn 三算子开发需求及 golden 新规范适配要求。

Benefit / Necessity (价值/作用)

  • 统一 kernel 与 GEIR 的 golden 注册和计算核心。
  • 支持 Promote 感知的高精度 true-value 与独立 Torch composition 交叉检查。
  • 明确各算子的 dtype、运算顺序及边界语义,降低后续精度验证偏差。
  • 在升级 TestSpec 的同时保留旧加载器兼容入口,避免已有测试用例失效。

Design(设计方案)

  • 为三个算子分别增加 snake_case __spec__ 注册,kernel 与 GEIR 复用同一 Spec。
  • 配置 Torch third-party composition 和 cross_check 容差。
  • INTrainingReduceV2 采用 FP64 true-value 处理大归约;SGD 对齐 arch35 DAG 的 Nesterov 运算顺序并保持 IEEE/动量写回语义;NormalizeBBox 对齐 kernel dtype cast 链并支持 batch=0。
  • 保留 __golden__ 兼容入口,并与 TestSpec 共用计算核心。
  • 不注册 ACLNN 或 e2e 通路。

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

likedislike
Rraoliang_sac成员
8月10日 添加了label:requirement
yuning_chenyuning_chen成员
8月10日 将 raoliang_sac 设为负责人
CANN-robotCANN-robot成员
8月10日 关闭了 issue
CANN-robotCANN-robot成员
8月10日 添加了label:resolved