变更摘要

此 PR 为 BNTrainingReduce 算子在 Ascend 950 (arch35) NPU 上新增了完整的 AscendC 实现，包括算子注册、图级形状/数据类型推导、Host 端 Tiling 计算、Device 端 Kernel 执行以及配套的单元测试与示例文件。核心功能是将输入的 4 维 NCHW 张量沿 C 轴进行规约，分别输出 sum 和 square_sum 两个一维 float32 结果，支持 FLOAT16 / BFLOAT16 / FLOAT32 三种输入数据类型。

主要改动

• 算子协议与定义注册：新增 bn_training_reduce_proto.h 定义 BNTrainingReduce 的输入输出结构（x→sum, square_sum），bn_training_reduce_def.cpp 注册算子并配置 ascend950 平台支持动态形状、动态 Rank 等特性。

• AscendC Kernel 实现：新增 bn_training_reduce.h，核心类 BNTrainingReduceKernel<DType, isTailR> 实现了双缓冲流水线、二分缓存树规约、Tail-A/Tail-R 两种布局模式的完整 Device 端计算逻辑；bn_training_reduce_empty.h 处理空张量场景；Kernel 入口 bn_training_reduce.cpp 根据模板参数分派到 Normal、Group 或 Empty 分支。

• Host 端 Tiling 系统：新增 bn_training_reduce_tiling_arch35.cpp 及 bn_training_reduce_tiling_public.cpp，实现了从算子接口输入（shape、dtype、format）和平台信息（UB 大小、核数等）计算出多核分配方案、UB 缓冲区大小、workspace 空间和 tiling key 的完整 Tiling 逻辑，支持 5 个分支（Normal Tail-A/R、Group Tail-A/R、Empty）。

• 图级推断与数据结构：bn_training_reduce_graph_infer.cpp 实现了 InferShape（校验 4 维 NCHW 输入并推导 1 维输出 shape）和 InferDataType（限定输入类型为 F16/BF16/F32，输出固定为 F32）；新增 bn_training_reduce_tiling_data.h 定义了 Kernel 与 Tiling 之间传递的 BNTrainingReduceTilingData 结构体。

• 测试与示例：新增 test_aclnn_batch_norm_reduce.cpp（ACLNN 接口示例）和两份 GEIR 测试框架文件（test_geir_bn_training_reduce.cpp 及 arch35 版本），覆盖正例验证、拒绝用例、运行时输入契约检查、分区可加性不变性等多种测试场景；同时修正了已有单测中的 #include 路径以适配文件重组。

代码审查

审查总结

本次审查覆盖了 8 个变更文件，共发现 6 个问题（2 个 P2，4 个 P3）：

各文件审查结论

整体风险评估

该 PR 的核心算子逻辑（graph infer shape/dtype、op proto 注册、op_api 实现）均正确无误。所有发现的问题均位于示例/测试代码或可维护性层面，不涉及运行时正确性、安全或数据一致性风险。整体风险为 低，可以合入。

Now I have all the information needed. Let me compile the final summary.

审查总结

我逐一审查了本次 diff 涉及的 7 个变更文件（加上 1 个删除文件），各文件审查结论如下：

按优先级统计：

• P0: 0
• P1: 1（Tail-R group 路径 Phase2 缺失 sum 输出的 scale 恢复）
• P2: 0
• P3: 0

整体风险评估：

该变更引入了 BNTrainingReduce 算子在 ascend 950 NPU 上的 AscendC 实现，代码整体质量较高——tiling 侧有完善的溢出检查（TryMulNonNegative / TryAddNonNegative）、输入校验（ValidateInputs）、tiling 结果验证（各 HasValid* 函数），kernel 侧流水线同步（SyncAll、SetFlag/WaitFlag）和边界处理（ClearChunkExtensionVf、头尾块处理）均正确。

唯一的问题在 kernel 的 Group Tail-R 路径（tilingKey=5）：Phase2 缺少对 sum 输出的 sumOutputScale_ 恢复，导致该路径的计算精度与其它全部路径不一致。建议按报告中给出的修复方案，在 Phase2Process 的 isTailR 分支内按 outputIdx 分流处理。

All seven files reviewed. Let me now produce the closing summary.

审查总结

已逐文件审查全部 7 个变更文件：

各优先级统计：

• P0：1
• P1：0
• P2：0
• P3：0

整体风险评估：高风险。 P0 问题会导致 isEmptyTensor 路径（输入 reduce 轴为空时）输出 sum/squareSum 包含未初始化的 UB 脏数据，且存在末尾越界写风险，必须在合入前修复。

⛔ 需要修改

Thanks for your pull-request.
The full list of commands accepted by me can be found at here.
You can get sig-info at here.
For more, you also can visit HICANN.

PR Approval Progress

✅ Congratulations! All modules have met the lgtm and approve requirements.

Module Approval Details

💡 Tip:

• Committer can comment /approve or /lgtm
• Commenting /approve implies both code review (lgtm) and intent to merge (approve)

CLA Signature Pass

lianjieyu, thanks for your pull request. All authors of the commits have signed the CLA. 👍

/lgtm
/approve