修复SyncBatchNormGatherStatsFused kernel 使用 SyncAll 但 Tiling 未设置 SetScheduleMode(BATCH

已合并

修复SyncBatchNormGatherStatsFused kernel 使用 SyncAll 但 Tiling 未设置 SetScheduleMode(BATCH_MODE) #7117

Chao-Wu创建于 7月7日从huangxiaochencha/ops-nn： 9.1.0 合入到 cann/ops-nn： 9.1.0

修复SyncBatchNormGatherStatsFused kernel 使用 SyncAll 但 Tiling 未设置 SetScheduleMode(BATCH_MODE) #7117

已合并

Chao-Wu创建于 7月7日

从huangxiaochencha/ops-nn 9.1.0合入到cann/ops-nn 9.1.0

Chao-Wu成员

7月7日

描述

SyncBatchNormGatherStatsFused kernel 使用 SyncAll 但 Tiling 并设置 SetScheduleMode(BATCH_MODE)

关联的Issue

https://gitcode.com/cann/ops-nn/issues/3872

测试

冒烟，测试均通过

文档更新

类型标签

Bug修复
新特性
性能优化
文档更新
其他，请描述：

AI/Agent生成声明

AI辅助编写

Pull Request已成功合入, 合并人@CANN-robot

（感谢 Chao-Wu 的贡献）

Chao-Wu成员

7月7日创建了 pull request，commit e8bbc69b

Chao-Wu成员

7月7日关联了issue：[Bug][arch35][norm] AddLayerNormQuant/SyncBatchNormGatherStatsFused kernel 使用 SyncAll 但 Tiling 未设置 SetScheduleMode(BATCH_MODE)

atomgit-bot

7月7日评论：

变更摘要

本 PR 修复了 SyncBatchNormGatherStatsFused kernel 在 Tiling 阶段遗漏设置调度模式的 Bug。该 kernel 内部使用了 SyncAll 全核同步机制，但对应的 Tiling 实现未调用 SetScheduleMode 将其置为 BATCH_MODE，可能导致调度行为不符合预期。修复方式是在四个 Tiling 子类的 PostTiling() 方法中统一添加 context_->SetScheduleMode(1) 调用。

主要改动

SyncBatchNormGatherStatsFusedCommonTiling::PostTiling 补充调度模式设置：在 sync_batch_norm_gather_stats_fused_common_tiling.cpp 中，SetBlockDim 之后新增 context_->SetScheduleMode(1)，将调度模式显式设为 BATCH_MODE，与 kernel 的 SyncAll 行为对齐。
SyncBatchNormGatherStatsFusedFirstAxisCommonTiling::PostTiling 补充调度模式设置：在 sync_batch_norm_gather_stats_fused_first_axis_common_tiling.cpp 中同步添加相同调用，覆盖 first-axis 分支的通用 Tiling 场景。
SyncBatchNormGatherStatsFusedFirstAxisWorkspaceTiling::PostTiling 补充调度模式设置：在 sync_batch_norm_gather_stats_fused_first_axis_workspace_tiling.cpp 中同步添加相同调用，覆盖 first-axis 分支的 Workspace Tiling 场景。
SyncBatchNormGatherStatsFusedWorkspaceTiling::PostTiling 补充调度模式设置：在 sync_batch_norm_gather_stats_fused_workspace_tiling.cpp 中同步添加相同调用，覆盖通用 Workspace Tiling 场景。

atomgit-bot

7月7日评论：

代码审查

审查总结

变更概述

本 PR 在 SyncBatchNormGatherStatsFused 算子的四个 tiling 变体的 PostTiling() 方法中分别添加了 context_->SetScheduleMode(1) 调用，将调度模式设置为 BatchMode，以确保 kernel 中的 SyncAll() 全核同步正确工作（所有核同时启动，避免死锁或归约结果错误）。

审查结论

文件	审查结果
`sync_batch_norm_gather_stats_fused_common_tiling.cpp`	1 个 P3 建议
`sync_batch_norm_gather_stats_fused_first_axis_common_tiling.cpp`	1 个 P3 建议
`sync_batch_norm_gather_stats_fused_first_axis_workspace_tiling.cpp`	1 个 P3 建议
`sync_batch_norm_gather_stats_fused_workspace_tiling.cpp`	1 个 P3 建议

P0: 0
P1: 0
P2: 0
P3: 4（均为同一建议：检查 SetScheduleMode 返回值，避免静默失败）

整体风险评估：低风险

核心修复逻辑正确且完整——四个 tiling 变体与四个使用 SyncAll 的 kernel 变体一一对应，添加的 SetScheduleMode(1) 调用位置（PostTiling 中 SetBlockDim 之后）与仓库中大多数参考实现一致。唯一可改进之处是返回值未检查，多数同类代码也存在此模式，属于可选的健壮性增强。

⚠️ 已识别出整体风险，但无法提取行内评论，请参考整体评估。

CANN-robot成员

7月7日添加了label：cann-cla/yes

CANN-robot成员

7月7日评论：

CLA Signature Pass

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

CANN-robot成员

7月7日评论：

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

PR Approval Progress

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

Module Approval Details

module	lgtm status	approve status
**	✅ gcw_kUomxQ2l, 唐玮玮 (2/2)	✅ gcw_kUomxQ2l (1/1)

💡 Tip:

Committer can comment /approve or /lgtm

Commenting /approve implies both code review (lgtm) and intent to merge (approve)

CANN-robot成员

7月7日将crystalhu,yangyang016,fanqirui,chenqi317,liubo75,tangweiwei2,gcw_kUomxQ2l,Andy-lb设为评审人

CANN-robot成员

7月7日将gcw_kUomxQ2l,Andy-lb设为审查人

Chao-Wu成员

7月7日评论：

compile

Chao-Wu成员

7月7日 update merge request[project id: 7665709, iid: 7117, commit_id: 23746b27b6c6694dcf745b087f9f7b334d742de5] virtual merging success

CANN-robot成员

7月7日添加了label：ci-pipeline-running

CANN-robot成员

7月7日评论：

流水线任务触发成功
任务链接 [fc68dac933fc4d048475ff4beb6a7f69][流水线指导]

任务名称	状态	日志	下载链接
Compile_Ascend_X86_mobile_station	✅ SUCCESS	>>>>>	>>>>>
Compile_Ascend_ARM	✅ SUCCESS	>>>>>	>>>>>
Compile_Ascend_single	✅ SUCCESS	>>>>>	>>>>>
Compile_Ascend_experimental	✅ SUCCESS	>>>>>	>>>>>
Compile_Ascend_X86_950	✅ SUCCESS	>>>>>	>>>>>
Compile_Ascend_ARM_950	✅ SUCCESS	>>>>>	>>>>>
Compile_Pre	✅ SUCCESS	>>>>>
pre_comment	✅ SUCCESS	>>>>>
Compile_Ascend_X86	✅ SUCCESS	>>>>>	>>>>>
Compile_X86_monitor_910b	✅ SUCCESS	>>>>>	>>>>>
Compile_X86_monitor_910c	✅ SUCCESS	>>>>>	>>>>>
Compile_X86_monitor_950	✅ SUCCESS	>>>>>	>>>>>
Compile_harmony-infer-chs-nn	✅ SUCCESS	>>>>>
UT_Test_ophost	✅ SUCCESS
UT_Test_opapi	✅ SUCCESS
UT_Test_kernel	✅ SUCCESS
UT_Test_opgraph	✅ SUCCESS
PreSmoke_A900	✅ SUCCESS		>>>>>
PreSmoke_ATK_Test_A2	✅ SUCCESS		>>>>>

[2026-07-07 10:15:07] CI执行结束

CANN-robot成员

7月7日评论：

流水线任务触发成功
任务链接 [0aa07f16a06044e58891f6179f3cdecf][流水线指导]

任务名称	状态	日志	下载链接
codecheck	✅ SUCCESS	>>>>>
antipoison	✅ SUCCESS	>>>>>
codecheck_checkpr	✅ SUCCESS
StaticCheck_codespell_check	✅ SUCCESS
StaticCheck_link_validity_check	✅ SUCCESS
StaticCheck_resource_existence_check	✅ SUCCESS
StaticCheck_tag_closed_check	✅ SUCCESS
StaticCheck_markdownlint	✅ SUCCESS
codecheck_style	✅ SUCCESS	>>>>>
codecheck_precommit	⚠️ WARNING	>>>>>	>>>>>
SCA	✅ SUCCESS	>>>>>