From 30ae5dbb1014d0accbda20e9acadbfc538805479 Mon Sep 17 00:00:00 2001
From: baker <chenhua26@hisilicon.com>
Date: Thu, 17 Dec 2020 19:32:50 +0800
Subject: [PATCH] add onnx model parse api

---
 inc/framework/common/ge_types.h             | 3 +++
 inc/framework/omg/parser/parser_inner_ctx.h | 2 +-
 2 files changed, 4 insertions(+), 1 deletion(-)

diff --git a/inc/framework/common/ge_types.h b/inc/framework/common/ge_types.h
index fb1f0be1..eea9824b 100644
--- a/inc/framework/common/ge_types.h
+++ b/inc/framework/common/ge_types.h
@@ -37,6 +37,9 @@ enum FrameworkType {
   MINDSPORE = 1,
   TENSORFLOW = 3,
   ANDROID_NN,
+#ifndef ONLY_COMPILE_OPEN_SRC
+  ONNX,
+#endif
   FRAMEWORK_RESERVED,
 };
 
diff --git a/inc/framework/omg/parser/parser_inner_ctx.h b/inc/framework/omg/parser/parser_inner_ctx.h
index f24e2639..5d91bd46 100644
--- a/inc/framework/omg/parser/parser_inner_ctx.h
+++ b/inc/framework/omg/parser/parser_inner_ctx.h
@@ -59,7 +59,7 @@ struct ParserContext {
   bool train_flag = false;
   domi::domiTensorFormat_t format = domi::DOMI_TENSOR_ND;
   domi::FrameworkType type = domi::FRAMEWORK_RESERVED;
-  RunMode run_mode = ONLY_PRE_CHECK;
+  RunMode run_mode = GEN_OM_MODEL;
   // save caffe custom proto path, used by caffe parse
   std::string custom_proto_path;
   // save caffe proto path, used by caffe parse