Skip to content

Commit eb14f47

Browse files
authored
[mlir][sparse][NFC] fix variable naming convension (llvm#69232)
1 parent fcc13c0 commit eb14f47

File tree

1 file changed

+13
-15
lines changed

1 file changed

+13
-15
lines changed

mlir/lib/Dialect/SparseTensor/Transforms/SparseTensorRewriting.cpp

Lines changed: 13 additions & 15 deletions
Original file line numberDiff line numberDiff line change
@@ -665,29 +665,27 @@ struct TensorReshapeRewriter : public OpRewritePattern<tensor::ReshapeOp> {
665665
srcDcvs.push_back(srcLcvs[lvl]);
666666
}
667667

668-
Value collapsed_size = constantIndex(builder, loc, 1);
668+
Value collapseSize = constantIndex(builder, loc, 1);
669669
for (Dimension d = 0; d < srcRank; d++)
670-
collapsed_size =
671-
builder.create<arith::MulIOp>(loc, collapsed_size, srcSizes[d]);
672-
SmallVector<Value, 1> collapsedSizes = {collapsed_size};
670+
collapseSize =
671+
builder.create<arith::MulIOp>(loc, collapseSize, srcSizes[d]);
672+
SmallVector<Value, 1> collapsedSizes = {collapseSize};
673673

674-
ReassociationIndices collapse_indices;
674+
ReassociationIndices collapseIdx;
675675
for (Dimension i = 0; i < srcRank; i++)
676-
collapse_indices.push_back(i);
677-
SmallVector<ReassociationIndices, 1> collapse_reassociation = {
678-
collapse_indices};
676+
collapseIdx.push_back(i);
677+
SmallVector<ReassociationIndices, 1> collapseReass = {collapseIdx};
679678
SmallVector<Value, 1> collapsedDcvs;
680-
reshapeCvs(builder, loc, collapse_reassociation, srcSizes, srcDcvs,
679+
reshapeCvs(builder, loc, collapseReass, srcSizes, srcDcvs,
681680
collapsedSizes, collapsedDcvs);
682681

683-
ReassociationIndices expand_indices;
682+
ReassociationIndices expandIdx;
684683
for (Dimension i = 0; i < dstTp.getDimRank(); i++)
685-
expand_indices.push_back(i);
686-
SmallVector<ReassociationIndices, 1> expand_reassociation = {
687-
expand_indices};
684+
expandIdx.push_back(i);
685+
SmallVector<ReassociationIndices, 1> expandReass = {expandIdx};
688686
SmallVector<Value> dstDcvs;
689-
reshapeCvs(builder, loc, expand_reassociation, collapsedSizes,
690-
collapsedDcvs, dstSizes, dstDcvs);
687+
reshapeCvs(builder, loc, expandReass, collapsedSizes, collapsedDcvs,
688+
dstSizes, dstDcvs);
691689

692690
auto t = builder.create<InsertOp>(loc, v, reduc.front(), dstDcvs);
693691
builder.create<sparse_tensor::YieldOp>(loc, t);

0 commit comments

Comments
 (0)