!152 bugfix for plugin load.

Merge pull request !152 from yangyongqiang/dev_bugfix

parser/common/acl_graph_parser_util.cc

@@ -220,11 +220,19 @@ domi::Status AclGrphParseUtil::AclParserInitialize(const std::map<std::string, s
     return FAILED;
   }
 
+  auto it = options.find(ge::FRAMEWORK_TYPE);
+  if (it == options.end()) {
+    GELOGE(FAILED, "Can not find ge.frameworkType in options");
+    return FAILED;
+  }
+  std::string fmk_type = it->second;
   std::vector<OpRegistrationData> registrationDatas = op_registry->registrationDatas;
   GELOGI("The size of registrationDatas in parser is: %zu", registrationDatas.size());
   for (OpRegistrationData &reg_data : registrationDatas) {
-    (void)OpRegistrationTbe::Instance()->Finalize(reg_data, false);
-    domi::OpRegistry::Instance()->Register(reg_data);
+    if (std::to_string(reg_data.GetFrameworkType()) == fmk_type) {
+      (void)OpRegistrationTbe::Instance()->Finalize(reg_data, false);
+      (void)domi::OpRegistry::Instance()->Register(reg_data);
+    }
   }
 
   // set init status

parser/common/parser_api.cc

@@ -38,10 +38,17 @@ Status ParserInitialize(const std::map<std::string, std::string> &options) {
   // load custom op plugin
   TBEPluginLoader::Instance().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;
+  }
   std::vector<OpRegistrationData> registrationDatas = domi::OpRegistry::Instance()->registrationDatas;
   GELOGI("The size of registrationDatas in parser is: %zu", registrationDatas.size());
   for (OpRegistrationData &reg_data : registrationDatas) {
-    (void)OpRegistrationTbe::Instance()->Finalize(reg_data, true);
+    if (std::to_string(reg_data.GetFrameworkType()) == fmk_type) {
+      (void)OpRegistrationTbe::Instance()->Finalize(reg_data, true);
+    }
   }
 
   auto iter = options.find(ge::OPTION_EXEC_ENABLE_SCOPE_FUSION_PASSES);