diff --git a/mlir/lib/Dialect/SparseTensor/Transforms/SparseTensorRewriting.cpp b/mlir/lib/Dialect/SparseTensor/Transforms/SparseTensorRewriting.cpp index f16d08b86a1a1..a1ab2495f5f7b 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 { 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(loc, collapsed_size, srcSizes[d]); - SmallVector collapsedSizes = {collapsed_size}; + collapseSize = + builder.create(loc, collapseSize, srcSizes[d]); + SmallVector collapsedSizes = {collapseSize}; - ReassociationIndices collapse_indices; + ReassociationIndices collapseIdx; for (Dimension i = 0; i < srcRank; i++) - collapse_indices.push_back(i); - SmallVector collapse_reassociation = { - collapse_indices}; + collapseIdx.push_back(i); + SmallVector collapseReass = {collapseIdx}; SmallVector 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 expand_reassociation = { - expand_indices}; + expandIdx.push_back(i); + SmallVector expandReass = {expandIdx}; SmallVector dstDcvs; - reshapeCvs(builder, loc, expand_reassociation, collapsedSizes, - collapsedDcvs, dstSizes, dstDcvs); + reshapeCvs(builder, loc, expandReass, collapsedSizes, collapsedDcvs, + dstSizes, dstDcvs); auto t = builder.create(loc, v, reduc.front(), dstDcvs); builder.create(loc, t);