MLIR  22.0.0git
TosaToMLProgram.cpp
Go to the documentation of this file.
1 //===- TosaToMLProgram.cpp - Lowering Tosa to MLProgram 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 // These rewriters lower from the TOSA dialect to the MLProgram dialect.
10 //
11 //===----------------------------------------------------------------------===//
12 
16 #include "mlir/IR/PatternMatch.h"
17 
18 using namespace mlir;
19 using namespace tosa;
20 namespace {
21 
22 class VariableOpConverter : public OpRewritePattern<tosa::VariableOp> {
23 public:
25 
26  LogicalResult matchAndRewrite(tosa::VariableOp op,
27  PatternRewriter &rewriter) const final {
28  auto variableType = tosa::getVariableType(op);
29  auto newVariable = mlir::ml_program::GlobalOp::create(
30  rewriter, op.getLoc(), op.getName(), variableType, /*is_mutable=*/true,
31  op.getInitialValueAttr(), /*sym_visibility=*/nullptr);
32  newVariable.setPrivate();
33  rewriter.replaceOp(op, newVariable);
34  return success();
35  }
36 };
37 
38 class VariableWriteOpConverter
39  : public OpRewritePattern<tosa::VariableWriteOp> {
40 public:
42 
43  LogicalResult matchAndRewrite(tosa::VariableWriteOp op,
44  PatternRewriter &rewriter) const final {
45  auto globalSymbolRef =
46  SymbolRefAttr::get(rewriter.getContext(), op.getName());
47  auto newVariableWrite = ml_program::GlobalStoreOp::create(
48  rewriter, op.getLoc(), globalSymbolRef, op.getInput1());
49  rewriter.replaceOp(op, newVariableWrite);
50  return success();
51  }
52 };
53 
54 class VariableReadOpConverter : public OpRewritePattern<tosa::VariableReadOp> {
55 public:
57 
58  LogicalResult matchAndRewrite(tosa::VariableReadOp op,
59  PatternRewriter &rewriter) const final {
60  auto globalSymbolRef =
61  SymbolRefAttr::get(rewriter.getContext(), op.getName());
62  auto newVariableRead = ml_program::GlobalLoadOp::create(
63  rewriter, op.getLoc(), op.getType(), globalSymbolRef);
64  rewriter.replaceOp(op, newVariableRead);
65 
66  return success();
67  }
68 };
69 
70 } // namespace
71 
74  patterns->add<VariableOpConverter, VariableWriteOpConverter,
75  VariableReadOpConverter>(patterns->getContext());
76 }
A special type of RewriterBase that coordinates the application of a rewrite pattern on the current I...
Definition: PatternMatch.h:783
RankedTensorType getVariableType(VariableOp variableOp)
void populateTosaToMLProgramConversionPatterns(RewritePatternSet *patterns)
Include the generated interface declarations.
const FrozenRewritePatternSet & patterns
auto get(MLIRContext *context, Ts &&...params)
Helper method that injects context only if needed, this helps unify some of the attribute constructio...
OpRewritePattern is a wrapper around RewritePattern that allows for matching and rewriting against an...
Definition: PatternMatch.h:314