operator.h 5.7 KB
Newer Older
Z
zhaojiaying01 已提交
1 2 3 4 5 6 7 8 9 10 11 12 13
/* Copyright (c) 2018 PaddlePaddle Authors. All Rights Reserved.

Licensed under the Apache License, Version 2.0 (the "License");
you may not use this file except in compliance with the License.
You may obtain a copy of the License at

    http://www.apache.org/licenses/LICENSE-2.0

Unless required by applicable law or agreed to in writing, software
distributed under the License is distributed on an "AS IS" BASIS,
WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
See the License for the specific language governing permissions and
limitations under the License. */
朔-望's avatar
朔-望 已提交
14 15 16

#pragma once

L
liuruilong 已提交
17
#include <map>
Z
zhaojiaying01 已提交
18
#include <string>
H
hjchen2 已提交
19
#include <utility>
Z
zhaojiaying01 已提交
20
#include <vector>
L
liuruilong 已提交
21

L
liuruilong 已提交
22 23
#include "common/enforce.h"
#include "common/type_define.h"
L
liuruilong 已提交
24 25
#include "common/types.h"
#include "common/variant.h"
L
liuruilong 已提交
26
#include "framework/attribute.h"
L
liuruilong 已提交
27
#include "framework/op_info.h"
L
liuruilong 已提交
28
#include "framework/op_kernel_type.h"
L
liuruilong 已提交
29 30
#include "framework/op_registry.h"
#include "framework/program/block_desc.h"
L
liuruilong 已提交
31
#include "framework/program/program-optimize/node.h"
L
liuruilong 已提交
32 33 34
#include "framework/scope.h"
#include "framework/tensor.h"
#include "framework/variable.h"
朔-望's avatar
朔-望 已提交
35 36

namespace paddle_mobile {
朔-望's avatar
朔-望 已提交
37
namespace framework {
38 39
using std::string;
using std::vector;
朔-望's avatar
朔-望 已提交
40

W
wangliu 已提交
41 42 43 44 45 46 47 48 49 50 51 52
template <typename T>
static T *GetVarValue(const string &key, const VariableNameMap &var_map,
                      const Scope &scope) {
  auto var_vec = var_map.at(key);
  if (!var_vec.empty()) {
    auto var = scope.FindVar(var_vec[0]);
    return var->GetMutable<T>();
  } else {
    return nullptr;
  }
}

朔-望's avatar
朔-望 已提交
53
template <typename Dtype>
L
liuruilong 已提交
54
class OperatorBase {
朔-望's avatar
朔-望 已提交
55
 public:
L
liuruilong 已提交
56 57 58
  /*
   *  @b op 基类的实例化方法, op 获取到了 输入、参数以及提前分配好的输出 tensor
   * */
59 60 61 62
  OperatorBase(const std::string &type, const VariableNameMap &inputs,
               const VariableNameMap &outputs, const AttributeMap &attrs,
               std::shared_ptr<Scope> scope);
  virtual ~OperatorBase() {}
63
  void Run() const;
L
liuruilong 已提交
64
  std::vector<string> GetOutKeys() const;
65
  std::vector<string> GetInputKeys() const;
66
  virtual void RunImpl() const = 0;
朔-望's avatar
朔-望 已提交
67

E
eclipsess 已提交
68
  virtual void Init() = 0;
L
liuruilong 已提交
69 70 71
  /*
   * @b op 运算所需的输入, 如上一层的输出结果、卷积核
   * */
72
  const VariableNameMap &Inputs() const { return inputs_; }
L
liuruilong 已提交
73 74 75
  /*
   * @b op 的输出, 内存会提前被分配好, 运算结果会被存到分配好的内存内
   * */
76
  const VariableNameMap &Outputs() const { return outputs_; }
L
liuruilong 已提交
77 78 79
  /*
   * @b op 类型
   * */
80
  const std::string &Type() const { return type_; }
L
liuruilong 已提交
81 82 83
  /*
   * @b op 运算所需要用到的参数: 如 conv 运算所需要用到的 stride
   * */
84 85 86 87
  const AttributeMap &Attrs() const { return attrs_; }
  void ClearVariables(const std::vector<std::string> &var_names) const {
    if (this->scope_) {
      this->scope_->EraseVars(var_names);
朔-望's avatar
朔-望 已提交
88
    }
89
  }
L
liuruilong 已提交
90 91 92 93
  /*
   * @b 根据输入形状和参数计算出输出形状
   * */
  virtual void InferShape() const = 0;
L
liuruilong 已提交
94

朔-望's avatar
朔-望 已提交
95
 protected:
96 97 98 99 100
  std::shared_ptr<Scope> scope_;
  std::string type_;
  VariableNameMap inputs_;
  VariableNameMap outputs_;
  AttributeMap attrs_;
L
liuruilong 已提交
101

朔-望's avatar
朔-望 已提交
102
 private:
103
  void CheckAllInputOutputSet() const;
朔-望's avatar
朔-望 已提交
104
};
朔-望's avatar
朔-望 已提交
105

L
liuruilong 已提交
106 107 108
/*
 * @b 这个类为所有带有运算的 op 的父类, 这个 op 继承与 OperatorBase
 * */
L
liuruilong 已提交
109
template <typename Dtype, typename ParamType, typename KernelType>
朔-望's avatar
朔-望 已提交
110
class OperatorWithKernel : public OperatorBase<Dtype> {
朔-望's avatar
朔-望 已提交
111
 public:
112 113 114
  OperatorWithKernel(const std::string &type, const VariableNameMap &inputs,
                     const VariableNameMap &outputs, const AttributeMap &attrs,
                     std::shared_ptr<Scope> scope)
L
liuruilong 已提交
115
      : OperatorBase<Dtype>(type, inputs, outputs, attrs, scope),
L
liuruilong 已提交
116
        param_(inputs, outputs, attrs, *scope) {}
L
liuruilong 已提交
117

L
liuruilong 已提交
118
  virtual void RunImpl() const { this->kernel_.Compute(this->param_); }
119

W
wangliu 已提交
120
  virtual void InferShape() const = 0;
L
liuruilong 已提交
121

E
eclipsess 已提交
122 123
  void Init() {
    PADDLE_MOBILE_ENFORCE(kernel_.Init(&param_), "  %s kernel init failed",
L
liuruilong 已提交
124 125 126
                          this->type_.c_str());
  }

L
liuruilong 已提交
127 128 129
 protected:
  KernelType kernel_;
  ParamType param_;
朔-望's avatar
朔-望 已提交
130
};
朔-望's avatar
朔-望 已提交
131

L
liuruilong 已提交
132 133 134
/*
 * @b 所有kernel的父类
 * */
朔-望's avatar
朔-望 已提交
135
template <typename Dtype, typename P>
L
liuruilong 已提交
136
class OpKernelBase {
朔-望's avatar
朔-望 已提交
137
 public:
L
liuruilong 已提交
138 139 140 141 142
  /*
   * @b 所有kernel 需实现 Compute 方法
   * @p para 这个参数为 kernel 运算时所需要用到参数组成的一个结构体,
   *    所有结构体存在与: paddle-mobile/src/operators/op_param.h
   * */
E
eclipsess 已提交
143 144 145 146 147 148 149
#ifdef PADDLE_MOBILE_MALI_GPU
  OpKernelBase() { acl_op_ = nullptr; }
  void *GetAclOp() const { return acl_op_; }
  void SetAclOp(void *op, void *ob) const {
    reinterpret_cast<OpKernelBase<Dtype, P> *>(ob)->acl_op_ = op;
  }
#endif
150
  virtual void Compute(const P &para) const = 0;
151
  virtual bool Init(P *para) { return true; }
152
  virtual ~OpKernelBase() = default;
E
eclipsess 已提交
153 154 155 156 157

 private:
#ifdef PADDLE_MOBILE_MALI_GPU
  void *acl_op_;
#endif
朔-望's avatar
朔-望 已提交
158
};
朔-望's avatar
朔-望 已提交
159

Z
zhaojiaying01 已提交
160 161 162 163 164 165 166
#define DEFINE_OP_CONSTRUCTOR(cls, parent_cls)                                 \
  cls(const std::string &type, const ::paddle_mobile::VariableNameMap &inputs, \
      const ::paddle_mobile::VariableNameMap &outputs,                         \
      const ::paddle_mobile::framework::AttributeMap &attrs,                   \
      std::shared_ptr<::paddle_mobile::framework::Scope> scope)                \
      : parent_cls<Dtype, T>(type, inputs, outputs, attrs, scope) {}

L
liuruilong 已提交
167
class FusionOpMatcher {
L
liuruilong 已提交
168 169 170 171 172
 public:
  FusionOpMatcher() {}

  virtual std::string Type() = 0;

L
liuruilong 已提交
173 174 175
  virtual void FolderNodes(
      Node *node,
      std::vector<std::shared_ptr<framework::Node>> *removed_nodes) {
L
liuruilong 已提交
176
    node->Folder(node_.Depth(), Type(), {}, removed_nodes);
L
liuruilong 已提交
177 178
  }

L
liuruilong 已提交
179
  virtual Node &BeginNode() { return node_; }
L
liuruilong 已提交
180

L
liuruilong 已提交
181
  std::string BeginType() { return node_.Type(); }
L
liuruilong 已提交
182

183 184
  virtual std::vector<std::pair<int, std::string>> NeedCheck() { return {}; }

L
for  
liuruilong 已提交
185
  //  virtual  bool Fusion();
L
liuruilong 已提交
186 187 188 189 190 191
 protected:
  Node node_;
  std::string type_;
  std::shared_ptr<OpDesc> new_opdesc_;
};

朔-望's avatar
朔-望 已提交
192 193
}  // namespace framework
}  // namespace paddle_mobile