b1d861cf83c20e1566c18f96817f97ed6c0974a0
[platform/core/ml/nnfw.git] / runtime / onert / core / include / compiler / CodeMap.h
1 /*
2  * Copyright (c) 2020 Samsung Electronics Co., Ltd. All Rights Reserved
3  *
4  * Licensed under the Apache License, Version 2.0 (the "License");
5  * you may not use this file except in compliance with the License.
6  * You may obtain a copy of the License at
7  *
8  *      http://www.apache.org/licenses/LICENSE-2.0
9  *
10  * Unless required by applicable law or agreed to in writing, software
11  * distributed under the License is distributed on an "AS IS" BASIS,
12  * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
13  * See the License for the specific language governing permissions and
14  * limitations under the License.
15  */
16
17 #ifndef __ONERT_COMPILER_CODE_MAP_H__
18 #define __ONERT_COMPILER_CODE_MAP_H__
19
20 #include <unordered_map>
21 #include "ir/Index.h"
22 #include "ir/Operation.h"
23 #include "exec/FunctionSequence.h"
24 #include "OperationLowerInfo.h"
25
26 namespace onert
27 {
28 namespace compiler
29 {
30
31 struct CodeAndInfo
32 {
33   ir::OperationIndex op_ind;
34   const ir::Operation *op;
35   const OperationLowerInfo *lower_info;
36   std::unique_ptr<exec::FunctionSequence> fn_seq;
37
38   CodeAndInfo(const ir::OperationIndex op_ind, const ir::Operation *op,
39               const OperationLowerInfo *lower_info,
40               std::unique_ptr<exec::FunctionSequence> &&fn_seq)
41     : op_ind{op_ind}, op{op}, lower_info{lower_info}, fn_seq{std::move(fn_seq)}
42   {
43   }
44 };
45
46 using CodeMap = std::unordered_map<ir::OperationIndex, CodeAndInfo>;
47
48 } // namespace compiler
49 } // namespace onert
50
51 #endif // __ONERT_COMPILER_CODE_MAP_H__