Skip to content

Commit 82004a2

Browse files
committed
Minor changes
1 parent 6894af0 commit 82004a2

File tree

2 files changed

+5
-5
lines changed

2 files changed

+5
-5
lines changed

mlir/lib/Dialect/Vector/Transforms/LowerVectorToElements.cpp

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -20,7 +20,7 @@ using namespace mlir;
2020

2121
namespace {
2222

23-
struct UnrollToElements final : OpRewritePattern<vector::ToElementsOp> {
23+
struct UnrollToElements final : public OpRewritePattern<vector::ToElementsOp> {
2424
using OpRewritePattern::OpRewritePattern;
2525

2626
LogicalResult matchAndRewrite(vector::ToElementsOp op,
@@ -38,7 +38,7 @@ struct UnrollToElements final : OpRewritePattern<vector::ToElementsOp> {
3838
SmallVector<Value, 0> results;
3939
for (const Value &vector : vectors) {
4040
auto subElements =
41-
rewriter.create<vector::ToElementsOp>(op.getLoc(), vector);
41+
vector::ToElementsOp::create(rewriter, op.getLoc(), vector);
4242
llvm::append_range(results, subElements.getResults());
4343
}
4444
rewriter.replaceOp(op, results);

mlir/lib/Dialect/Vector/Utils/VectorUtils.cpp

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -401,10 +401,10 @@ vector::isValidMaskedInputVector(ArrayRef<int64_t> shape,
401401
///
402402
/// will produce the following IR changes
403403
///
404-
/// %v0 = vector.extract %v[0] : vector<axbx...>
405-
/// %v1 = vector.extract %v[1] : vector<axbx...>
404+
/// %v0 = vector.extract %v[0] : vector<axbx...> from vector<nxaxb...>
405+
/// %v1 = vector.extract %v[1] : vector<axbx...> from vector<nxaxb...>
406406
/// ...
407-
/// %vnminusone = vector.extract %v[n-1] : vector<axbx...>
407+
/// %vnminusone = vector.extract %v[n-1] : vector<axbx...> from ...
408408
///
409409
/// and returns SmallVector<Value> r = {[[%v0]], [[%v1]], ..., [[%vnminusone]]}
410410
FailureOr<SmallVector<Value>>

0 commit comments

Comments
 (0)