diff --git a/parser/common/acl_graph_parser_util.cc b/parser/common/acl_graph_parser_util.cc index c4ea63d..61c82a3 100644 --- a/parser/common/acl_graph_parser_util.cc +++ b/parser/common/acl_graph_parser_util.cc @@ -217,7 +217,7 @@ domi::Status AclGrphParseUtil::AclParserInitialize(const std::map registrationDatas = op_registry->registrationDatas; GELOGI("The size of registrationDatas in parser is: %zu", registrationDatas.size()); for (OpRegistrationData ®_data : registrationDatas) { - if (ge::TypeUtils::FmkTypeToSerialString(reg_data.GetFrameworkType()) == fmk_type) { + if (std::to_string(reg_data.GetFrameworkType()) == fmk_type) { (void)OpRegistrationTbe::Instance()->Finalize(reg_data, false); (void)domi::OpRegistry::Instance()->Register(reg_data); } diff --git a/parser/common/parser_api.cc b/parser/common/parser_api.cc index 94b5999..651d1b1 100644 --- a/parser/common/parser_api.cc +++ b/parser/common/parser_api.cc @@ -39,7 +39,7 @@ Status ParserInitialize(const std::map &options) { // load custom op plugin TBEPluginManager::Instance().LoadPluginSo(options); - std::string fmk_type = ge::TypeUtils::FmkTypeToSerialString(domi::TENSORFLOW); + std::string fmk_type = std::to_string(domi::TENSORFLOW); auto it = options.find(ge::FRAMEWORK_TYPE); if (it != options.end()) { fmk_type = it->second; @@ -48,7 +48,7 @@ Status ParserInitialize(const std::map &options) { std::vector registrationDatas = domi::OpRegistry::Instance()->registrationDatas; GELOGI("The size of registrationDatas in parser is: %zu", registrationDatas.size()); for (OpRegistrationData ®_data : registrationDatas) { - if (ge::TypeUtils::FmkTypeToSerialString(reg_data.GetFrameworkType()) == fmk_type) { + if (std::to_string(reg_data.GetFrameworkType()) == fmk_type) { (void)OpRegistrationTbe::Instance()->Finalize(reg_data, true); } }