address comments

Author: Peiming Liu

mlir/include/mlir/Dialect/SparseTensor/IR/SparseTensor.h

@@ -58,7 +58,9 @@ struct COOSegment {
   }
 };
 
-/// A simple wrapper to encode a bitset of defined (at most 64) levels.
+/// A simple wrapper to encode a bitset of (at most 64) levels, currently used
+/// by `sparse_tensor.iterate` operation for the set of levels on which the
+/// coordinates should be loaded.
 class LevelSet {
   uint64_t bits = 0;
 

mlir/include/mlir/Dialect/SparseTensor/IR/SparseTensorOps.td

@@ -1526,7 +1526,7 @@ def IterateOp : SparseTensor_Op<"iterate",
       ["getEntrySuccessorOperands"]>,
      SingleBlockImplicitTerminator<"sparse_tensor::YieldOp">]> {
 
-  let summary = "Iterate over a sparse iteration space";
+  let summary = "Iterates over a sparse iteration space";
   let description = [{
       The `sparse_tensor.iterate` operation represents a loop (nest) over
       the provided iteration space extracted from a specific sparse tensor.

mlir/lib/Dialect/SparseTensor/IR/SparseTensorDialect.cpp

@@ -2137,7 +2137,7 @@ parseSparseSpaceLoop(OpAsmParser &parser, OperationState &state,
   SmallVector<OpAsmParser::UnresolvedOperand> spaces;
   SmallVector<OpAsmParser::UnresolvedOperand> initArgs;
 
-  // Parses "%iters, ... in %spaces, ..."
+  // Parse "%iters, ... in %spaces, ..."
   if (parser.parseArgumentList(iterators) || parser.parseKeyword("in") ||
       parser.parseOperandList(spaces))
     return failure();