commit f34e534a16f9cfe328f0a513ef474fd44af29a30
author mariecwhite <mariewhite@google.com> Wed Mar 06 06:55:27 2024 +0000
committer mariecwhite <mariewhite@google.com> Wed Mar 06 18:47:00 2024 +1100
tree d66323a1bb2c12b09602aece5449f1f5a44576ce
parent 4d3c93f7582621ca2aad1dfc55708e09b73a8da7

Replace k with m

runtime/src/iree/builtins/ukernel/arch/arm_64/mmt4d_arm_64_dotprod.c

diff --git a/runtime/src/iree/builtins/ukernel/arch/arm_64/mmt4d_arm_64_dotprod.c b/runtime/src/iree/builtins/ukernel/arch/arm_64/mmt4d_arm_64_dotprod.c
index 89aedc9..165ec35 100644
--- a/runtime/src/iree/builtins/ukernel/arch/arm_64/mmt4d_arm_64_dotprod.c
+++ b/runtime/src/iree/builtins/ukernel/arch/arm_64/mmt4d_arm_64_dotprod.c
@@ -142,11 +142,11 @@
       // 4x8 * 8x8 -> 4x8.
       IREE_UK_UNROLL for (int i = 0; i < 2; i++) {
         IREE_UK_UNROLL for (int j = 0; j < 2; j++) {
-          IREE_UK_UNROLL for (int k = 0; k < 2; k++) {
-            acc[4 * k + j] = vdotq_lane_s32(acc[4 * k + j], rhs[2 * i + j],
-                                            lhs[2 * k + i], 0);
-            acc[4 * k + j + 2] = vdotq_lane_s32(
-                acc[4 * k + j + 2], rhs[2 * i + j], lhs[2 * k + i], 1);
+          IREE_UK_UNROLL for (int m = 0; m < 2; m++) {
+            acc[4 * m + j] = vdotq_lane_s32(acc[4 * m + j], rhs[2 * i + j],
+                                            lhs[2 * m + i], 0);
+            acc[4 * m + j + 2] = vdotq_lane_s32(
+                acc[4 * m + j + 2], rhs[2 * i + j], lhs[2 * m + i], 1);
           }
         }
       }
@@ -154,11 +154,11 @@
       // 8x8 * 8x8 -> 8x8.
       IREE_UK_UNROLL for (int i = 0; i < 2; i++) {
         IREE_UK_UNROLL for (int j = 0; j < 2; j++) {
-          IREE_UK_UNROLL for (int k = 0; k < 2; k++) {
-            acc[8 + 4 * k + j] = vdotq_lane_s32(
-                acc[8 + 4 * k + j], rhs[2 * i + j], lhs[4 + 2 * i + k], 0);
-            acc[10 + 4 * k + j] = vdotq_lane_s32(
-                acc[10 + 4 * k + j], rhs[2 * i + j], lhs[4 + 2 * i + k], 1);
+          IREE_UK_UNROLL for (int m = 0; m < 2; m++) {
+            acc[8 + 4 * m + j] = vdotq_lane_s32(
+                acc[8 + 4 * m + j], rhs[2 * i + j], lhs[4 + 2 * i + m], 0);
+            acc[10 + 4 * m + j] = vdotq_lane_s32(
+                acc[10 + 4 * m + j], rhs[2 * i + j], lhs[4 + 2 * i + m], 1);
           }
         }
       }