Skip to content

Commit f97bf75

Browse files
author
Peiming Liu
committed
address comments
1 parent 709b06f commit f97bf75

File tree

1 file changed

+5
-1
lines changed

1 file changed

+5
-1
lines changed

mlir/lib/Dialect/SparseTensor/Transforms/Utils/SparseTensorLevel.h

Lines changed: 5 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -14,6 +14,9 @@
1414
namespace mlir {
1515
namespace sparse_tensor {
1616

17+
/// The base class for all types of sparse tensor levels. It provides interface
18+
/// to query the loop range (see `peekRangeAt`) and look up the coordinates (see
19+
/// `peekCrdAt`).
1720
class SparseTensorLevel {
1821
SparseTensorLevel(SparseTensorLevel &&) = delete;
1922
SparseTensorLevel(const SparseTensorLevel &) = delete;
@@ -89,8 +92,9 @@ class SparseIterator {
8992
virtual ~SparseIterator() = default;
9093

9194
Value getCrd() const { return crd; }
92-
9395
ValueRange getItVals() const { return itVals; };
96+
97+
// Sets the iterate to the specified position.
9498
void seek(ValueRange vals) {
9599
assert(vals.size() == itVals.size());
96100
std::copy(vals.begin(), vals.end(), itVals.begin());

0 commit comments

Comments
 (0)