norm 族 4 个算子的测试侧 golden 在三方比对方面存在问题,影响比对结果的有效性:
norm/bn3_d_training_update
norm/bn3_d_training_reduce_grad
norm/batch_norm_ext2
norm/gn_training_reduce
CANN
对 4 个算子的 tests/assets/golden.py 做规范化,仅改动测试侧,不动算子实现、接口与 tiling:
tests/assets/golden.py
astype(np.float32)
F.batch_norm
Backgroud(背景信息)
norm 族 4 个算子的测试侧 golden 在三方比对方面存在问题,影响比对结果的有效性:
norm/bn3_d_training_updatenorm/bn3_d_training_reduce_gradnorm/batch_norm_ext2norm/gn_training_reduceOrigin(信息来源)
CANN
Benefit / Necessity (价值/作用)
Design(设计方案)
对 4 个算子的
tests/assets/golden.py做规范化,仅改动测试侧,不动算子实现、接口与 tiling:astype(np.float32)与出口按声明 dtype 的窄回,改为按运行时下发的 dtype 取值。F.batch_norm)实现,与 golden 计算核解耦。