[Mlir-commits] [mlir] [mlir][ArmNeon] Update `LowerContractionToSMMLAPattern` to support proper unrolling for k dimension (PR #88591)

Diego Caballero llvmlistbot at llvm.org
Tue Apr 16 08:57:05 PDT 2024


================
@@ -133,8 +133,12 @@ class LowerContractionToSMMLAPattern
       smmlaShape.insert(smmlaShape.begin(), isVecmat ? 1 : 2);
       loopOrder.push_back(2);
     }
+
+    // Keep track of the previous accumulator when tiling over K
----------------
dcaballe wrote:

nit: period at the end of comments per guidelines

https://github.com/llvm/llvm-project/pull/88591


More information about the Mlir-commits mailing list