File tree Expand file tree Collapse file tree 1 file changed +5
-1
lines changed
mlir/lib/Dialect/SparseTensor/Transforms/Utils Expand file tree Collapse file tree 1 file changed +5
-1
lines changed Original file line number Diff line number Diff line change 14
14
namespace mlir {
15
15
namespace sparse_tensor {
16
16
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`).
17
20
class SparseTensorLevel {
18
21
SparseTensorLevel (SparseTensorLevel &&) = delete ;
19
22
SparseTensorLevel (const SparseTensorLevel &) = delete ;
@@ -89,8 +92,9 @@ class SparseIterator {
89
92
virtual ~SparseIterator () = default ;
90
93
91
94
Value getCrd () const { return crd; }
92
-
93
95
ValueRange getItVals () const { return itVals; };
96
+
97
+ // Sets the iterate to the specified position.
94
98
void seek (ValueRange vals) {
95
99
assert (vals.size () == itVals.size ());
96
100
std::copy (vals.begin (), vals.end (), itVals.begin ());
You can’t perform that action at this time.
0 commit comments