From df5f18c9ba69b9d664dcf7167fbb13551ed553a7 Mon Sep 17 00:00:00 2001
From: "wangwenhua1@huawei.com" <wangwenhua1@huawei.com>
Date: Wed, 23 Dec 2020 14:54:45 +0800
Subject: [PATCH] display model info

---
 ge/generator/ge_generator.cc | 10 +++++++++-
 1 file changed, 9 insertions(+), 1 deletion(-)

diff --git a/ge/generator/ge_generator.cc b/ge/generator/ge_generator.cc
index a3ad443d..81d4a8cb 100644
--- a/ge/generator/ge_generator.cc
+++ b/ge/generator/ge_generator.cc
@@ -538,10 +538,18 @@ bool GeGenerator::Impl::SetOmSystemInfo(AttrHolder &obj) {
     GELOGW("SetStr of soc_version failed.");
     return false;
   }
+
+  // 0(Caffe) 1(MindSpore) 3(TensorFlow) 5(Onnx)
+  std::map<string, string> framework_type_to_string = {
+    {"0", "Caffe"},
+    {"1", "MindSpore"},
+    {"3", "TensorFlow"},
+    {"5", "Onnx"}
+  };
   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", framework_type)) {
+  if (!ge::AttrUtils::SetStr(obj, "framework_type", framework_type_to_string[framework_type.c_str()])) {
     GELOGW("SetStr of framework_type failed.");
     return false;
   }