display model info

pull/857/head
wangwenhua1@huawei.com 5 years ago
parent ae345567d6
commit a7c576ad93

@ -559,7 +559,12 @@ bool GeGenerator::Impl::SetOmSystemInfo(AttrHolder &obj) {
std::string framework_type;
(void)ge::GetContext().GetOption(ge::FRAMEWORK_TYPE, framework_type);
GELOGI("SetOmSystemInfo framework_type: %s", framework_type.c_str());
if (!ge::AttrUtils::SetStr(obj, "framework_type", ge::kFwkTypeToStr[framework_type])) {
auto iter = ge::kFwkTypeToStr.find(framework_type);
if (iter == ge::kFwkTypeToStr.end()) {
GELOGW("can not find framework_type in map");
return false;
}
if (!ge::AttrUtils::SetStr(obj, "framework_type", iter->second) {
GELOGW("SetStr of framework_type failed.");
return false;
}

Loading…
Cancel
Save