!140 add validation of fmk type for plugin load.

Merge pull request !140 from yangyongqiang/master

parser/common/acl_graph_parser_util.cc

@@ -27,6 +27,7 @@
 
 #include "ge/ge_api_types.h"
 #include "graph/opsproto_manager.h"
+#include "graph/utils/type_utils.h"
 #include "omg/parser/parser_inner_ctx.h"
 #include "framework/common/debug/ge_log.h"
 #include "parser/common/register_tbe.h"
@@ -206,11 +207,20 @@ 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;
+  GELOGD("frameworkType is %s", fmk_type.c_str());
   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 (ge::TypeUtils::FmkTypeToSerialString(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

@@ -19,6 +19,7 @@
 
 #include "common/ge/tbe_plugin_manager.h"
 #include "framework/common/debug/ge_log.h"
+#include "graph/utils/type_utils.h"
 #include "parser/common/register_tbe.h"
 #include "framework/omg/parser/parser_inner_ctx.h"
 #include "external/ge/ge_api_types.h"
@@ -38,10 +39,18 @@ Status ParserInitialize(const std::map<std::string, std::string> &options) {
   // load custom op plugin
   TBEPluginManager::Instance().LoadPluginSo(options);
 
+  std::string fmk_type = ge::TypeUtils::FmkTypeToSerialString(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> 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 (ge::TypeUtils::FmkTypeToSerialString(reg_data.GetFrameworkType()) == fmk_type) {
+      (void)OpRegistrationTbe::Instance()->Finalize(reg_data, true);
+    }
   }
 
   auto iter = options.find(ge::OPTION_EXEC_ENABLE_SCOPE_FUSION_PASSES);