Skip to content

Commit

Permalink
format
Browse files Browse the repository at this point in the history
Signed-off-by: Alexandre Eichenberger <alexe@us.ibm.com>
  • Loading branch information
AlexandreEichenberger committed Sep 10, 2024
1 parent b50bd6a commit 3d376cf
Showing 1 changed file with 6 additions and 6 deletions.
12 changes: 6 additions & 6 deletions src/Conversion/KrnlToAffine/KrnlMatmul.cpp
Original file line number Diff line number Diff line change
Expand Up @@ -299,10 +299,10 @@ class KrnlMatmulLowering : public ConversionPattern {
// For i, j loops.
LiteralIndexExpr zeroIE(0);
Value jSaved;
createAffine.forLoopIE(zeroIE, I, 1,
createAffine.forLoopIE(zeroIE, I, 1,
[&](AffineBuilderKrnlMem &createAffine, ValueRange loopInd) {
Value i = loopInd[0];
createAffine.forLoopIE(zeroIE, J, 1,
createAffine.forLoopIE(zeroIE, J, 1,
[&](AffineBuilderKrnlMem &createAffine, ValueRange loopInd) {
MathBuilder createMath(createAffine);
Value j = loopInd[0];
Expand All @@ -314,7 +314,7 @@ class KrnlMatmulLowering : public ConversionPattern {
// TTmpC() = affine_load(C, cAccess);
createAffine.store(initVal, TmpC, tmpCAccess);
// Sum over k.
createAffine.forLoopIE(zeroIE, K, 1,
createAffine.forLoopIE(zeroIE, K, 1,
[&](AffineBuilderKrnlMem &createAffine,
ValueRange loopInd) {
MathBuilder createMath(createAffine);
Expand Down Expand Up @@ -383,7 +383,7 @@ class KrnlMatmulLowering : public ConversionPattern {
LiteralIndexExpr zeroIE(0);
Value iZero = create.math.constantIndex(0);

create.affineKMem.forLoopIE(zeroIE, K, VL,
create.affineKMem.forLoopIE(zeroIE, K, VL,
[&](AffineBuilderKrnlMem &createAffine, ValueRange loopInd) {
MultiDialectBuilder<MathBuilder, VectorBuilder> create(createAffine);
Value k = loopInd[0];
Expand Down Expand Up @@ -465,7 +465,7 @@ class KrnlMatmulLowering : public ConversionPattern {
LiteralIndexExpr zeroIE(0);
Value iZero = create.math.constantIndex(0);

createAffine.forLoopIE(zeroIE, I, 1,
createAffine.forLoopIE(zeroIE, I, 1,
[&](AffineBuilderKrnlMem &createAffine, ValueRange loopInd) {
MultiDialectBuilder<MathBuilder, VectorBuilder> create(createAffine);
Value i = loopInd[0];
Expand All @@ -475,7 +475,7 @@ class KrnlMatmulLowering : public ConversionPattern {
Value tmpCAccess = (unrollFactor > 1) ? i : zeroIE.getValue();
createAffine.store(initVal, TmpC, tmpCAccess);
// Sum over k.
createAffine.forLoopIE(zeroIE, K, 1,
createAffine.forLoopIE(zeroIE, K, 1,
[&](AffineBuilderKrnlMem &createAffine, ValueRange loopInd) {
MultiDialectBuilder<MathBuilder, VectorBuilder> create(
createAffine);
Expand Down

0 comments on commit 3d376cf

Please sign in to comment.