Skip to content

Commit 1ee000f

Browse files
committed
fix(Unfuse.cpp): adapt to new IteratorType enum from StringRef
1 parent 4eb1e1f commit 1ee000f

File tree

1 file changed

+5
-3
lines changed

1 file changed

+5
-3
lines changed

mlir/lib/Dialect/Linalg/Transforms/Unfuse.cpp

Lines changed: 5 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -523,8 +523,9 @@ Value sumKeepDim(OpBuilder &builder, Location loc, Value in, unsigned dim) {
523523
AffineMap indexingMaps[] = {inMap, outMap};
524524

525525
// Compute the iterator types.
526-
SmallVector<StringRef> iteratorTypes(rank, "parallel");
527-
iteratorTypes.back() = "reduction";
526+
SmallVector<utils::IteratorType> iteratorTypes(rank,
527+
utils::IteratorType::parallel);
528+
iteratorTypes.back() = utils::IteratorType::reduction;
528529

529530
return builder
530531
.create<linalg::GenericOp>(
@@ -580,7 +581,8 @@ struct SoftmaxLowering : OpRewritePattern<SoftmaxOp> {
580581
rank, rewriter.getContext())};
581582

582583
// Compute the iterator types.
583-
SmallVector<StringRef> iteratorTypes(rank, "parallel");
584+
SmallVector<utils::IteratorType> iteratorTypes(
585+
rank, utils::IteratorType::parallel);
584586

585587
rewriter.replaceOpWithNewOp<linalg::GenericOp>(
586588
op,

0 commit comments

Comments
 (0)