From 882c14f34e97dd615d95ef17818f08d2c86d7f9c Mon Sep 17 00:00:00 2001
From: wangyang519 <wangyang519@huawei.com>
Date: Fri, 22 Aug 2025 15:15:31 +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