MLIR  20.0.0git
TosaToLinalgNamedPass.cpp
Go to the documentation of this file.
1 //===- TosaToLinalgPass.cpp - Lowering Tosa to Linalg 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 Linalg dialect.
10 //
11 //===----------------------------------------------------------------------===//
12 
14 
24 #include "mlir/IR/PatternMatch.h"
25 #include "mlir/Pass/PassManager.h"
28 
29 namespace mlir {
30 #define GEN_PASS_DEF_TOSATOLINALGNAMED
31 #include "mlir/Conversion/Passes.h.inc"
32 } // namespace mlir
33 
34 using namespace mlir;
35 
36 namespace {
37 struct TosaToLinalgNamed
38  : public impl::TosaToLinalgNamedBase<TosaToLinalgNamed> {
39 public:
40  TosaToLinalgNamed(const TosaToLinalgNamedOptions &options)
41  : impl::TosaToLinalgNamedBase<TosaToLinalgNamed>(options) {}
42 
43  void getDependentDialects(DialectRegistry &registry) const override {
44  registry
45  .insert<arith::ArithDialect, linalg::LinalgDialect, math::MathDialect,
46  tensor::TensorDialect, scf::SCFDialect>();
47  }
48 
49  void runOnOperation() override {
50  RewritePatternSet patterns(&getContext());
51  ConversionTarget target(getContext());
52  target.addLegalDialect<linalg::LinalgDialect, tosa::TosaDialect,
53  tensor::TensorDialect, scf::SCFDialect>();
54 
55  // Not every TOSA op can be legalized to linalg.
56  target.addIllegalOp<tosa::Conv2DOp>();
57  target.addIllegalOp<tosa::Conv3DOp>();
58  target.addIllegalOp<tosa::DepthwiseConv2DOp>();
59  target.addIllegalOp<tosa::MaxPool2dOp>();
60  target.addIllegalOp<tosa::AvgPool2dOp>();
61  target.addIllegalOp<tosa::MatMulOp>();
62  target.addIllegalOp<tosa::FullyConnectedOp>();
63  target.addIllegalOp<tosa::TransposeOp>();
64 
65  target.markUnknownOpDynamicallyLegal([](Operation *) { return true; });
66 
67  FunctionOpInterface func = getOperation();
68  TosaToLinalgNamedOptions options;
69  options.preferConv2DKernelLayoutHWCF = preferConv2DKernelLayoutHWCF;
71  if (failed(applyFullConversion(func, target, std::move(patterns))))
72  signalPassFailure();
73  }
74 };
75 } // namespace
76 
77 std::unique_ptr<Pass>
78 mlir::tosa::createTosaToLinalgNamed(const TosaToLinalgNamedOptions &options) {
79  return std::make_unique<TosaToLinalgNamed>(options);
80 }
static MLIRContext * getContext(OpFoldResult val)
static llvm::ManagedStatic< PassManagerOptions > options
This class describes a specific conversion target.
void addLegalDialect(StringRef name, Names... names)
Register the operations of the given dialects as legal.
void markUnknownOpDynamicallyLegal(const DynamicLegalityCallbackFn &fn)
Register unknown operations as dynamically legal.
void addIllegalOp(OperationName op)
Register the given operation as illegal, i.e.
The DialectRegistry maps a dialect namespace to a constructor for the matching dialect.
Operation is the basic unit of execution within MLIR.
Definition: Operation.h:88
std::unique_ptr< Pass > createTosaToLinalgNamed(const TosaToLinalgNamedOptions &options=TosaToLinalgNamedOptions())
void populateTosaToLinalgNamedConversionPatterns(RewritePatternSet *patterns, const TosaToLinalgNamedOptions &options)
Populates conversion passes from TOSA dialect to Linalg named operations.
Include the generated interface declarations.
LogicalResult applyFullConversion(ArrayRef< Operation * > ops, const ConversionTarget &target, const FrozenRewritePatternSet &patterns, ConversionConfig config=ConversionConfig())
Apply a complete conversion on the given operations, and all nested operations.