[Mlir-commits] [mlir] 490c77e - [mlir][vector] Fix typo, NFC.

llvmlistbot at llvm.org llvmlistbot at llvm.org
Tue Dec 27 19:46:53 PST 2022


Author: jacquesguan
Date: 2022-12-28T11:46:45+08:00
New Revision: 490c77e46d8d2ccda36a1ff3983b937b3bf2c772

URL: https://github.com/llvm/llvm-project/commit/490c77e46d8d2ccda36a1ff3983b937b3bf2c772
DIFF: https://github.com/llvm/llvm-project/commit/490c77e46d8d2ccda36a1ff3983b937b3bf2c772.diff

LOG: [mlir][vector] Fix typo, NFC.

Reviewed By: aartbik

Differential Revision: https://reviews.llvm.org/D140681

Added: 
    

Modified: 
    mlir/lib/Dialect/Vector/Transforms/VectorUnroll.cpp

Removed: 
    


################################################################################
diff  --git a/mlir/lib/Dialect/Vector/Transforms/VectorUnroll.cpp b/mlir/lib/Dialect/Vector/Transforms/VectorUnroll.cpp
index c59972a867be9..1b316083258ce 100644
--- a/mlir/lib/Dialect/Vector/Transforms/VectorUnroll.cpp
+++ b/mlir/lib/Dialect/Vector/Transforms/VectorUnroll.cpp
@@ -591,23 +591,23 @@ struct UnrollReductionPattern : public OpRewritePattern<vector::ReductionOp> {
   const vector::UnrollVectorOptions options;
 };
 
-struct UnrollTranposePattern : public OpRewritePattern<vector::TransposeOp> {
-  UnrollTranposePattern(MLIRContext *context,
-                        const vector::UnrollVectorOptions &options,
-                        PatternBenefit benefit = 1)
+struct UnrollTransposePattern : public OpRewritePattern<vector::TransposeOp> {
+  UnrollTransposePattern(MLIRContext *context,
+                         const vector::UnrollVectorOptions &options,
+                         PatternBenefit benefit = 1)
       : OpRewritePattern<vector::TransposeOp>(context, benefit),
         options(options) {}
 
-  LogicalResult matchAndRewrite(vector::TransposeOp tranposeOp,
+  LogicalResult matchAndRewrite(vector::TransposeOp transposeOp,
                                 PatternRewriter &rewriter) const override {
-    if (tranposeOp.getResultType().getRank() == 0)
+    if (transposeOp.getResultType().getRank() == 0)
       return failure();
-    auto targetShape = getTargetShape(options, tranposeOp);
+    auto targetShape = getTargetShape(options, transposeOp);
     if (!targetShape)
       return failure();
-    auto originalVectorType = tranposeOp.getResultType();
+    auto originalVectorType = transposeOp.getResultType();
     SmallVector<int64_t> strides(targetShape->size(), 1);
-    Location loc = tranposeOp.getLoc();
+    Location loc = transposeOp.getLoc();
     ArrayRef<int64_t> originalSize = originalVectorType.getShape();
     SmallVector<int64_t> ratio = *computeShapeRatio(originalSize, *targetShape);
     int64_t sliceCount = computeMaxLinearIndex(ratio);
@@ -615,7 +615,7 @@ struct UnrollTranposePattern : public OpRewritePattern<vector::TransposeOp> {
     Value result = rewriter.create<arith::ConstantOp>(
         loc, originalVectorType, rewriter.getZeroAttr(originalVectorType));
     SmallVector<int64_t> permutation;
-    tranposeOp.getTransp(permutation);
+    transposeOp.getTransp(permutation);
 
     // Stride of the ratios, this gives us the offsets of sliceCount in a basis
     // of multiples of the targetShape.
@@ -631,13 +631,14 @@ struct UnrollTranposePattern : public OpRewritePattern<vector::TransposeOp> {
         permutedShape[indices.value()] = (*targetShape)[indices.index()];
       }
       Value slicedOperand = rewriter.create<vector::ExtractStridedSliceOp>(
-          loc, tranposeOp.getVector(), permutedOffsets, permutedShape, strides);
-      Value tranposedSlice =
+          loc, transposeOp.getVector(), permutedOffsets, permutedShape,
+          strides);
+      Value transposedSlice =
           rewriter.create<vector::TransposeOp>(loc, slicedOperand, permutation);
       result = rewriter.create<vector::InsertStridedSliceOp>(
-          loc, tranposedSlice, result, elementOffsets, strides);
+          loc, transposedSlice, result, elementOffsets, strides);
     }
-    rewriter.replaceOp(tranposeOp, result);
+    rewriter.replaceOp(transposeOp, result);
     return success();
   }
 
@@ -653,5 +654,5 @@ void mlir::vector::populateVectorUnrollPatterns(
   patterns.add<UnrollTransferReadPattern, UnrollTransferWritePattern,
                UnrollContractionPattern, UnrollElementwisePattern,
                UnrollReductionPattern, UnrollMultiReductionPattern,
-               UnrollTranposePattern>(patterns.getContext(), options, benefit);
+               UnrollTransposePattern>(patterns.getContext(), options, benefit);
 }


        


More information about the Mlir-commits mailing list