fix(rms_norm): add tail>0 guard for level1 increment and multi-level reduce in stage2-ops-nn-AtomGit

已合并

fix(rms_norm): add tail>0 guard for level1 increment and multi-level reduce in stage2 #5771

raoliang_sac创建于 6月5日从raoliang_sac/ops-nn： fix/rms-norm-tail-zero-guard 合入到 cann/ops-nn： master

fix(rms_norm): add tail>0 guard for level1 increment and multi-level reduce in stage2 #5771

已合并

raoliang_sac创建于 6月5日

从raoliang_sac/ops-nn fix/rms-norm-tail-zero-guard合入到cann/ops-nn master

raoliang_sac成员

6月5日

关联 Issue

Closes #3158

问题描述

在 norm 目录下多个 rms_norm 变体的 split_reduce / split_d kernel 中，stage2 处理非整尾块逻辑时，ComputeSum + level1 += 1 + ComputeMultiLevelReduce 在 if/else-if 分支之后无条件执行，缺少 if (tail > 0) 保护。

当 tail == 0（剩余列数恰好被 ubFactor 整除）时，两个 if/else-if 分支均不进入，ComputeSum 不执行，level1Local[level1] 无有效数据。但 level1 += 1 和 ComputeMultiLevelReduce 仍然执行，向归约树写入空槽位，导致 level2/level3 读取位置错位，最终 rstd 计算结果不正确。

修复方案

将 stage2 之后的 ComputeSum + level1 += 1 + ComputeMultiLevelReduce 用 if (tail > 0) 包裹，与 add_rms_norm_quant_regbase_split_reduce.h line 288 的已有模式保持一致。

变更文件（3 个）

文件	修复位置
`norm/rms_norm/op_kernel/arch35/rms_norm_regbase_split_d.h`	line 182-183
`norm/add_rms_norm_cast/op_kernel/arch35/add_rms_norm_cast_regbase_spilt_reduce.h`	line 259-261
`norm/add_rms_norm_dynamic_quant/op_kernel/arch35/add_rms_norm_dynamic_quant_regbase_split_reduce.h`	line 311-313

变更详情

         }
-        level1 += 1;
-        ComputeMultiLevelReduce(...);
+        if (tail > 0) {
+            level1 += 1;
+            ComputeMultiLevelReduce(...);
+        }
         // Stage3: Cal MasterLoop

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

（感谢 raoliang_sac 的贡献）

R raoliang_sac成员

6月5日创建了 pull request，commit 2c1860b3

R raoliang_sac成员

6月5日关联了issue：[Bug] rms_norm_regbase_split_d.h: tail==0 时 level1 和 ComputeMultiLevelReduce 无条件执行导致 rstd 计算错误

任务名称	状态	日志	下载链接
Compile_Ascend_X86_ubuntu24	✅ SUCCESS	>>>>>	>>>>>
Compile_Ascend_X86_mobile_station	✅ SUCCESS	>>>>>	>>>>>
Compile_Ascend_ARM	✅ SUCCESS	>>>>>	>>>>>
Compile_Ascend_single	❌ FAILED	>>>>>
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_Ascend_X86_mobile_station_ubuntu24	❌ FAILED	>>>>>
Compile_Ascend_X86_950_ubuntu24	✅ SUCCESS	>>>>>	>>>>>
Compile_Ascend_ARM_ubuntu24	✅ SUCCESS	>>>>>	>>>>>
Compile_Ascend_single_ubuntu24	✅ SUCCESS	>>>>>	>>>>>
Compile_Ascend_experimental_ubuntu24	✅ SUCCESS	>>>>>	>>>>>
Compile_Ascend_ARM_950_ubuntu24	✅ SUCCESS	>>>>>	>>>>>
Compile_X86_monitor_910b	✅ SUCCESS	>>>>>	>>>>>
Compile_X86_monitor_910c	✅ SUCCESS	>>>>>	>>>>>
Compile_X86_monitor_950	✅ SUCCESS	>>>>>	>>>>>
Compile_classify	✅ SUCCESS	>>>>>

任务名称	状态	日志	下载链接
codecheck	✅ SUCCESS	>>>>>
antipoison	✅ SUCCESS	>>>>>
codecheck_checkpr	✅ SUCCESS
StaticCheck_codespell_check	❌ FAILED
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	>>>>>

关联 Issue

问题描述

修复方案

变更文件（3 个）

变更详情

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

PR Approval Progress

Module Approval Details

CLA Signature Pass