diff --git a/src/include/blockwise_gemm.hip.hpp b/src/include/blockwise_gemm.hip.hpp
index bdf79540c3..d540dd8f4f 100644
--- a/src/include/blockwise_gemm.hip.hpp
+++ b/src/include/blockwise_gemm.hip.hpp
@@ -385,9 +385,9 @@ struct BlockwiseGemmBlockABlockBThreadCTransANormalBNormalC_v2
         void* a_loc   = (void*)(p_a_block + mMyThreadOffsetA);
         void* b_loc   = (void*)(p_b_block + mMyThreadOffsetB);
         // loop over k
-        int k_chunk = 2;
-#pragma unroll
-        for(index_t k_begin = 0; k_begin < K; k_begin += KPerThreadLoop * k_chunk)
+        int k_chunk = K;
+        //for(index_t k_begin = 0; k_begin < K; k_begin += KPerThreadLoop * k_chunk)
+        index_t k_begin = 0;
         {
 
 #if 0
@@ -417,6 +417,7 @@ struct BlockwiseGemmBlockABlockBThreadCTransANormalBNormalC_v2
             lgkmcnt(1);
             outerProduct4x4(reg_a[0], reg_b[1], reg_c[1], reg_c[3], reg_c[5], reg_c[7]);
             lgkmcnt(0);
+#pragma unroll
             for(int i = 0; i < k_chunk - 1; i++)
             {
                 k = k + 1;