问题描述

在 norm/rms_norm/op_kernel/arch35/rms_norm_regbase_split_d.h 的 ComputeFormer 函数中，stage2 处理非整尾块逻辑时，level1 += 1 和 ComputeMultiLevelReduce 在 line 182-183 无条件执行，缺少 if (tail > 0) 保护。

触发条件

当 colTail 恰好被 ubFactor 整除时（tail = colTail % ubFactor == 0），stage2 的两个 if/else-if 分支均不进入，ComputeSum 不执行，level1Local[level1] 无有效数据。

影响

• level1 += 1 无条件执行，向归约树写入一个空槽位（初始化为 0.0）
• ComputeMultiLevelReduce 使用错误的 level1 计数执行，导致 level2/level3 读取位置错位
• 最终 ComputeMultiLevelRstd 计算出的 rstd 值不正确

对比参考

norm/add_rms_norm_quant/op_kernel/arch35/add_rms_norm_quant_regbase_split_reduce.h 的 line 288 已正确使用 if (tail > 0) 包裹了相同逻辑。

修复方案

将 line 182-183 用 if (tail > 0) 包裹：

if (tail > 0) {
    level1 += 1;
    ComputeMultiLevelReduce(level1Local, level2Local, level3Local, level1, level2, level3);
}