Browse Source

!641 restore code

From: @yangyongqiang5033
Reviewed-by: @xchu42,@wqtshg
Signed-off-by: @wqtshg
pull/641/MERGE
mindspore-ci-bot Gitee 4 years ago
parent
commit
49a014c687
2 changed files with 4 additions and 12 deletions
  1. +3
    -11
      ge/common/ge/tbe_plugin_manager.cc
  2. +1
    -1
      parser

+ 3
- 11
ge/common/ge/tbe_plugin_manager.cc View File

@@ -184,20 +184,12 @@ void TBEPluginManager::GetCustomOpPath(std::string &customop_path) {
void TBEPluginManager::LoadCustomOpLib() { void TBEPluginManager::LoadCustomOpLib() {
LoadPluginSo(options_); LoadPluginSo(options_);


std::string fmk_type = std::to_string(domi::TENSORFLOW);
auto it = options_.find(ge::FRAMEWORK_TYPE);
if (it != options_.end()) {
fmk_type = it->second;
}
GELOGD("frameworkType is %s", fmk_type.c_str());
std::vector<OpRegistrationData> registration_datas = domi::OpRegistry::Instance()->registrationDatas; std::vector<OpRegistrationData> registration_datas = domi::OpRegistry::Instance()->registrationDatas;
GELOGI("The size of registration_datas is: %zu", registration_datas.size()); GELOGI("The size of registration_datas is: %zu", registration_datas.size());
for (OpRegistrationData reg_data : registration_datas) { for (OpRegistrationData reg_data : registration_datas) {
if (std::to_string(reg_data.GetFrameworkType()) == fmk_type) {
GELOGD("Begin to register optype: %s, imply_type: %s", reg_data.GetOmOptype().c_str(),
TypeUtils::ImplyTypeToSerialString(reg_data.GetImplyType()).c_str());
domi::OpRegistry::Instance()->Register(reg_data);
}
GELOGD("Begin to register optype: %s, imply_type: %s", reg_data.GetOmOptype().c_str(),
TypeUtils::ImplyTypeToSerialString(reg_data.GetImplyType()).c_str());
domi::OpRegistry::Instance()->Register(reg_data);
} }
} }




+ 1
- 1
parser

@@ -1 +1 @@
Subproject commit befc2aac08de4f1f1c38e476c4d3fd53174653ff
Subproject commit c1d95bac3a38b398641ddc67b935937fde699d3d

Loading…
Cancel
Save