From 9e3d7c40a537165d6a4bfb14ead882c2b6807ad9 Mon Sep 17 00:00:00 2001
From: Diego Caballero <diegocaballero@google.com>
Date: Fri, 17 Nov 2023 00:17:27 +0100
Subject: [PATCH] Add comment

---
 .../compiler/Codegen/Common/CPU/CPUMaterializeEncodingPass.cpp | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/compiler/src/iree/compiler/Codegen/Common/CPU/CPUMaterializeEncodingPass.cpp b/compiler/src/iree/compiler/Codegen/Common/CPU/CPUMaterializeEncodingPass.cpp
index ffd3f30974aa0..6f3858f591345 100644
--- a/compiler/src/iree/compiler/Codegen/Common/CPU/CPUMaterializeEncodingPass.cpp
+++ b/compiler/src/iree/compiler/Codegen/Common/CPU/CPUMaterializeEncodingPass.cpp
@@ -172,7 +172,8 @@ enumerateMatmulTileX86_64(EncodingUser user, TypeRange elementTypes,
             TileMxNxK{1, 16, 1},  // Truncation of the above.
         };
       } else {
-        // Code generation tile sizes.
+        // Code generation tile sizes. We try to keep certain level of unrolling
+        // on the N dimension when the M dimension is too narrow for it.
         return {
             TileMxNxK{16, 16, 1}, // Aim to use VFMADD* (zmm).
             TileMxNxK{8, 32, 1},  // Truncation of the above.