Skip to content

Commit c640e20

Browse files
committed
Run both patterns at the same time
1 parent f84789f commit c640e20

File tree

1 file changed

+3
-17
lines changed

1 file changed

+3
-17
lines changed

mlir/lib/Conversion/ConvertToSPIRV/ConvertToSPIRVPass.cpp

Lines changed: 3 additions & 17 deletions
Original file line numberDiff line numberDiff line change
@@ -39,39 +39,25 @@ namespace {
3939
/// A pass to perform the SPIR-V conversion.
4040
class ConvertToSPIRVPass
4141
: public impl::ConvertToSPIRVPassBase<ConvertToSPIRVPass> {
42-
using impl::ConvertToSPIRVPassBase::ConvertToSPIRVPassBase;
42+
using impl::ConvertToSPIRVPassBase<
43+
ConvertToSPIRVPass>::ConvertToSPIRVPassBase;
4344

4445
void runOnOperation() override {
4546
MLIRContext *context = &getContext();
4647
Operation *op = getOperation();
4748

4849
if (runSignatureConversion) {
49-
// Unroll vectors in function inputs to native vector size.
50-
llvm::errs() << "Start unrolling function inputs\n";
50+
// Unroll vectors in function signatures to native vector size.
5151
{
5252
RewritePatternSet patterns(context);
5353
populateFuncOpVectorRewritePatterns(patterns);
54-
GreedyRewriteConfig config;
55-
config.strictMode = GreedyRewriteStrictness::ExistingOps;
56-
if (failed(
57-
applyPatternsAndFoldGreedily(op, std::move(patterns), config)))
58-
return signalPassFailure();
59-
}
60-
llvm::errs() << "Finish unrolling function inputs\n";
61-
62-
// Unroll vectors in function outputs to native vector size.
63-
llvm::errs() << "Start unrolling function outputs\n";
64-
{
65-
RewritePatternSet patterns(context);
6654
populateReturnOpVectorRewritePatterns(patterns);
6755
GreedyRewriteConfig config;
6856
config.strictMode = GreedyRewriteStrictness::ExistingOps;
6957
if (failed(
7058
applyPatternsAndFoldGreedily(op, std::move(patterns), config)))
7159
return signalPassFailure();
7260
}
73-
llvm::errs() << "Finish unrolling function outputs\n";
74-
7561
return;
7662
}
7763

0 commit comments

Comments
 (0)