[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