1/**
2 * Copyright (c) Glow Contributors. See CONTRIBUTORS file.
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#ifndef GLOW_OPTIMIZER_IROPTIMIZER_IROPTIMIZER_H
17#define GLOW_OPTIMIZER_IROPTIMIZER_IROPTIMIZER_H
18
19#include <memory>
20
21namespace glow {
22
23class IRFunction;
24class IRContainer;
25class Backend;
26
27/// Perform target-independent optimizations on the IR representation of \p M.
28/// Apply the buffer sharing optimizations if \p shouldShareBuffers is true.
29void optimize(IRFunction &M, bool shouldShareBuffers);
30
31/// Perform backend-specific optimizations on the IR representation of \p M.
32/// Apply the buffer sharing optimizations if \p shouldShareBuffers is true.
33void optimize(IRFunction &M, const Backend &B, bool shouldShareBuffers);
34
35/// Helper to generate and optimize IR from given Function \p F. \p
36/// shouldShareBuffers signifies whether to use the share buffers optimization.
37/// Backend /p B is used to allow for custom lowering from Node to
38/// Instruction IR.
39std::unique_ptr<IRFunction> generateAndOptimizeIR(IRContainer *F,
40 const Backend &B,
41 bool shouldShareBuffers);
42} // namespace glow
43
44#endif // GLOW_OPTIMIZER_IROPTIMIZER_IROPTIMIZER_H
45