From 5c5c138941d2de638a01ae9922b358dbfac4db69 Mon Sep 17 00:00:00 2001
From: baker <chenhua26@hisilicon.com>
Date: Thu, 10 Dec 2020 22:22:04 +0800
Subject: [PATCH] add onnx model parse api

---
 parser/onnx/onnx_parser.cc | 6 +++---
 1 file changed, 3 insertions(+), 3 deletions(-)

diff --git a/parser/onnx/onnx_parser.cc b/parser/onnx/onnx_parser.cc
index 009a447..404c177 100644
--- a/parser/onnx/onnx_parser.cc
+++ b/parser/onnx/onnx_parser.cc
@@ -111,7 +111,7 @@ graphStatus aclgrphParseONNXFromMem(const char *buffer, size_t size,
   GE_CHECK_NOTNULL(model_parser);
 
   // parse caffe model_file to GE graph
-  ge::graphStatus ret = model_parser->Parse(buffer, (uint32_t)size, graph);
+  ge::graphStatus ret = model_parser->ParseFromMemory(buffer, (uint32_t)size, graph);
   if (ret != ge::SUCCESS) {
     GELOGE(ret, "Parser graph %s failed.", graph.GetName().c_str());
     return ge::FAILED;
@@ -506,7 +506,7 @@ Status OnnxModelParser::GetModelFromFile(const char *file, ge::onnx::ModelProto
   GELOGI("File path is %s.", file);
 
   // 1. Get graph from onnx model file.
-  if (!ge::parser::ReadProtoFromBinaryFile(file, onnx_model)) {
+  if (!ge::parser::ReadProtoFromBinaryFile(file, &onnx_model)) {
     ErrorManager::GetInstance().ATCReportErrMessage(
         "E19021", {"reason"}, {"Read onnx model file failed."});
     GELOGE(PARAM_INVALID, "Read onnx model file failed.");
@@ -645,7 +645,7 @@ Status OnnxModelParser::ParseFromMemory(const char *data, uint32_t size, ge::Gra
     GELOGE(FAILED, "get model from file failed.");
     return FAILED;
   }
-  ret = RealParse(onnx_model, graph)
+  ret = RealParse(onnx_model, graph);
   if (ret != SUCCESS) {
     GELOGE(FAILED, "parse model failed.");
     return FAILED;