You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
66 lines
2.3 KiB
66 lines
2.3 KiB
/* 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. */
|
|
|
|
#pragma once
|
|
|
|
/*
|
|
* This file contains Analyzer, an class that exposed as a library that analyze
|
|
* and optimize Fluid ProgramDesc for inference. Similar to LLVM, it has
|
|
* multiple flags to
|
|
* control whether an process is applied on the program.
|
|
*
|
|
* The processes are called Passes in analysis, the Passes are placed in a
|
|
* pipeline, the first Pass is the FluidToDataFlowGraphPass which transforms a
|
|
* Fluid ProgramDesc to
|
|
* a data flow graph, the last Pass is DataFlowGraphToFluidPass which transforms
|
|
* a data flow graph to a Fluid ProgramDesc. The passes in the middle of the
|
|
* pipeline can be any Passes
|
|
* which take a node or data flow graph as input.
|
|
*
|
|
* The Analyzer can be used in two methods, the first is a executable file which
|
|
* can be used to pre-process the inference model and can be controlled by
|
|
* passing difference command flags;
|
|
* the other way is to compose inside the inference API as a runtime pre-process
|
|
* phase in the inference service.
|
|
*/
|
|
|
|
#include <gflags/gflags.h>
|
|
#include "paddle/fluid/inference/analysis/pass.h"
|
|
#include "paddle/fluid/inference/analysis/pass_manager.h"
|
|
|
|
namespace paddle {
|
|
|
|
// TODO(Superjomn) add a definition flag like PADDLE_WITH_TENSORRT and hide this
|
|
// flag if not available.
|
|
DECLARE_bool(inference_analysis_enable_tensorrt_subgraph_engine);
|
|
DECLARE_string(inference_analysis_graphviz_log_root);
|
|
DECLARE_string(inference_analysis_output_storage_path);
|
|
|
|
namespace inference {
|
|
namespace analysis {
|
|
|
|
class Analyzer : public OrderedRegistry<PassManager> {
|
|
public:
|
|
// Register all the pass-managers.
|
|
Analyzer();
|
|
|
|
void Run(Argument* argument);
|
|
|
|
DISABLE_COPY_AND_ASSIGN(Analyzer);
|
|
};
|
|
|
|
} // namespace analysis
|
|
} // namespace inference
|
|
} // namespace paddle
|