[Mlir-commits] [mlir] eb14f47 - [mlir][sparse][NFC] fix variable naming convension (#69232)
llvmlistbot at llvm.org
llvmlistbot at llvm.org
Mon Oct 16 10:51:32 PDT 2023
Author: Peiming Liu
Date: 2023-10-16T10:51:28-07:00
New Revision: eb14f47bf1ccfda500ba3c3092d70e269f6f0b56
URL: https://github.com/llvm/llvm-project/commit/eb14f47bf1ccfda500ba3c3092d70e269f6f0b56
DIFF: https://github.com/llvm/llvm-project/commit/eb14f47bf1ccfda500ba3c3092d70e269f6f0b56.diff
LOG: [mlir][sparse][NFC] fix variable naming convension (#69232)
Added:
Modified:
mlir/lib/Dialect/SparseTensor/Transforms/SparseTensorRewriting.cpp
Removed:
################################################################################
diff --git a/mlir/lib/Dialect/SparseTensor/Transforms/SparseTensorRewriting.cpp b/mlir/lib/Dialect/SparseTensor/Transforms/SparseTensorRewriting.cpp
index f16d08b86a1a14f..a1ab2495f5f7b5e 100644
--- a/mlir/lib/Dialect/SparseTensor/Transforms/SparseTensorRewriting.cpp
+++ b/mlir/lib/Dialect/SparseTensor/Transforms/SparseTensorRewriting.cpp
@@ -665,29 +665,27 @@ struct TensorReshapeRewriter : public OpRewritePattern<tensor::ReshapeOp> {
srcDcvs.push_back(srcLcvs[lvl]);
}
- Value collapsed_size = constantIndex(builder, loc, 1);
+ Value collapseSize = constantIndex(builder, loc, 1);
for (Dimension d = 0; d < srcRank; d++)
- collapsed_size =
- builder.create<arith::MulIOp>(loc, collapsed_size, srcSizes[d]);
- SmallVector<Value, 1> collapsedSizes = {collapsed_size};
+ collapseSize =
+ builder.create<arith::MulIOp>(loc, collapseSize, srcSizes[d]);
+ SmallVector<Value, 1> collapsedSizes = {collapseSize};
- ReassociationIndices collapse_indices;
+ ReassociationIndices collapseIdx;
for (Dimension i = 0; i < srcRank; i++)
- collapse_indices.push_back(i);
- SmallVector<ReassociationIndices, 1> collapse_reassociation = {
- collapse_indices};
+ collapseIdx.push_back(i);
+ SmallVector<ReassociationIndices, 1> collapseReass = {collapseIdx};
SmallVector<Value, 1> collapsedDcvs;
- reshapeCvs(builder, loc, collapse_reassociation, srcSizes, srcDcvs,
+ reshapeCvs(builder, loc, collapseReass, srcSizes, srcDcvs,
collapsedSizes, collapsedDcvs);
- ReassociationIndices expand_indices;
+ ReassociationIndices expandIdx;
for (Dimension i = 0; i < dstTp.getDimRank(); i++)
- expand_indices.push_back(i);
- SmallVector<ReassociationIndices, 1> expand_reassociation = {
- expand_indices};
+ expandIdx.push_back(i);
+ SmallVector<ReassociationIndices, 1> expandReass = {expandIdx};
SmallVector<Value> dstDcvs;
- reshapeCvs(builder, loc, expand_reassociation, collapsedSizes,
- collapsedDcvs, dstSizes, dstDcvs);
+ reshapeCvs(builder, loc, expandReass, collapsedSizes, collapsedDcvs,
+ dstSizes, dstDcvs);
auto t = builder.create<InsertOp>(loc, v, reduc.front(), dstDcvs);
builder.create<sparse_tensor::YieldOp>(loc, t);
More information about the Mlir-commits
mailing list