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 1104b540bcd04279d2333d29e590776fb8e5bd9b..2ca3e8ae36772905d3b79b1d68c3fcacc60e96fb 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);