From 19f15bf465feb840eac41f6bc40181174358c8c3 Mon Sep 17 00:00:00 2001
From: wangyang519 <wangyang519@huawei.com>
Date: Fri, 22 Aug 2025 15:16:46 +0800
Subject: [PATCH] fix code check of ATB for coherenorm

---
 .../norm/coherelayernorm/op_kernel/cohere_layer_norm.cpp        | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/src/kernels/kernels/norm/coherelayernorm/op_kernel/cohere_layer_norm.cpp b/src/kernels/kernels/norm/coherelayernorm/op_kernel/cohere_layer_norm.cpp
index 1104b540..2ca3e8ae 100644
--- a/src/kernels/kernels/norm/coherelayernorm/op_kernel/cohere_layer_norm.cpp
+++ b/src/kernels/kernels/norm/coherelayernorm/op_kernel/cohere_layer_norm.cpp
@@ -107,7 +107,7 @@ private:
 
     __aicore__ inline void ProcessSingleRows()
     {
-        for (uint32_t rowRepeatIndex = 0; rowRepeatIndex < rowRepeats_; rowRepeatIndex++) {
+        for (uint64_t rowRepeatIndex = 0; rowRepeatIndex < rowRepeats_; rowRepeatIndex++) {
             uint64_t rowOffset = rowRepeatIndex * numColumns_;
             float mean = ComputeMean(rowOffset);
             float variance = ComputeSquareSum(rowOffset, mean);
-- 
Gitee