File tree Expand file tree Collapse file tree 1 file changed +4
-2
lines changed
mlir/lib/Dialect/SparseTensor/Transforms Expand file tree Collapse file tree 1 file changed +4
-2
lines changed Original file line number Diff line number Diff line change @@ -38,7 +38,8 @@ static std::optional<LogicalResult>
38
38
convertIteratorType (IteratorType itTp, SmallVectorImpl<Type> &fields) {
39
39
// The actually Iterator Values (that are updated every iteration).
40
40
auto idxTp = IndexType::get (itTp.getContext ());
41
- // TODO: This assumes there is no batch dimenstion in the sparse tensor.
41
+ // TODO: handle batch dimension.
42
+ assert (itTp.getEncoding ().getBatchLvlRank () == 0 );
42
43
if (!itTp.isUnique ()) {
43
44
// Segment high for non-unqiue iterator.
44
45
fields.push_back (idxTp);
@@ -77,7 +78,8 @@ class SparseIterateOpConverter : public OneToNOpConversionPattern<IterateOp> {
77
78
matchAndRewrite (IterateOp op, OpAdaptor adaptor,
78
79
OneToNPatternRewriter &rewriter) const override {
79
80
if (!op.getCrdUsedLvls ().empty ())
80
- llvm_unreachable (" Not implemented." );
81
+ return rewriter.notifyMatchFailure (
82
+ op, " non-empty coordinates list not implemented." );
81
83
82
84
Location loc = op.getLoc ();
83
85
You can’t perform that action at this time.
0 commit comments