未验证 提交 53db8325 编写于 作者: H hong19860320 提交者: GitHub

[Core] Adjust the file path of subgraph_bridge_registry and subgraph_engine_base (#4270)

上级 1716db63
...@@ -142,6 +142,15 @@ if (NOT LITE_ON_TINY_PUBLISH) ...@@ -142,6 +142,15 @@ if (NOT LITE_ON_TINY_PUBLISH)
add_subdirectory(arena) add_subdirectory(arena)
endif() endif()
if(LITE_WITH_NPU OR LITE_WITH_XTCL OR LITE_WITH_BM OR LITE_WITH_RKNPU OR LITE_WITH_MLU OR LITE_WITH_APU OR LITE_WITH_HUAWEI_ASCEND_NPU)
lite_cc_library(subgraph_bridge_registry
SRCS subgraph_bridge_registry.cc
DEPS op)
lite_cc_library(subgraph_engine_base
SRCS subgraph_engine_base.cc
DEPS tensor op scope program)
endif()
# for mobile, unnecessary to compile the following testings. # for mobile, unnecessary to compile the following testings.
if (LITE_WITH_LIGHT_WEIGHT_FRAMEWORK) if (LITE_WITH_LIGHT_WEIGHT_FRAMEWORK)
return() return()
......
...@@ -12,19 +12,19 @@ ...@@ -12,19 +12,19 @@
// See the License for the specific language governing permissions and // See the License for the specific language governing permissions and
// limitations under the License. // limitations under the License.
#include "lite/kernels/npu/bridges/registry.h" #include "lite/core/subgraph_bridge_registry.h"
#include <utility> #include <utility>
namespace paddle { namespace paddle {
namespace lite { namespace lite {
namespace subgraph { namespace subgraph {
Registry& Registry::Instance() { SubgraphBridgeRegistry& SubgraphBridgeRegistry::Instance() {
static Registry x; static SubgraphBridgeRegistry x;
return x; return x;
} }
void Registry::Insert(const std::string& op_type, void SubgraphBridgeRegistry::Insert(const std::string& op_type,
const TargetType& target, const TargetType& target,
const cvt_func_type& cvt_func_name) { const cvt_func_type& cvt_func_name) {
int key = static_cast<int>(target); int key = static_cast<int>(target);
...@@ -35,13 +35,13 @@ void Registry::Insert(const std::string& op_type, ...@@ -35,13 +35,13 @@ void Registry::Insert(const std::string& op_type,
map_.at(key).insert(std::make_pair(op_type, cvt_func_name)); map_.at(key).insert(std::make_pair(op_type, cvt_func_name));
} }
const cvt_func_type& Registry::Select(const std::string& op_type, const cvt_func_type& SubgraphBridgeRegistry::Select(
const TargetType& target) const { const std::string& op_type, const TargetType& target) const {
int key = static_cast<int>(target); int key = static_cast<int>(target);
return map_.at(key).at(op_type); return map_.at(key).at(op_type);
} }
bool Registry::Exists(const std::string& op_type, bool SubgraphBridgeRegistry::Exists(const std::string& op_type,
const TargetType& target) const { const TargetType& target) const {
int key = static_cast<int>(target); int key = static_cast<int>(target);
bool found = map_.find(key) != map_.end(); bool found = map_.find(key) != map_.end();
......
...@@ -36,9 +36,9 @@ inline bool CHECK_REBUILD_WHEN_SHAPE_CHANGED(int status) { ...@@ -36,9 +36,9 @@ inline bool CHECK_REBUILD_WHEN_SHAPE_CHANGED(int status) {
using cvt_func_type = using cvt_func_type =
std::function<int(void* ctx, OpLite* op, KernelBase* kernel)>; std::function<int(void* ctx, OpLite* op, KernelBase* kernel)>;
using cvt_map_type = std::map<int, std::map<std::string, cvt_func_type>>; using cvt_map_type = std::map<int, std::map<std::string, cvt_func_type>>;
class Registry { class SubgraphBridgeRegistry {
public: public:
static Registry& Instance(); static SubgraphBridgeRegistry& Instance();
void Insert(const std::string& op_type, void Insert(const std::string& op_type,
const TargetType& target, const TargetType& target,
...@@ -46,11 +46,11 @@ class Registry { ...@@ -46,11 +46,11 @@ class Registry {
const cvt_func_type& Select(const std::string& op_type, const cvt_func_type& Select(const std::string& op_type,
const TargetType& target) const; const TargetType& target) const;
bool Exists(const std::string& op_type, const TargetType& target) const; bool Exists(const std::string& op_type, const TargetType& target) const;
Registry() = default; SubgraphBridgeRegistry() = default;
private: private:
cvt_map_type map_; cvt_map_type map_;
DISALLOW_COPY_AND_ASSIGN(Registry); DISALLOW_COPY_AND_ASSIGN(SubgraphBridgeRegistry);
}; };
} // namespace subgraph } // namespace subgraph
...@@ -69,7 +69,7 @@ class Registry { ...@@ -69,7 +69,7 @@ class Registry {
"REGISTER_SUBGRAPH_BRIDGE must be called in global namespace only " \ "REGISTER_SUBGRAPH_BRIDGE must be called in global namespace only " \
"once!"); \ "once!"); \
int __reg_subgraph_bridge_##op_type__##_##target__##_Insert() { \ int __reg_subgraph_bridge_##op_type__##_##target__##_Insert() { \
paddle::lite::subgraph::Registry::Instance().Insert( \ paddle::lite::subgraph::SubgraphBridgeRegistry::Instance().Insert( \
#op_type__, TARGET(target__), cvt_func_name); \ #op_type__, TARGET(target__), cvt_func_name); \
return 0; \ return 0; \
} }
......
...@@ -12,18 +12,18 @@ ...@@ -12,18 +12,18 @@
// See the License for the specific language governing permissions and // See the License for the specific language governing permissions and
// limitations under the License. // limitations under the License.
#include "lite/kernels/npu/bridges/engine.h" #include "lite/core/subgraph_engine_base.h"
#include <sys/time.h> #include <sys/time.h>
#include <time.h> #include <time.h>
#include <algorithm> #include <algorithm>
#include <utility> #include <utility>
#include "lite/kernels/npu/bridges/registry.h"
namespace paddle { namespace paddle {
namespace lite { namespace lite {
namespace subgraph { namespace subgraph {
Engine::Engine(KernelContext *ctx, SubgraphEngineBase::SubgraphEngineBase(
KernelContext *ctx,
int block_idx, int block_idx,
const std::shared_ptr<const cpp::ProgramDesc> &program_desc, const std::shared_ptr<const cpp::ProgramDesc> &program_desc,
Scope *exec_scope, Scope *exec_scope,
...@@ -43,7 +43,7 @@ Engine::Engine(KernelContext *ctx, ...@@ -43,7 +43,7 @@ Engine::Engine(KernelContext *ctx,
std::stable_sort(output_names_.begin(), output_names_.end()); std::stable_sort(output_names_.begin(), output_names_.end());
} }
bool Engine::Run() { bool SubgraphEngineBase::Run() {
if (is_first_epoch_) { if (is_first_epoch_) {
PrepareWorkspaceForDeviceProgram(); PrepareWorkspaceForDeviceProgram();
is_first_epoch_ = false; is_first_epoch_ = false;
...@@ -54,7 +54,7 @@ bool Engine::Run() { ...@@ -54,7 +54,7 @@ bool Engine::Run() {
return LaunchDeviceProgram(); return LaunchDeviceProgram();
} }
bool Engine::PrepareWorkspaceForOriginProgram() { bool SubgraphEngineBase::PrepareWorkspaceForOriginProgram() {
origin_idims_.resize(input_names_.size()); origin_idims_.resize(input_names_.size());
origin_itensors_.resize(input_names_.size()); origin_itensors_.resize(input_names_.size());
for (int i = 0; i < input_names_.size(); i++) { for (int i = 0; i < input_names_.size(); i++) {
...@@ -69,7 +69,7 @@ bool Engine::PrepareWorkspaceForOriginProgram() { ...@@ -69,7 +69,7 @@ bool Engine::PrepareWorkspaceForOriginProgram() {
return true; return true;
} }
bool Engine::BuildOriginProgram() { bool SubgraphEngineBase::BuildOriginProgram() {
// TODO(hong19860320) The block_desc need to be divided into subgraphs during // TODO(hong19860320) The block_desc need to be divided into subgraphs during
// the exection time. But only see them as a subgraph now. // the exection time. But only see them as a subgraph now.
if (!origin_program_) { if (!origin_program_) {
...@@ -79,7 +79,7 @@ bool Engine::BuildOriginProgram() { ...@@ -79,7 +79,7 @@ bool Engine::BuildOriginProgram() {
return true; return true;
} }
bool Engine::LaunchOriginProgram() { bool SubgraphEngineBase::LaunchOriginProgram() {
if (!origin_program_) { if (!origin_program_) {
BuildOriginProgram(); BuildOriginProgram();
} }
...@@ -91,15 +91,15 @@ bool Engine::LaunchOriginProgram() { ...@@ -91,15 +91,15 @@ bool Engine::LaunchOriginProgram() {
return false; return false;
} }
bool Engine::PrepareWorkspaceForDeviceProgram() { bool SubgraphEngineBase::PrepareWorkspaceForDeviceProgram() {
return PrepareWorkspaceForOriginProgram(); return PrepareWorkspaceForOriginProgram();
} }
bool Engine::BuildDeviceProgram() { return BuildOriginProgram(); } bool SubgraphEngineBase::BuildDeviceProgram() { return BuildOriginProgram(); }
bool Engine::LaunchDeviceProgram() { return LaunchOriginProgram(); } bool SubgraphEngineBase::LaunchDeviceProgram() { return LaunchOriginProgram(); }
bool Engine::InputShapeChanged() { bool SubgraphEngineBase::InputShapeChanged() {
bool changed = false; bool changed = false;
for (size_t i = 0; i < origin_itensors_.size(); i++) { for (size_t i = 0; i < origin_itensors_.size(); i++) {
auto origin_idim = origin_itensors_[i]->dims().Vectorize(); auto origin_idim = origin_itensors_[i]->dims().Vectorize();
......
...@@ -26,20 +26,21 @@ namespace paddle { ...@@ -26,20 +26,21 @@ namespace paddle {
namespace lite { namespace lite {
namespace subgraph { namespace subgraph {
class Engine { class SubgraphEngineBase {
public: public:
Engine(KernelContext *ctx, SubgraphEngineBase(
KernelContext *ctx,
int block_idx, int block_idx,
const std::shared_ptr<const cpp::ProgramDesc> &program_desc, const std::shared_ptr<const cpp::ProgramDesc> &program_desc,
Scope *exec_scope, Scope *exec_scope,
const std::vector<std::string> &input_names, const std::vector<std::string> &input_names,
const std::vector<std::string> &output_names); const std::vector<std::string> &output_names);
virtual ~Engine() = default; virtual ~SubgraphEngineBase() = default;
virtual bool Run(); virtual bool Run();
private: private:
Engine(const Engine &) = delete; SubgraphEngineBase(const SubgraphEngineBase &) = delete;
protected: protected:
virtual bool PrepareWorkspaceForOriginProgram(); virtual bool PrepareWorkspaceForOriginProgram();
......
add_subdirectory(bridges) add_subdirectory(bridges)
add_kernel(subgraph_compute_apu APU basic SRCS subgraph_compute.cc DEPS ${lite_kernel_deps} device_apu neuron_adapter subgraph_bridge_engine ${apu_subgraph_bridges}) add_kernel(subgraph_compute_apu APU basic SRCS subgraph_compute.cc DEPS ${lite_kernel_deps} device_apu neuron_adapter subgraph_engine_base ${apu_subgraph_bridges})
...@@ -12,8 +12,9 @@ ...@@ -12,8 +12,9 @@
// See the License for the specific language governing permissions and // See the License for the specific language governing permissions and
// limitations under the License. // limitations under the License.
#include "lite/core/subgraph_bridge_registry.h"
#include "lite/kernels/apu/bridges/graph.h"
#include "lite/kernels/apu/bridges/utility.h" #include "lite/kernels/apu/bridges/utility.h"
#include "lite/kernels/npu/bridges/registry.h"
namespace paddle { namespace paddle {
namespace lite { namespace lite {
......
...@@ -15,9 +15,9 @@ ...@@ -15,9 +15,9 @@
#include "lite/operators/conv_op.h" #include "lite/operators/conv_op.h"
#include <iostream> #include <iostream>
#include <vector> #include <vector>
#include "lite/core/subgraph_bridge_registry.h"
#include "lite/kernels/apu/bridges/graph.h" #include "lite/kernels/apu/bridges/graph.h"
#include "lite/kernels/apu/bridges/utility.h" #include "lite/kernels/apu/bridges/utility.h"
#include "lite/kernels/npu/bridges/registry.h"
namespace paddle { namespace paddle {
namespace lite { namespace lite {
......
...@@ -12,8 +12,9 @@ ...@@ -12,8 +12,9 @@
// See the License for the specific language governing permissions and // See the License for the specific language governing permissions and
// limitations under the License. // limitations under the License.
#include "lite/core/subgraph_bridge_registry.h"
#include "lite/kernels/apu/bridges/graph.h" #include "lite/kernels/apu/bridges/graph.h"
#include "lite/kernels/npu/bridges/registry.h" #include "lite/kernels/apu/bridges/utility.h"
namespace paddle { namespace paddle {
namespace lite { namespace lite {
......
...@@ -12,9 +12,9 @@ ...@@ -12,9 +12,9 @@
// See the License for the specific language governing permissions and // See the License for the specific language governing permissions and
// limitations under the License. // limitations under the License.
#include "lite/core/subgraph_bridge_registry.h"
#include "lite/kernels/apu/bridges/graph.h" #include "lite/kernels/apu/bridges/graph.h"
#include "lite/kernels/apu/bridges/utility.h" #include "lite/kernels/apu/bridges/utility.h"
#include "lite/kernels/npu/bridges/registry.h"
namespace paddle { namespace paddle {
namespace lite { namespace lite {
......
...@@ -13,9 +13,9 @@ ...@@ -13,9 +13,9 @@
// limitations under the License. // limitations under the License.
#include "lite/operators/pool_op.h" #include "lite/operators/pool_op.h"
#include "lite/core/subgraph_bridge_registry.h"
#include "lite/kernels/apu/bridges/graph.h" #include "lite/kernels/apu/bridges/graph.h"
#include "lite/kernels/apu/bridges/utility.h" #include "lite/kernels/apu/bridges/utility.h"
#include "lite/kernels/npu/bridges/registry.h"
namespace paddle { namespace paddle {
namespace lite { namespace lite {
......
...@@ -12,9 +12,9 @@ ...@@ -12,9 +12,9 @@
// See the License for the specific language governing permissions and // See the License for the specific language governing permissions and
// limitations under the License. // limitations under the License.
#include "lite/core/subgraph_bridge_registry.h"
#include "lite/kernels/apu/bridges/graph.h" #include "lite/kernels/apu/bridges/graph.h"
#include "lite/kernels/apu/bridges/utility.h" #include "lite/kernels/apu/bridges/utility.h"
#include "lite/kernels/npu/bridges/registry.h"
namespace paddle { namespace paddle {
namespace lite { namespace lite {
......
...@@ -50,7 +50,7 @@ bool SubgraphEngine::BuildDeviceProgram() { ...@@ -50,7 +50,7 @@ bool SubgraphEngine::BuildDeviceProgram() {
// Convert all of ops and their input vars and weights and added into the APU // Convert all of ops and their input vars and weights and added into the APU
// NIR graph // NIR graph
const auto& bridges = subgraph::Registry::Instance(); const auto& bridges = subgraph::SubgraphBridgeRegistry::Instance();
const auto& insts = origin_program_->instructions(kRootBlockIdx); const auto& insts = origin_program_->instructions(kRootBlockIdx);
for (auto& inst : insts) { for (auto& inst : insts) {
auto op = const_cast<OpLite*>(inst.op()); auto op = const_cast<OpLite*>(inst.op());
......
...@@ -19,15 +19,15 @@ ...@@ -19,15 +19,15 @@
#include <vector> #include <vector>
#include "NeuronAdapter.h" #include "NeuronAdapter.h"
#include "lite/core/kernel.h" #include "lite/core/kernel.h"
#include "lite/kernels/npu/bridges/engine.h" #include "lite/core/subgraph_bridge_registry.h"
#include "lite/kernels/npu/bridges/registry.h" #include "lite/core/subgraph_engine_base.h"
namespace paddle { namespace paddle {
namespace lite { namespace lite {
namespace kernels { namespace kernels {
namespace apu { namespace apu {
class SubgraphEngine : public subgraph::Engine { class SubgraphEngine : public subgraph::SubgraphEngineBase {
public: public:
SubgraphEngine(KernelContext *ctx, SubgraphEngine(KernelContext *ctx,
int block_idx, int block_idx,
...@@ -35,7 +35,7 @@ class SubgraphEngine : public subgraph::Engine { ...@@ -35,7 +35,7 @@ class SubgraphEngine : public subgraph::Engine {
Scope *exec_scope, Scope *exec_scope,
const std::vector<std::string> &input_names, const std::vector<std::string> &input_names,
const std::vector<std::string> &output_names) const std::vector<std::string> &output_names)
: subgraph::Engine(ctx, : subgraph::SubgraphEngineBase(ctx,
block_idx, block_idx,
program_desc, program_desc,
exec_scope, exec_scope,
......
...@@ -3,4 +3,4 @@ if(NOT LITE_WITH_BM) ...@@ -3,4 +3,4 @@ if(NOT LITE_WITH_BM)
endif() endif()
add_subdirectory(bridges) add_subdirectory(bridges)
add_kernel(subgraph_compute_bm BM basic SRCS subgraph_compute.cc DEPS ${lite_kernel_deps} ${bm_subgraph_bridges}) add_kernel(subgraph_compute_bm BM basic SRCS subgraph_compute.cc DEPS ${lite_kernel_deps} subgraph_engine_base ${bm_subgraph_bridges})
...@@ -5,7 +5,7 @@ endif() ...@@ -5,7 +5,7 @@ endif()
lite_cc_library(subgraph_bridge_utility_bm SRCS utility.cc DEPS) lite_cc_library(subgraph_bridge_utility_bm SRCS utility.cc DEPS)
lite_cc_library(subgraph_bridge_graph_bm SRCS graph.cc DEPS subgraph_bridge_utility_bm) lite_cc_library(subgraph_bridge_graph_bm SRCS graph.cc DEPS subgraph_bridge_utility_bm)
set(bm_subgraph_bridge_deps subgraph_bridge_registry subgraph_bridge_engine subgraph_bridge_utility_bm subgraph_bridge_graph_bm) set(bm_subgraph_bridge_deps subgraph_bridge_registry subgraph_bridge_utility_bm subgraph_bridge_graph_bm)
lite_cc_library(subgraph_bridge_act_op_bm SRCS act_op.cc DEPS ${bm_subgraph_bridge_deps}) lite_cc_library(subgraph_bridge_act_op_bm SRCS act_op.cc DEPS ${bm_subgraph_bridge_deps})
lite_cc_library(subgraph_bridge_conv_op_bm SRCS conv_op.cc DEPS ${bm_subgraph_bridge_deps}) lite_cc_library(subgraph_bridge_conv_op_bm SRCS conv_op.cc DEPS ${bm_subgraph_bridge_deps})
...@@ -40,7 +40,6 @@ lite_cc_library(subgraph_bridge_swish_op_bm SRCS swish_op.cc DEPS ${bm_subgraph_ ...@@ -40,7 +40,6 @@ lite_cc_library(subgraph_bridge_swish_op_bm SRCS swish_op.cc DEPS ${bm_subgraph_
set(bm_subgraph_bridges set(bm_subgraph_bridges
subgraph_bridge_registry subgraph_bridge_registry
subgraph_bridge_engine
subgraph_bridge_graph_bm subgraph_bridge_graph_bm
subgraph_bridge_act_op_bm subgraph_bridge_act_op_bm
subgraph_bridge_conv_op_bm subgraph_bridge_conv_op_bm
......
...@@ -15,8 +15,8 @@ ...@@ -15,8 +15,8 @@
#include <bmcompiler_if.h> #include <bmcompiler_if.h>
#include <bmcompiler_if_lite.h> #include <bmcompiler_if_lite.h>
#include <bmcompiler_op_code.h> #include <bmcompiler_op_code.h>
#include "lite/core/subgraph_bridge_registry.h"
#include "lite/kernels/bm/bridges/graph.h" #include "lite/kernels/bm/bridges/graph.h"
#include "lite/kernels/npu/bridges/registry.h"
namespace paddle { namespace paddle {
namespace lite { namespace lite {
......
...@@ -15,9 +15,9 @@ ...@@ -15,9 +15,9 @@
#include <bmcompiler_defs.h> #include <bmcompiler_defs.h>
#include <bmcompiler_if.h> #include <bmcompiler_if.h>
#include <bmcompiler_if_lite.h> #include <bmcompiler_if_lite.h>
#include "lite/core/subgraph_bridge_registry.h"
#include "lite/kernels/bm/bridges/graph.h" #include "lite/kernels/bm/bridges/graph.h"
#include "lite/kernels/bm/bridges/utility.h" #include "lite/kernels/bm/bridges/utility.h"
#include "lite/kernels/npu/bridges/registry.h"
namespace paddle { namespace paddle {
namespace lite { namespace lite {
......
...@@ -14,9 +14,9 @@ ...@@ -14,9 +14,9 @@
#include <bmcompiler_if.h> #include <bmcompiler_if.h>
#include <math.h> #include <math.h>
#include "lite/core/subgraph_bridge_registry.h"
#include "lite/kernels/bm/bridges/graph.h" #include "lite/kernels/bm/bridges/graph.h"
#include "lite/kernels/bm/bridges/utility.h" #include "lite/kernels/bm/bridges/utility.h"
#include "lite/kernels/npu/bridges/registry.h"
namespace paddle { namespace paddle {
namespace lite { namespace lite {
......
...@@ -16,9 +16,9 @@ ...@@ -16,9 +16,9 @@
#include <iostream> #include <iostream>
#include <string> #include <string>
#include <vector> #include <vector>
#include "lite/core/subgraph_bridge_registry.h"
#include "lite/kernels/bm/bridges/graph.h" #include "lite/kernels/bm/bridges/graph.h"
#include "lite/kernels/bm/bridges/utility.h" #include "lite/kernels/bm/bridges/utility.h"
#include "lite/kernels/npu/bridges/registry.h"
namespace paddle { namespace paddle {
namespace lite { namespace lite {
......
...@@ -14,9 +14,9 @@ ...@@ -14,9 +14,9 @@
#include <bmcompiler_defs.h> #include <bmcompiler_defs.h>
#include <bmcompiler_if.h> #include <bmcompiler_if.h>
#include "lite/core/subgraph_bridge_registry.h"
#include "lite/kernels/bm/bridges/graph.h" #include "lite/kernels/bm/bridges/graph.h"
#include "lite/kernels/bm/bridges/utility.h" #include "lite/kernels/bm/bridges/utility.h"
#include "lite/kernels/npu/bridges/registry.h"
namespace paddle { namespace paddle {
namespace lite { namespace lite {
......
...@@ -11,10 +11,11 @@ ...@@ -11,10 +11,11 @@
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. // WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
// See the License for the specific language governing permissions and // See the License for the specific language governing permissions and
// limitations under the License. // limitations under the License.
#include <bmcompiler_if.h> #include <bmcompiler_if.h>
#include "lite/core/subgraph_bridge_registry.h"
#include "lite/kernels/bm/bridges/graph.h" #include "lite/kernels/bm/bridges/graph.h"
#include "lite/kernels/bm/bridges/utility.h" #include "lite/kernels/bm/bridges/utility.h"
#include "lite/kernels/npu/bridges/registry.h"
namespace paddle { namespace paddle {
namespace lite { namespace lite {
......
...@@ -13,9 +13,9 @@ ...@@ -13,9 +13,9 @@
// limitations under the License. // limitations under the License.
#include <bmcompiler_if.h> #include <bmcompiler_if.h>
#include "lite/core/subgraph_bridge_registry.h"
#include "lite/kernels/bm/bridges/graph.h" #include "lite/kernels/bm/bridges/graph.h"
#include "lite/kernels/bm/bridges/utility.h" #include "lite/kernels/bm/bridges/utility.h"
#include "lite/kernels/npu/bridges/registry.h"
namespace paddle { namespace paddle {
namespace lite { namespace lite {
......
...@@ -13,9 +13,9 @@ ...@@ -13,9 +13,9 @@
// limitations under the License. // limitations under the License.
#include <bmcompiler_if.h> #include <bmcompiler_if.h>
#include "lite/core/subgraph_bridge_registry.h"
#include "lite/kernels/bm/bridges/graph.h" #include "lite/kernels/bm/bridges/graph.h"
#include "lite/kernels/bm/bridges/utility.h" #include "lite/kernels/bm/bridges/utility.h"
#include "lite/kernels/npu/bridges/registry.h"
namespace paddle { namespace paddle {
namespace lite { namespace lite {
......
...@@ -14,9 +14,9 @@ ...@@ -14,9 +14,9 @@
#include <bmcompiler_if.h> #include <bmcompiler_if.h>
#include <math.h> #include <math.h>
#include "lite/core/subgraph_bridge_registry.h"
#include "lite/kernels/bm/bridges/graph.h" #include "lite/kernels/bm/bridges/graph.h"
#include "lite/kernels/bm/bridges/utility.h" #include "lite/kernels/bm/bridges/utility.h"
#include "lite/kernels/npu/bridges/registry.h"
namespace paddle { namespace paddle {
namespace lite { namespace lite {
......
...@@ -11,11 +11,12 @@ ...@@ -11,11 +11,12 @@
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. // WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
// See the License for the specific language governing permissions and // See the License for the specific language governing permissions and
// limitations under the License. // limitations under the License.
#include <bmcompiler_if.h> #include <bmcompiler_if.h>
#include <bmcompiler_op_code.h> #include <bmcompiler_op_code.h>
#include "lite/core/subgraph_bridge_registry.h"
#include "lite/kernels/bm/bridges/graph.h" #include "lite/kernels/bm/bridges/graph.h"
#include "lite/kernels/bm/bridges/utility.h" #include "lite/kernels/bm/bridges/utility.h"
#include "lite/kernels/npu/bridges/registry.h"
namespace paddle { namespace paddle {
namespace lite { namespace lite {
......
...@@ -11,13 +11,14 @@ ...@@ -11,13 +11,14 @@
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. // WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
// See the License for the specific language governing permissions and // See the License for the specific language governing permissions and
// limitations under the License. // limitations under the License.
#include <bmcompiler_defs.h> #include <bmcompiler_defs.h>
#include <bmcompiler_if.h> #include <bmcompiler_if.h>
#include <bmcompiler_if_lite.h> #include <bmcompiler_if_lite.h>
#include <bmcompiler_op_code.h> #include <bmcompiler_op_code.h>
#include "lite/core/subgraph_bridge_registry.h"
#include "lite/kernels/bm/bridges/graph.h" #include "lite/kernels/bm/bridges/graph.h"
#include "lite/kernels/bm/bridges/utility.h" #include "lite/kernels/bm/bridges/utility.h"
#include "lite/kernels/npu/bridges/registry.h"
namespace paddle { namespace paddle {
namespace lite { namespace lite {
......
...@@ -15,9 +15,9 @@ ...@@ -15,9 +15,9 @@
#include <bmcompiler_defs.h> #include <bmcompiler_defs.h>
#include <bmcompiler_if.h> #include <bmcompiler_if.h>
#include <bmcompiler_if_lite.h> #include <bmcompiler_if_lite.h>
#include "lite/core/subgraph_bridge_registry.h"
#include "lite/kernels/bm/bridges/graph.h" #include "lite/kernels/bm/bridges/graph.h"
#include "lite/kernels/bm/bridges/utility.h" #include "lite/kernels/bm/bridges/utility.h"
#include "lite/kernels/npu/bridges/registry.h"
namespace paddle { namespace paddle {
namespace lite { namespace lite {
......
...@@ -13,8 +13,8 @@ ...@@ -13,8 +13,8 @@
// limitations under the License. // limitations under the License.
#include <bmcompiler_if.h> #include <bmcompiler_if.h>
#include "lite/core/subgraph_bridge_registry.h"
#include "lite/kernels/bm/bridges/graph.h" #include "lite/kernels/bm/bridges/graph.h"
#include "lite/kernels/npu/bridges/registry.h"
namespace paddle { namespace paddle {
namespace lite { namespace lite {
......
...@@ -14,9 +14,9 @@ ...@@ -14,9 +14,9 @@
#include <bmcompiler_if.h> #include <bmcompiler_if.h>
#include <bmcompiler_op_code.h> #include <bmcompiler_op_code.h>
#include "lite/core/subgraph_bridge_registry.h"
#include "lite/kernels/bm/bridges/graph.h" #include "lite/kernels/bm/bridges/graph.h"
#include "lite/kernels/bm/bridges/utility.h" #include "lite/kernels/bm/bridges/utility.h"
#include "lite/kernels/npu/bridges/registry.h"
namespace paddle { namespace paddle {
namespace lite { namespace lite {
......
...@@ -11,10 +11,11 @@ ...@@ -11,10 +11,11 @@
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. // WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
// See the License for the specific language governing permissions and // See the License for the specific language governing permissions and
// limitations under the License. // limitations under the License.
#include <bmcompiler_if.h> #include <bmcompiler_if.h>
#include "lite/core/subgraph_bridge_registry.h"
#include "lite/kernels/bm/bridges/graph.h" #include "lite/kernels/bm/bridges/graph.h"
#include "lite/kernels/bm/bridges/utility.h" #include "lite/kernels/bm/bridges/utility.h"
#include "lite/kernels/npu/bridges/registry.h"
namespace paddle { namespace paddle {
namespace lite { namespace lite {
......
...@@ -14,9 +14,9 @@ ...@@ -14,9 +14,9 @@
#include <bmcompiler_if.h> #include <bmcompiler_if.h>
#include <user_bmcpu_common.h> #include <user_bmcpu_common.h>
#include "lite/core/subgraph_bridge_registry.h"
#include "lite/kernels/bm/bridges/graph.h" #include "lite/kernels/bm/bridges/graph.h"
#include "lite/kernels/bm/bridges/utility.h" #include "lite/kernels/bm/bridges/utility.h"
#include "lite/kernels/npu/bridges/registry.h"
namespace paddle { namespace paddle {
namespace lite { namespace lite {
......
...@@ -13,9 +13,9 @@ ...@@ -13,9 +13,9 @@
// limitations under the License. // limitations under the License.
#include <bmcompiler_if.h> #include <bmcompiler_if.h>
#include "lite/core/subgraph_bridge_registry.h"
#include "lite/kernels/bm/bridges/graph.h" #include "lite/kernels/bm/bridges/graph.h"
#include "lite/kernels/bm/bridges/utility.h" #include "lite/kernels/bm/bridges/utility.h"
#include "lite/kernels/npu/bridges/registry.h"
namespace paddle { namespace paddle {
namespace lite { namespace lite {
......
...@@ -11,13 +11,14 @@ ...@@ -11,13 +11,14 @@
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. // WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
// See the License for the specific language governing permissions and // See the License for the specific language governing permissions and
// limitations under the License. // limitations under the License.
#include <bmcompiler_defs.h> #include <bmcompiler_defs.h>
#include <bmcompiler_if.h> #include <bmcompiler_if.h>
#include <bmcompiler_if_lite.h> #include <bmcompiler_if_lite.h>
#include <user_bmcpu_common.h> #include <user_bmcpu_common.h>
#include "lite/core/subgraph_bridge_registry.h"
#include "lite/kernels/bm/bridges/graph.h" #include "lite/kernels/bm/bridges/graph.h"
#include "lite/kernels/bm/bridges/utility.h" #include "lite/kernels/bm/bridges/utility.h"
#include "lite/kernels/npu/bridges/registry.h"
namespace paddle { namespace paddle {
namespace lite { namespace lite {
......
...@@ -13,9 +13,9 @@ ...@@ -13,9 +13,9 @@
// limitations under the License. // limitations under the License.
#include <bmcompiler_if.h> #include <bmcompiler_if.h>
#include "lite/core/subgraph_bridge_registry.h"
#include "lite/kernels/bm/bridges/graph.h" #include "lite/kernels/bm/bridges/graph.h"
#include "lite/kernels/bm/bridges/utility.h" #include "lite/kernels/bm/bridges/utility.h"
#include "lite/kernels/npu/bridges/registry.h"
namespace paddle { namespace paddle {
namespace lite { namespace lite {
......
...@@ -14,9 +14,9 @@ ...@@ -14,9 +14,9 @@
#include <bmcompiler_if.h> #include <bmcompiler_if.h>
#include <bmcompiler_op_code.h> #include <bmcompiler_op_code.h>
#include "lite/core/subgraph_bridge_registry.h"
#include "lite/kernels/bm/bridges/graph.h" #include "lite/kernels/bm/bridges/graph.h"
#include "lite/kernels/bm/bridges/utility.h" #include "lite/kernels/bm/bridges/utility.h"
#include "lite/kernels/npu/bridges/registry.h"
namespace paddle { namespace paddle {
namespace lite { namespace lite {
......
...@@ -13,8 +13,8 @@ ...@@ -13,8 +13,8 @@
// limitations under the License. // limitations under the License.
#include <bmcompiler_if.h> #include <bmcompiler_if.h>
#include "lite/core/subgraph_bridge_registry.h"
#include "lite/kernels/bm/bridges/graph.h" #include "lite/kernels/bm/bridges/graph.h"
#include "lite/kernels/npu/bridges/registry.h"
namespace paddle { namespace paddle {
namespace lite { namespace lite {
......
...@@ -14,9 +14,9 @@ ...@@ -14,9 +14,9 @@
#include <bmcompiler_if.h> #include <bmcompiler_if.h>
#include <bmcompiler_op_code.h> #include <bmcompiler_op_code.h>
#include "lite/core/subgraph_bridge_registry.h"
#include "lite/kernels/bm/bridges/graph.h" #include "lite/kernels/bm/bridges/graph.h"
#include "lite/kernels/bm/bridges/utility.h" #include "lite/kernels/bm/bridges/utility.h"
#include "lite/kernels/npu/bridges/registry.h"
namespace paddle { namespace paddle {
namespace lite { namespace lite {
......
...@@ -15,9 +15,9 @@ ...@@ -15,9 +15,9 @@
#include <bmcompiler_defs.h> #include <bmcompiler_defs.h>
#include <bmcompiler_if.h> #include <bmcompiler_if.h>
#include <bmcompiler_if_lite.h> #include <bmcompiler_if_lite.h>
#include "lite/core/subgraph_bridge_registry.h"
#include "lite/kernels/bm/bridges/graph.h" #include "lite/kernels/bm/bridges/graph.h"
#include "lite/kernels/bm/bridges/utility.h" #include "lite/kernels/bm/bridges/utility.h"
#include "lite/kernels/npu/bridges/registry.h"
namespace paddle { namespace paddle {
namespace lite { namespace lite {
......
...@@ -14,9 +14,9 @@ ...@@ -14,9 +14,9 @@
#include <bmcompiler_if.h> #include <bmcompiler_if.h>
#include <bmcompiler_op_code.h> #include <bmcompiler_op_code.h>
#include "lite/core/subgraph_bridge_registry.h"
#include "lite/kernels/bm/bridges/graph.h" #include "lite/kernels/bm/bridges/graph.h"
#include "lite/kernels/bm/bridges/utility.h" #include "lite/kernels/bm/bridges/utility.h"
#include "lite/kernels/npu/bridges/registry.h"
namespace paddle { namespace paddle {
namespace lite { namespace lite {
......
...@@ -11,10 +11,11 @@ ...@@ -11,10 +11,11 @@
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. // WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
// See the License for the specific language governing permissions and // See the License for the specific language governing permissions and
// limitations under the License. // limitations under the License.
#include <bmcompiler_if.h> #include <bmcompiler_if.h>
#include "lite/core/subgraph_bridge_registry.h"
#include "lite/kernels/bm/bridges/graph.h" #include "lite/kernels/bm/bridges/graph.h"
#include "lite/kernels/bm/bridges/utility.h" #include "lite/kernels/bm/bridges/utility.h"
#include "lite/kernels/npu/bridges/registry.h"
namespace paddle { namespace paddle {
namespace lite { namespace lite {
......
...@@ -14,9 +14,9 @@ ...@@ -14,9 +14,9 @@
#include <bmcompiler_if.h> #include <bmcompiler_if.h>
#include <bmcompiler_op_code.h> #include <bmcompiler_op_code.h>
#include "lite/core/subgraph_bridge_registry.h"
#include "lite/kernels/bm/bridges/graph.h" #include "lite/kernels/bm/bridges/graph.h"
#include "lite/kernels/bm/bridges/utility.h" #include "lite/kernels/bm/bridges/utility.h"
#include "lite/kernels/npu/bridges/registry.h"
namespace paddle { namespace paddle {
namespace lite { namespace lite {
......
...@@ -14,9 +14,9 @@ ...@@ -14,9 +14,9 @@
#include <bmcompiler_if.h> #include <bmcompiler_if.h>
#include <bmcompiler_op_code.h> #include <bmcompiler_op_code.h>
#include "lite/core/subgraph_bridge_registry.h"
#include "lite/kernels/bm/bridges/graph.h" #include "lite/kernels/bm/bridges/graph.h"
#include "lite/kernels/bm/bridges/utility.h" #include "lite/kernels/bm/bridges/utility.h"
#include "lite/kernels/npu/bridges/registry.h"
namespace paddle { namespace paddle {
namespace lite { namespace lite {
......
...@@ -11,11 +11,12 @@ ...@@ -11,11 +11,12 @@
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. // WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
// See the License for the specific language governing permissions and // See the License for the specific language governing permissions and
// limitations under the License. // limitations under the License.
#include <bmcompiler_if.h> #include <bmcompiler_if.h>
#include <bmcompiler_op_code.h> #include <bmcompiler_op_code.h>
#include "lite/core/subgraph_bridge_registry.h"
#include "lite/kernels/bm/bridges/graph.h" #include "lite/kernels/bm/bridges/graph.h"
#include "lite/kernels/bm/bridges/utility.h" #include "lite/kernels/bm/bridges/utility.h"
#include "lite/kernels/npu/bridges/registry.h"
namespace paddle { namespace paddle {
namespace lite { namespace lite {
......
...@@ -14,9 +14,9 @@ ...@@ -14,9 +14,9 @@
#include <bmcompiler_defs.h> #include <bmcompiler_defs.h>
#include <bmcompiler_if.h> #include <bmcompiler_if.h>
#include "lite/core/subgraph_bridge_registry.h"
#include "lite/kernels/bm/bridges/graph.h" #include "lite/kernels/bm/bridges/graph.h"
#include "lite/kernels/bm/bridges/utility.h" #include "lite/kernels/bm/bridges/utility.h"
#include "lite/kernels/npu/bridges/registry.h"
namespace paddle { namespace paddle {
namespace lite { namespace lite {
......
...@@ -17,9 +17,9 @@ ...@@ -17,9 +17,9 @@
#include <iostream> #include <iostream>
#include <string> #include <string>
#include <vector> #include <vector>
#include "lite/core/subgraph_bridge_registry.h"
#include "lite/kernels/bm/bridges/graph.h" #include "lite/kernels/bm/bridges/graph.h"
#include "lite/kernels/bm/bridges/utility.h" #include "lite/kernels/bm/bridges/utility.h"
#include "lite/kernels/npu/bridges/registry.h"
namespace paddle { namespace paddle {
namespace lite { namespace lite {
......
...@@ -31,7 +31,7 @@ namespace bm { ...@@ -31,7 +31,7 @@ namespace bm {
bool SubgraphEngine::BuildDeviceProgram() { bool SubgraphEngine::BuildDeviceProgram() {
int status = 0; int status = 0;
subgraph::bm::Graph graph; subgraph::bm::Graph graph;
const auto& bridges = subgraph::Registry::Instance(); const auto& bridges = subgraph::SubgraphBridgeRegistry::Instance();
graph.CreateCompilerHandle(); graph.CreateCompilerHandle();
auto& ctx = this->ctx_->template As<BMContext>(); auto& ctx = this->ctx_->template As<BMContext>();
for (size_t i = 0; i < input_names_.size(); i++) { for (size_t i = 0; i < input_names_.size(); i++) {
......
...@@ -23,16 +23,16 @@ ...@@ -23,16 +23,16 @@
#include "lite/core/kernel.h" #include "lite/core/kernel.h"
#include "lite/core/op_registry.h" #include "lite/core/op_registry.h"
#include "lite/core/program.h" #include "lite/core/program.h"
#include "lite/core/subgraph_bridge_registry.h"
#include "lite/core/subgraph_engine_base.h"
#include "lite/core/types.h" #include "lite/core/types.h"
#include "lite/kernels/npu/bridges/engine.h"
#include "lite/kernels/npu/bridges/registry.h"
namespace paddle { namespace paddle {
namespace lite { namespace lite {
namespace kernels { namespace kernels {
namespace bm { namespace bm {
class SubgraphEngine : public subgraph::Engine { class SubgraphEngine : public subgraph::SubgraphEngineBase {
public: public:
SubgraphEngine(KernelContext *ctx, SubgraphEngine(KernelContext *ctx,
int block_idx, int block_idx,
...@@ -40,7 +40,7 @@ class SubgraphEngine : public subgraph::Engine { ...@@ -40,7 +40,7 @@ class SubgraphEngine : public subgraph::Engine {
Scope *exec_scope, Scope *exec_scope,
const std::vector<std::string> &input_names, const std::vector<std::string> &input_names,
const std::vector<std::string> &output_names) const std::vector<std::string> &output_names)
: subgraph::Engine(ctx, : subgraph::SubgraphEngineBase(ctx,
block_idx, block_idx,
program_desc, program_desc,
exec_scope, exec_scope,
......
add_subdirectory(bridges) add_subdirectory(bridges)
add_kernel(subgraph_compute_huawei_ascend_npu HUAWEI_ASCEND_NPU basic SRCS subgraph_compute.cc DEPS ${lite_kernel_deps} device_huawei_ascend_npu subgraph_bridge_engine ${huawei_ascend_npu_subgraph_bridges}) add_kernel(subgraph_compute_huawei_ascend_npu HUAWEI_ASCEND_NPU basic SRCS subgraph_compute.cc DEPS ${lite_kernel_deps} device_huawei_ascend_npu subgraph_engine_base ${huawei_ascend_npu_subgraph_bridges})
...@@ -5,7 +5,7 @@ endif() ...@@ -5,7 +5,7 @@ endif()
lite_cc_library(subgraph_bridge_utility_huawei_ascend_npu SRCS utility.cc DEPS) lite_cc_library(subgraph_bridge_utility_huawei_ascend_npu SRCS utility.cc DEPS)
lite_cc_library(subgraph_bridge_graph_huawei_ascend_npu SRCS graph.cc DEPS subgraph_bridge_utility_huawei_ascend_npu) lite_cc_library(subgraph_bridge_graph_huawei_ascend_npu SRCS graph.cc DEPS subgraph_bridge_utility_huawei_ascend_npu)
set(huawei_ascend_npu_subgraph_bridge_deps subgraph_bridge_registry subgraph_bridge_engine subgraph_bridge_utility_huawei_ascend_npu subgraph_bridge_graph_huawei_ascend_npu) set(huawei_ascend_npu_subgraph_bridge_deps subgraph_bridge_registry subgraph_bridge_utility_huawei_ascend_npu subgraph_bridge_graph_huawei_ascend_npu)
lite_cc_library(subgraph_bridge_act_op_huawei_ascend_npu SRCS act_op.cc DEPS ${huawei_ascend_npu_subgraph_bridge_deps}) lite_cc_library(subgraph_bridge_act_op_huawei_ascend_npu SRCS act_op.cc DEPS ${huawei_ascend_npu_subgraph_bridge_deps})
lite_cc_library(subgraph_bridge_conv_op_huawei_ascend_npu SRCS conv_op.cc DEPS ${huawei_ascend_npu_subgraph_bridge_deps}) lite_cc_library(subgraph_bridge_conv_op_huawei_ascend_npu SRCS conv_op.cc DEPS ${huawei_ascend_npu_subgraph_bridge_deps})
...@@ -29,7 +29,6 @@ lite_cc_library(subgraph_bridge_gather_op_huawei_ascend_npu SRCS gather_op.cc DE ...@@ -29,7 +29,6 @@ lite_cc_library(subgraph_bridge_gather_op_huawei_ascend_npu SRCS gather_op.cc DE
set(huawei_ascend_npu_subgraph_bridges set(huawei_ascend_npu_subgraph_bridges
subgraph_bridge_registry subgraph_bridge_registry
subgraph_bridge_engine
subgraph_bridge_graph_huawei_ascend_npu subgraph_bridge_graph_huawei_ascend_npu
subgraph_bridge_act_op_huawei_ascend_npu subgraph_bridge_act_op_huawei_ascend_npu
subgraph_bridge_conv_op_huawei_ascend_npu subgraph_bridge_conv_op_huawei_ascend_npu
......
...@@ -12,9 +12,9 @@ ...@@ -12,9 +12,9 @@
// See the License for the specific language governing permissions and // See the License for the specific language governing permissions and
// limitations under the License. // limitations under the License.
#include "lite/core/subgraph_bridge_registry.h"
#include "lite/kernels/huawei_ascend_npu/bridges/graph.h" #include "lite/kernels/huawei_ascend_npu/bridges/graph.h"
#include "lite/kernels/huawei_ascend_npu/bridges/utility.h" #include "lite/kernels/huawei_ascend_npu/bridges/utility.h"
#include "lite/kernels/npu/bridges/registry.h"
namespace paddle { namespace paddle {
namespace lite { namespace lite {
......
...@@ -12,9 +12,9 @@ ...@@ -12,9 +12,9 @@
// See the License for the specific language governing permissions and // See the License for the specific language governing permissions and
// limitations under the License. // limitations under the License.
#include "lite/core/subgraph_bridge_registry.h"
#include "lite/kernels/huawei_ascend_npu/bridges/graph.h" #include "lite/kernels/huawei_ascend_npu/bridges/graph.h"
#include "lite/kernels/huawei_ascend_npu/bridges/utility.h" #include "lite/kernels/huawei_ascend_npu/bridges/utility.h"
#include "lite/kernels/npu/bridges/registry.h"
namespace paddle { namespace paddle {
namespace lite { namespace lite {
......
...@@ -12,9 +12,9 @@ ...@@ -12,9 +12,9 @@
// See the License for the specific language governing permissions and // See the License for the specific language governing permissions and
// limitations under the License. // limitations under the License.
#include "lite/core/subgraph_bridge_registry.h"
#include "lite/kernels/huawei_ascend_npu/bridges/graph.h" #include "lite/kernels/huawei_ascend_npu/bridges/graph.h"
#include "lite/kernels/huawei_ascend_npu/bridges/utility.h" #include "lite/kernels/huawei_ascend_npu/bridges/utility.h"
#include "lite/kernels/npu/bridges/registry.h"
namespace paddle { namespace paddle {
namespace lite { namespace lite {
......
...@@ -12,9 +12,9 @@ ...@@ -12,9 +12,9 @@
// See the License for the specific language governing permissions and // See the License for the specific language governing permissions and
// limitations under the License. // limitations under the License.
#include "lite/core/subgraph_bridge_registry.h"
#include "lite/kernels/huawei_ascend_npu/bridges/graph.h" #include "lite/kernels/huawei_ascend_npu/bridges/graph.h"
#include "lite/kernels/huawei_ascend_npu/bridges/utility.h" #include "lite/kernels/huawei_ascend_npu/bridges/utility.h"
#include "lite/kernels/npu/bridges/registry.h"
namespace paddle { namespace paddle {
namespace lite { namespace lite {
......
...@@ -13,9 +13,9 @@ ...@@ -13,9 +13,9 @@
// limitations under the License. // limitations under the License.
#include "lite/operators/conv_op.h" #include "lite/operators/conv_op.h"
#include "lite/core/subgraph_bridge_registry.h"
#include "lite/kernels/huawei_ascend_npu/bridges/graph.h" #include "lite/kernels/huawei_ascend_npu/bridges/graph.h"
#include "lite/kernels/huawei_ascend_npu/bridges/utility.h" #include "lite/kernels/huawei_ascend_npu/bridges/utility.h"
#include "lite/kernels/npu/bridges/registry.h"
namespace paddle { namespace paddle {
namespace lite { namespace lite {
......
...@@ -12,9 +12,9 @@ ...@@ -12,9 +12,9 @@
// See the License for the specific language governing permissions and // See the License for the specific language governing permissions and
// limitations under the License. // limitations under the License.
#include "lite/core/subgraph_bridge_registry.h"
#include "lite/kernels/huawei_ascend_npu/bridges/graph.h" #include "lite/kernels/huawei_ascend_npu/bridges/graph.h"
#include "lite/kernels/huawei_ascend_npu/bridges/utility.h" #include "lite/kernels/huawei_ascend_npu/bridges/utility.h"
#include "lite/kernels/npu/bridges/registry.h"
namespace paddle { namespace paddle {
namespace lite { namespace lite {
......
...@@ -12,9 +12,9 @@ ...@@ -12,9 +12,9 @@
// See the License for the specific language governing permissions and // See the License for the specific language governing permissions and
// limitations under the License. // limitations under the License.
#include "lite/core/subgraph_bridge_registry.h"
#include "lite/kernels/huawei_ascend_npu/bridges/graph.h" #include "lite/kernels/huawei_ascend_npu/bridges/graph.h"
#include "lite/kernels/huawei_ascend_npu/bridges/utility.h" #include "lite/kernels/huawei_ascend_npu/bridges/utility.h"
#include "lite/kernels/npu/bridges/registry.h"
namespace paddle { namespace paddle {
namespace lite { namespace lite {
......
...@@ -12,9 +12,9 @@ ...@@ -12,9 +12,9 @@
// See the License for the specific language governing permissions and // See the License for the specific language governing permissions and
// limitations under the License. // limitations under the License.
#include "lite/core/subgraph_bridge_registry.h"
#include "lite/kernels/huawei_ascend_npu/bridges/graph.h" #include "lite/kernels/huawei_ascend_npu/bridges/graph.h"
#include "lite/kernels/huawei_ascend_npu/bridges/utility.h" #include "lite/kernels/huawei_ascend_npu/bridges/utility.h"
#include "lite/kernels/npu/bridges/registry.h"
namespace paddle { namespace paddle {
namespace lite { namespace lite {
......
...@@ -12,9 +12,9 @@ ...@@ -12,9 +12,9 @@
// See the License for the specific language governing permissions and // See the License for the specific language governing permissions and
// limitations under the License. // limitations under the License.
#include "lite/core/subgraph_bridge_registry.h"
#include "lite/kernels/huawei_ascend_npu/bridges/graph.h" #include "lite/kernels/huawei_ascend_npu/bridges/graph.h"
#include "lite/kernels/huawei_ascend_npu/bridges/utility.h" #include "lite/kernels/huawei_ascend_npu/bridges/utility.h"
#include "lite/kernels/npu/bridges/registry.h"
namespace paddle { namespace paddle {
namespace lite { namespace lite {
......
...@@ -12,9 +12,9 @@ ...@@ -12,9 +12,9 @@
// See the License for the specific language governing permissions and // See the License for the specific language governing permissions and
// limitations under the License. // limitations under the License.
#include "lite/core/subgraph_bridge_registry.h"
#include "lite/kernels/huawei_ascend_npu/bridges/graph.h" #include "lite/kernels/huawei_ascend_npu/bridges/graph.h"
#include "lite/kernels/huawei_ascend_npu/bridges/utility.h" #include "lite/kernels/huawei_ascend_npu/bridges/utility.h"
#include "lite/kernels/npu/bridges/registry.h"
namespace paddle { namespace paddle {
namespace lite { namespace lite {
......
...@@ -13,9 +13,9 @@ ...@@ -13,9 +13,9 @@
// limitations under the License. // limitations under the License.
// #include "lite/operators/interpolate_op.h" // #include "lite/operators/interpolate_op.h"
#include "lite/core/subgraph_bridge_registry.h"
#include "lite/kernels/huawei_ascend_npu/bridges/graph.h" #include "lite/kernels/huawei_ascend_npu/bridges/graph.h"
#include "lite/kernels/huawei_ascend_npu/bridges/utility.h" #include "lite/kernels/huawei_ascend_npu/bridges/utility.h"
#include "lite/kernels/npu/bridges/registry.h"
namespace paddle { namespace paddle {
namespace lite { namespace lite {
......
...@@ -12,9 +12,9 @@ ...@@ -12,9 +12,9 @@
// See the License for the specific language governing permissions and // See the License for the specific language governing permissions and
// limitations under the License. // limitations under the License.
#include "lite/core/subgraph_bridge_registry.h"
#include "lite/kernels/huawei_ascend_npu/bridges/graph.h" #include "lite/kernels/huawei_ascend_npu/bridges/graph.h"
#include "lite/kernels/huawei_ascend_npu/bridges/utility.h" #include "lite/kernels/huawei_ascend_npu/bridges/utility.h"
#include "lite/kernels/npu/bridges/registry.h"
namespace paddle { namespace paddle {
namespace lite { namespace lite {
......
...@@ -12,9 +12,9 @@ ...@@ -12,9 +12,9 @@
// See the License for the specific language governing permissions and // See the License for the specific language governing permissions and
// limitations under the License. // limitations under the License.
#include "lite/core/subgraph_bridge_registry.h"
#include "lite/kernels/huawei_ascend_npu/bridges/graph.h" #include "lite/kernels/huawei_ascend_npu/bridges/graph.h"
#include "lite/kernels/huawei_ascend_npu/bridges/utility.h" #include "lite/kernels/huawei_ascend_npu/bridges/utility.h"
#include "lite/kernels/npu/bridges/registry.h"
namespace paddle { namespace paddle {
namespace lite { namespace lite {
......
...@@ -13,9 +13,9 @@ ...@@ -13,9 +13,9 @@
// limitations under the License. // limitations under the License.
#include "lite/operators/pool_op.h" #include "lite/operators/pool_op.h"
#include "lite/core/subgraph_bridge_registry.h"
#include "lite/kernels/huawei_ascend_npu/bridges/graph.h" #include "lite/kernels/huawei_ascend_npu/bridges/graph.h"
#include "lite/kernels/huawei_ascend_npu/bridges/utility.h" #include "lite/kernels/huawei_ascend_npu/bridges/utility.h"
#include "lite/kernels/npu/bridges/registry.h"
namespace paddle { namespace paddle {
namespace lite { namespace lite {
......
...@@ -13,9 +13,9 @@ ...@@ -13,9 +13,9 @@
// limitations under the License. // limitations under the License.
#include "lite/operators/reshape_op.h" #include "lite/operators/reshape_op.h"
#include "lite/core/subgraph_bridge_registry.h"
#include "lite/kernels/huawei_ascend_npu/bridges/graph.h" #include "lite/kernels/huawei_ascend_npu/bridges/graph.h"
#include "lite/kernels/huawei_ascend_npu/bridges/utility.h" #include "lite/kernels/huawei_ascend_npu/bridges/utility.h"
#include "lite/kernels/npu/bridges/registry.h"
namespace paddle { namespace paddle {
namespace lite { namespace lite {
......
...@@ -12,9 +12,9 @@ ...@@ -12,9 +12,9 @@
// See the License for the specific language governing permissions and // See the License for the specific language governing permissions and
// limitations under the License. // limitations under the License.
#include "lite/core/subgraph_bridge_registry.h"
#include "lite/kernels/huawei_ascend_npu/bridges/graph.h" #include "lite/kernels/huawei_ascend_npu/bridges/graph.h"
#include "lite/kernels/huawei_ascend_npu/bridges/utility.h" #include "lite/kernels/huawei_ascend_npu/bridges/utility.h"
#include "lite/kernels/npu/bridges/registry.h"
namespace paddle { namespace paddle {
namespace lite { namespace lite {
......
...@@ -12,9 +12,9 @@ ...@@ -12,9 +12,9 @@
// See the License for the specific language governing permissions and // See the License for the specific language governing permissions and
// limitations under the License. // limitations under the License.
#include "lite/core/subgraph_bridge_registry.h"
#include "lite/kernels/huawei_ascend_npu/bridges/graph.h" #include "lite/kernels/huawei_ascend_npu/bridges/graph.h"
#include "lite/kernels/huawei_ascend_npu/bridges/utility.h" #include "lite/kernels/huawei_ascend_npu/bridges/utility.h"
#include "lite/kernels/npu/bridges/registry.h"
namespace paddle { namespace paddle {
namespace lite { namespace lite {
......
...@@ -12,9 +12,9 @@ ...@@ -12,9 +12,9 @@
// See the License for the specific language governing permissions and // See the License for the specific language governing permissions and
// limitations under the License. // limitations under the License.
#include "lite/core/subgraph_bridge_registry.h"
#include "lite/kernels/huawei_ascend_npu/bridges/graph.h" #include "lite/kernels/huawei_ascend_npu/bridges/graph.h"
#include "lite/kernels/huawei_ascend_npu/bridges/utility.h" #include "lite/kernels/huawei_ascend_npu/bridges/utility.h"
#include "lite/kernels/npu/bridges/registry.h"
namespace paddle { namespace paddle {
namespace lite { namespace lite {
......
...@@ -12,9 +12,9 @@ ...@@ -12,9 +12,9 @@
// See the License for the specific language governing permissions and // See the License for the specific language governing permissions and
// limitations under the License. // limitations under the License.
#include "lite/core/subgraph_bridge_registry.h"
#include "lite/kernels/huawei_ascend_npu/bridges/graph.h" #include "lite/kernels/huawei_ascend_npu/bridges/graph.h"
#include "lite/kernels/huawei_ascend_npu/bridges/utility.h" #include "lite/kernels/huawei_ascend_npu/bridges/utility.h"
#include "lite/kernels/npu/bridges/registry.h"
namespace paddle { namespace paddle {
namespace lite { namespace lite {
......
...@@ -119,7 +119,7 @@ bool DeviceProgram::BuildGraphAndCacheToFile( ...@@ -119,7 +119,7 @@ bool DeviceProgram::BuildGraphAndCacheToFile(
// then added them into the IR graph // then added them into the IR graph
int status = 0; int status = 0;
subgraph::huawei_ascend_npu::Graph graph; subgraph::huawei_ascend_npu::Graph graph;
const auto& bridges = subgraph::Registry::Instance(); const auto& bridges = subgraph::SubgraphBridgeRegistry::Instance();
CHECK(origin_program) CHECK(origin_program)
<< "[HUAWEI_ASCEND_NPU] The origin program is not initialized!"; << "[HUAWEI_ASCEND_NPU] The origin program is not initialized!";
CHECK_GT(origin_program->instructions(kRootBlockIdx).size(), 0) CHECK_GT(origin_program->instructions(kRootBlockIdx).size(), 0)
......
...@@ -21,8 +21,8 @@ ...@@ -21,8 +21,8 @@
#include "graph/tensor.h" #include "graph/tensor.h"
#include "lite/backends/huawei_ascend_npu/device.h" #include "lite/backends/huawei_ascend_npu/device.h"
#include "lite/core/kernel.h" #include "lite/core/kernel.h"
#include "lite/kernels/npu/bridges/engine.h" #include "lite/core/subgraph_bridge_registry.h"
#include "lite/kernels/npu/bridges/registry.h" #include "lite/core/subgraph_engine_base.h"
namespace paddle { namespace paddle {
namespace lite { namespace lite {
...@@ -76,7 +76,7 @@ class DeviceProgram { ...@@ -76,7 +76,7 @@ class DeviceProgram {
std::vector<TensorDesc> device_odims_{}; std::vector<TensorDesc> device_odims_{};
}; };
class SubgraphEngine : public subgraph::Engine { class SubgraphEngine : public subgraph::SubgraphEngineBase {
public: public:
SubgraphEngine(KernelContext* ctx, SubgraphEngine(KernelContext* ctx,
int block_idx, int block_idx,
...@@ -84,7 +84,7 @@ class SubgraphEngine : public subgraph::Engine { ...@@ -84,7 +84,7 @@ class SubgraphEngine : public subgraph::Engine {
Scope* exec_scope, Scope* exec_scope,
const std::vector<std::string>& input_names, const std::vector<std::string>& input_names,
const std::vector<std::string>& output_names) const std::vector<std::string>& output_names)
: subgraph::Engine(ctx, : subgraph::SubgraphEngineBase(ctx,
block_idx, block_idx,
program_desc, program_desc,
exec_scope, exec_scope,
......
...@@ -3,7 +3,7 @@ if(NOT LITE_WITH_MLU) ...@@ -3,7 +3,7 @@ if(NOT LITE_WITH_MLU)
endif() endif()
add_subdirectory(bridges) add_subdirectory(bridges)
add_kernel(subgraph_compute_mlu MLU basic SRCS subgraph_compute.cc DEPS ${lite_kernel_deps} ${mlu_subgraph_bridges}) add_kernel(subgraph_compute_mlu MLU basic SRCS subgraph_compute.cc DEPS ${lite_kernel_deps} subgraph_engine_base ${mlu_subgraph_bridges})
add_kernel(io_copy_compute_mlu MLU basic SRCS io_copy_compute.cc DEPS ${lite_kernel_deps} ${target_wrapper_mlu}) add_kernel(io_copy_compute_mlu MLU basic SRCS io_copy_compute.cc DEPS ${lite_kernel_deps} ${target_wrapper_mlu})
add_kernel(calib_compute_mlu MLU basic SRCS calib_compute.cc DEPS ${lite_kernel_deps}) add_kernel(calib_compute_mlu MLU basic SRCS calib_compute.cc DEPS ${lite_kernel_deps})
# depend on transpose function in backend/x86/math/math_function # depend on transpose function in backend/x86/math/math_function
......
...@@ -6,7 +6,7 @@ lite_cc_library(subgraph_bridge_utility_mlu SRCS utility.cc DEPS ${mlu_builder_l ...@@ -6,7 +6,7 @@ lite_cc_library(subgraph_bridge_utility_mlu SRCS utility.cc DEPS ${mlu_builder_l
lite_cc_library(subgraph_bridge_tensor_mlu SRCS tensor.cc DEPS ${mlu_builder_libs} subgraph_bridge_utility_mlu) lite_cc_library(subgraph_bridge_tensor_mlu SRCS tensor.cc DEPS ${mlu_builder_libs} subgraph_bridge_utility_mlu)
lite_cc_library(subgraph_bridge_graph_mlu SRCS graph.cc DEPS subgraph_bridge_utility_mlu subgraph_bridge_tensor_mlu) lite_cc_library(subgraph_bridge_graph_mlu SRCS graph.cc DEPS subgraph_bridge_utility_mlu subgraph_bridge_tensor_mlu)
set(mlu_subgraph_bridge_deps subgraph_bridge_registry subgraph_bridge_engine subgraph_bridge_utility_mlu subgraph_bridge_graph_mlu) set(mlu_subgraph_bridge_deps subgraph_bridge_registry subgraph_bridge_utility_mlu subgraph_bridge_graph_mlu)
lite_cc_library(subgraph_bridge_act_op_mlu SRCS act_op.cc DEPS ${mlu_subgraph_bridge_deps}) lite_cc_library(subgraph_bridge_act_op_mlu SRCS act_op.cc DEPS ${mlu_subgraph_bridge_deps})
lite_cc_library(subgraph_bridge_batch_norm_op_mlu SRCS batch_norm_op.cc DEPS ${mlu_subgraph_bridge_deps}) lite_cc_library(subgraph_bridge_batch_norm_op_mlu SRCS batch_norm_op.cc DEPS ${mlu_subgraph_bridge_deps})
......
...@@ -12,9 +12,9 @@ ...@@ -12,9 +12,9 @@
// See the License for the specific language governing permissions and // See the License for the specific language governing permissions and
// limitations under the License. // limitations under the License.
#include "lite/core/subgraph_bridge_registry.h"
#include "lite/kernels/mlu/bridges/graph.h" #include "lite/kernels/mlu/bridges/graph.h"
#include "lite/kernels/mlu/bridges/utility.h" #include "lite/kernels/mlu/bridges/utility.h"
#include "lite/kernels/npu/bridges/registry.h"
namespace paddle { namespace paddle {
namespace lite { namespace lite {
......
...@@ -18,8 +18,8 @@ ...@@ -18,8 +18,8 @@
#include "lite/core/op_lite.h" #include "lite/core/op_lite.h"
#include "lite/core/op_registry.h" #include "lite/core/op_registry.h"
#include "lite/core/subgraph_bridge_registry.h"
#include "lite/kernels/mlu/bridges/test_helper.h" #include "lite/kernels/mlu/bridges/test_helper.h"
#include "lite/kernels/npu/bridges/registry.h"
#include "lite/operators/activation_ops.h" #include "lite/operators/activation_ops.h"
namespace paddle { namespace paddle {
......
...@@ -12,9 +12,9 @@ ...@@ -12,9 +12,9 @@
// See the License for the specific language governing permissions and // See the License for the specific language governing permissions and
// limitations under the License. // limitations under the License.
#include "lite/core/subgraph_bridge_registry.h"
#include "lite/kernels/mlu/bridges/graph.h" #include "lite/kernels/mlu/bridges/graph.h"
#include "lite/kernels/mlu/bridges/utility.h" #include "lite/kernels/mlu/bridges/utility.h"
#include "lite/kernels/npu/bridges/registry.h"
namespace paddle { namespace paddle {
namespace lite { namespace lite {
......
...@@ -20,9 +20,9 @@ ...@@ -20,9 +20,9 @@
#include <iostream> #include <iostream>
#include "lite/core/op_registry.h" #include "lite/core/op_registry.h"
#include "lite/core/subgraph_bridge_registry.h"
#include "lite/kernels/mlu/bridges/test_helper.h" #include "lite/kernels/mlu/bridges/test_helper.h"
#include "lite/kernels/mlu/bridges/utility.h" #include "lite/kernels/mlu/bridges/utility.h"
#include "lite/kernels/npu/bridges/registry.h"
namespace paddle { namespace paddle {
namespace lite { namespace lite {
namespace subgraph { namespace subgraph {
......
...@@ -12,9 +12,9 @@ ...@@ -12,9 +12,9 @@
// See the License for the specific language governing permissions and // See the License for the specific language governing permissions and
// limitations under the License. // limitations under the License.
#include "lite/core/subgraph_bridge_registry.h"
#include "lite/kernels/mlu/bridges/graph.h" #include "lite/kernels/mlu/bridges/graph.h"
#include "lite/kernels/mlu/bridges/utility.h" #include "lite/kernels/mlu/bridges/utility.h"
#include "lite/kernels/npu/bridges/registry.h"
namespace paddle { namespace paddle {
namespace lite { namespace lite {
......
...@@ -15,8 +15,8 @@ ...@@ -15,8 +15,8 @@
#include "lite/operators/batch_norm_op.h" #include "lite/operators/batch_norm_op.h"
#include <gtest/gtest.h> #include <gtest/gtest.h>
#include "lite/core/op_registry.h" #include "lite/core/op_registry.h"
#include "lite/core/subgraph_bridge_registry.h"
#include "lite/kernels/mlu/bridges/test_helper.h" #include "lite/kernels/mlu/bridges/test_helper.h"
#include "lite/kernels/npu/bridges/registry.h"
namespace paddle { namespace paddle {
namespace lite { namespace lite {
......
...@@ -12,9 +12,9 @@ ...@@ -12,9 +12,9 @@
// See the License for the specific language governing permissions and // See the License for the specific language governing permissions and
// limitations under the License. // limitations under the License.
#include "lite/core/subgraph_bridge_registry.h"
#include "lite/kernels/mlu/bridges/graph.h" #include "lite/kernels/mlu/bridges/graph.h"
#include "lite/kernels/mlu/bridges/utility.h" #include "lite/kernels/mlu/bridges/utility.h"
#include "lite/kernels/npu/bridges/registry.h"
namespace paddle { namespace paddle {
namespace lite { namespace lite {
......
...@@ -16,9 +16,9 @@ ...@@ -16,9 +16,9 @@
#include <gtest/gtest.h> #include <gtest/gtest.h>
#include "lite/core/op_lite.h" #include "lite/core/op_lite.h"
#include "lite/core/op_registry.h" #include "lite/core/op_registry.h"
#include "lite/core/subgraph_bridge_registry.h"
#include "lite/kernels/mlu/bridges/test_helper.h" #include "lite/kernels/mlu/bridges/test_helper.h"
#include "lite/kernels/mlu/bridges/utility.h" #include "lite/kernels/mlu/bridges/utility.h"
#include "lite/kernels/npu/bridges/registry.h"
namespace paddle { namespace paddle {
namespace lite { namespace lite {
......
...@@ -12,9 +12,9 @@ ...@@ -12,9 +12,9 @@
// See the License for the specific language governing permissions and // See the License for the specific language governing permissions and
// limitations under the License. // limitations under the License.
#include "lite/core/subgraph_bridge_registry.h"
#include "lite/kernels/mlu/bridges/graph.h" #include "lite/kernels/mlu/bridges/graph.h"
#include "lite/kernels/mlu/bridges/utility.h" #include "lite/kernels/mlu/bridges/utility.h"
#include "lite/kernels/npu/bridges/registry.h"
namespace paddle { namespace paddle {
namespace lite { namespace lite {
......
...@@ -17,8 +17,8 @@ ...@@ -17,8 +17,8 @@
#include <random> #include <random>
#include "lite/core/op_lite.h" #include "lite/core/op_lite.h"
#include "lite/core/op_registry.h" #include "lite/core/op_registry.h"
#include "lite/core/subgraph_bridge_registry.h"
#include "lite/kernels/mlu/bridges/test_helper.h" #include "lite/kernels/mlu/bridges/test_helper.h"
#include "lite/kernels/npu/bridges/registry.h"
namespace paddle { namespace paddle {
namespace lite { namespace lite {
......
...@@ -16,9 +16,9 @@ ...@@ -16,9 +16,9 @@
#include <algorithm> #include <algorithm>
#include "lite/core/subgraph_bridge_registry.h"
#include "lite/kernels/mlu/bridges/graph.h" #include "lite/kernels/mlu/bridges/graph.h"
#include "lite/kernels/mlu/bridges/utility.h" #include "lite/kernels/mlu/bridges/utility.h"
#include "lite/kernels/npu/bridges/registry.h"
namespace paddle { namespace paddle {
namespace lite { namespace lite {
......
...@@ -20,8 +20,8 @@ ...@@ -20,8 +20,8 @@
#include "lite/core/op_lite.h" #include "lite/core/op_lite.h"
#include "lite/core/op_registry.h" #include "lite/core/op_registry.h"
#include "lite/core/subgraph_bridge_registry.h"
#include "lite/kernels/mlu/bridges/test_helper.h" #include "lite/kernels/mlu/bridges/test_helper.h"
#include "lite/kernels/npu/bridges/registry.h"
namespace paddle { namespace paddle {
namespace lite { namespace lite {
......
...@@ -12,9 +12,9 @@ ...@@ -12,9 +12,9 @@
// See the License for the specific language governing permissions and // See the License for the specific language governing permissions and
// limitations under the License. // limitations under the License.
#include "lite/core/subgraph_bridge_registry.h"
#include "lite/kernels/mlu/bridges/graph.h" #include "lite/kernels/mlu/bridges/graph.h"
#include "lite/kernels/mlu/bridges/utility.h" #include "lite/kernels/mlu/bridges/utility.h"
#include "lite/kernels/npu/bridges/registry.h"
namespace paddle { namespace paddle {
namespace lite { namespace lite {
......
...@@ -16,8 +16,8 @@ ...@@ -16,8 +16,8 @@
#include <gtest/gtest.h> #include <gtest/gtest.h>
#include <random> #include <random>
#include "lite/core/op_registry.h" #include "lite/core/op_registry.h"
#include "lite/core/subgraph_bridge_registry.h"
#include "lite/kernels/mlu/bridges/test_helper.h" #include "lite/kernels/mlu/bridges/test_helper.h"
#include "lite/kernels/npu/bridges/registry.h"
namespace paddle { namespace paddle {
namespace lite { namespace lite {
......
...@@ -12,9 +12,9 @@ ...@@ -12,9 +12,9 @@
// See the License for the specific language governing permissions and // See the License for the specific language governing permissions and
// limitations under the License. // limitations under the License.
#include "lite/core/subgraph_bridge_registry.h"
#include "lite/kernels/mlu/bridges/graph.h" #include "lite/kernels/mlu/bridges/graph.h"
#include "lite/kernels/mlu/bridges/utility.h" #include "lite/kernels/mlu/bridges/utility.h"
#include "lite/kernels/npu/bridges/registry.h"
namespace paddle { namespace paddle {
namespace lite { namespace lite {
......
...@@ -16,8 +16,8 @@ ...@@ -16,8 +16,8 @@
#include <gtest/gtest.h> #include <gtest/gtest.h>
#include <random> #include <random>
#include "lite/core/op_registry.h" #include "lite/core/op_registry.h"
#include "lite/core/subgraph_bridge_registry.h"
#include "lite/kernels/mlu/bridges/test_helper.h" #include "lite/kernels/mlu/bridges/test_helper.h"
#include "lite/kernels/npu/bridges/registry.h"
namespace paddle { namespace paddle {
namespace lite { namespace lite {
......
...@@ -12,9 +12,9 @@ ...@@ -12,9 +12,9 @@
// See the License for the specific language governing permissions and // See the License for the specific language governing permissions and
// limitations under the License. // limitations under the License.
#include "lite/core/subgraph_bridge_registry.h"
#include "lite/kernels/mlu/bridges/graph.h" #include "lite/kernels/mlu/bridges/graph.h"
#include "lite/kernels/mlu/bridges/utility.h" #include "lite/kernels/mlu/bridges/utility.h"
#include "lite/kernels/npu/bridges/registry.h"
namespace paddle { namespace paddle {
namespace lite { namespace lite {
......
...@@ -16,8 +16,8 @@ ...@@ -16,8 +16,8 @@
#include <gtest/gtest.h> #include <gtest/gtest.h>
#include "lite/core/op_lite.h" #include "lite/core/op_lite.h"
#include "lite/core/op_registry.h" #include "lite/core/op_registry.h"
#include "lite/core/subgraph_bridge_registry.h"
#include "lite/kernels/mlu/bridges/test_helper.h" #include "lite/kernels/mlu/bridges/test_helper.h"
#include "lite/kernels/npu/bridges/registry.h"
namespace paddle { namespace paddle {
namespace lite { namespace lite {
......
...@@ -12,9 +12,9 @@ ...@@ -12,9 +12,9 @@
// See the License for the specific language governing permissions and // See the License for the specific language governing permissions and
// limitations under the License. // limitations under the License.
#include "lite/core/subgraph_bridge_registry.h"
#include "lite/kernels/mlu/bridges/graph.h" #include "lite/kernels/mlu/bridges/graph.h"
#include "lite/kernels/mlu/bridges/utility.h" #include "lite/kernels/mlu/bridges/utility.h"
#include "lite/kernels/npu/bridges/registry.h"
namespace paddle { namespace paddle {
namespace lite { namespace lite {
......
...@@ -19,8 +19,8 @@ ...@@ -19,8 +19,8 @@
#include <random> #include <random>
#include "lite/core/op_registry.h" #include "lite/core/op_registry.h"
#include "lite/core/subgraph_bridge_registry.h"
#include "lite/kernels/mlu/bridges/test_helper.h" #include "lite/kernels/mlu/bridges/test_helper.h"
#include "lite/kernels/npu/bridges/registry.h"
namespace paddle { namespace paddle {
namespace lite { namespace lite {
......
...@@ -12,9 +12,9 @@ ...@@ -12,9 +12,9 @@
// See the License for the specific language governing permissions and // See the License for the specific language governing permissions and
// limitations under the License. // limitations under the License.
#include "lite/core/subgraph_bridge_registry.h"
#include "lite/kernels/mlu/bridges/graph.h" #include "lite/kernels/mlu/bridges/graph.h"
#include "lite/kernels/mlu/bridges/utility.h" #include "lite/kernels/mlu/bridges/utility.h"
#include "lite/kernels/npu/bridges/registry.h"
namespace paddle { namespace paddle {
namespace lite { namespace lite {
......
...@@ -16,9 +16,9 @@ ...@@ -16,9 +16,9 @@
#include <gtest/gtest.h> #include <gtest/gtest.h>
#include "lite/core/op_lite.h" #include "lite/core/op_lite.h"
#include "lite/core/op_registry.h" #include "lite/core/op_registry.h"
#include "lite/core/subgraph_bridge_registry.h"
#include "lite/kernels/mlu/bridges/test_helper.h" #include "lite/kernels/mlu/bridges/test_helper.h"
#include "lite/kernels/mlu/bridges/utility.h" #include "lite/kernels/mlu/bridges/utility.h"
#include "lite/kernels/npu/bridges/registry.h"
namespace paddle { namespace paddle {
namespace lite { namespace lite {
......
...@@ -12,9 +12,9 @@ ...@@ -12,9 +12,9 @@
// See the License for the specific language governing permissions and // See the License for the specific language governing permissions and
// limitations under the License. // limitations under the License.
#include "lite/core/subgraph_bridge_registry.h"
#include "lite/kernels/mlu/bridges/graph.h" #include "lite/kernels/mlu/bridges/graph.h"
#include "lite/kernels/mlu/bridges/utility.h" #include "lite/kernels/mlu/bridges/utility.h"
#include "lite/kernels/npu/bridges/registry.h"
namespace paddle { namespace paddle {
namespace lite { namespace lite {
......
...@@ -18,9 +18,9 @@ ...@@ -18,9 +18,9 @@
#include "lite/core/device_info.h" #include "lite/core/device_info.h"
#include "lite/core/op_lite.h" #include "lite/core/op_lite.h"
#include "lite/core/op_registry.h" #include "lite/core/op_registry.h"
#include "lite/core/subgraph_bridge_registry.h"
#include "lite/kernels/mlu/bridges/test_helper.h" #include "lite/kernels/mlu/bridges/test_helper.h"
#include "lite/kernels/mlu/bridges/utility.h" #include "lite/kernels/mlu/bridges/utility.h"
#include "lite/kernels/npu/bridges/registry.h"
namespace paddle { namespace paddle {
namespace lite { namespace lite {
......
...@@ -12,9 +12,9 @@ ...@@ -12,9 +12,9 @@
// See the License for the specific language governing permissions and // See the License for the specific language governing permissions and
// limitations under the License. // limitations under the License.
#include "lite/core/subgraph_bridge_registry.h"
#include "lite/kernels/mlu/bridges/graph.h" #include "lite/kernels/mlu/bridges/graph.h"
#include "lite/kernels/mlu/bridges/utility.h" #include "lite/kernels/mlu/bridges/utility.h"
#include "lite/kernels/npu/bridges/registry.h"
namespace paddle { namespace paddle {
namespace lite { namespace lite {
......
...@@ -16,9 +16,9 @@ ...@@ -16,9 +16,9 @@
#include <gtest/gtest.h> #include <gtest/gtest.h>
#include "lite/core/op_lite.h" #include "lite/core/op_lite.h"
#include "lite/core/op_registry.h" #include "lite/core/op_registry.h"
#include "lite/core/subgraph_bridge_registry.h"
#include "lite/kernels/mlu/bridges/test_helper.h" #include "lite/kernels/mlu/bridges/test_helper.h"
#include "lite/kernels/mlu/bridges/utility.h" #include "lite/kernels/mlu/bridges/utility.h"
#include "lite/kernels/npu/bridges/registry.h"
namespace paddle { namespace paddle {
namespace lite { namespace lite {
......
...@@ -12,9 +12,9 @@ ...@@ -12,9 +12,9 @@
// See the License for the specific language governing permissions and // See the License for the specific language governing permissions and
// limitations under the License. // limitations under the License.
#include "lite/core/subgraph_bridge_registry.h"
#include "lite/kernels/mlu/bridges/graph.h" #include "lite/kernels/mlu/bridges/graph.h"
#include "lite/kernels/mlu/bridges/utility.h" #include "lite/kernels/mlu/bridges/utility.h"
#include "lite/kernels/npu/bridges/registry.h"
namespace paddle { namespace paddle {
namespace lite { namespace lite {
......
此差异已折叠。
此差异已折叠。
此差异已折叠。
此差异已折叠。
此差异已折叠。
此差异已折叠。
此差异已折叠。
此差异已折叠。
此差异已折叠。
此差异已折叠。
此差异已折叠。
此差异已折叠。
此差异已折叠。
此差异已折叠。
此差异已折叠。
此差异已折叠。
此差异已折叠。
此差异已折叠。
此差异已折叠。
此差异已折叠。
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册