1 //===- TosaToSCFPass.cpp - Lowering Tosa to SCF Dialect -------------------===//
2 //
3 // Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions.
4 // See https://llvm.org/LICENSE.txt for license information.
5 // SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
6 //
7 //===----------------------------------------------------------------------===//
8 //
9 // This transformation pass legalizes Tosa operations to the SCF dialect.
10 //
11 //===----------------------------------------------------------------------===//
12 
13 #include "../PassDetail.h"
14 #include "mlir/Conversion/TosaToSCF/TosaToSCF.h"
15 #include "mlir/Dialect/SCF/IR/SCF.h"
16 #include "mlir/Dialect/Tensor/IR/Tensor.h"
17 #include "mlir/Dialect/Tosa/IR/TosaOps.h"
18 #include "mlir/Dialect/Tosa/Transforms/PassDetail.h"
19 #include "mlir/Dialect/Tosa/Transforms/Passes.h"
20 #include "mlir/IR/PatternMatch.h"
21 #include "mlir/Pass/PassManager.h"
22 #include "mlir/Transforms/DialectConversion.h"
23 #include "mlir/Transforms/GreedyPatternRewriteDriver.h"
24 
25 using namespace mlir;
26 using namespace tosa;
27 
28 namespace {
29 struct TosaToSCF : public TosaToSCFBase<TosaToSCF> {
30 public:
runOnOperation__anon57e36ba00111::TosaToSCF31   void runOnOperation() override {
32     RewritePatternSet patterns(&getContext());
33     ConversionTarget target(getContext());
34     target.addLegalDialect<tensor::TensorDialect, scf::SCFDialect>();
35     target.addIllegalOp<tosa::IfOp, tosa::WhileOp>();
36     target.markUnknownOpDynamicallyLegal([](Operation *) { return true; });
37 
38     auto *op = getOperation();
39     mlir::tosa::populateTosaToSCFConversionPatterns(&patterns);
40     if (failed(applyPartialConversion(op, target, std::move(patterns))))
41       signalPassFailure();
42   }
43 };
44 } // namespace
45 
createTosaToSCF()46 std::unique_ptr<Pass> mlir::tosa::createTosaToSCF() {
47   return std::make_unique<TosaToSCF>();
48 }
49 
addTosaToSCFPasses(OpPassManager & pm)50 void mlir::tosa::addTosaToSCFPasses(OpPassManager &pm) {
51   pm.addNestedPass<func::FuncOp>(createTosaToSCF());
52 }
53