From b3c3584c969b929bc823d0dce11b3341f685babc Mon Sep 17 00:00:00 2001 From: baker Date: Wed, 6 Jan 2021 09:47:54 +0800 Subject: [PATCH] fix bug --- inc/external/parser/onnx_parser.h | 6 ++++-- metadef | 2 +- parser/onnx/onnx_parser.cc | 10 +++++++--- 3 files changed, 12 insertions(+), 6 deletions(-) diff --git a/inc/external/parser/onnx_parser.h b/inc/external/parser/onnx_parser.h index 7678218..116fc72 100644 --- a/inc/external/parser/onnx_parser.h +++ b/inc/external/parser/onnx_parser.h @@ -27,11 +27,13 @@ namespace ge { graphStatus aclgrphParseONNX(const char *model_file, - std::map &parser_params, + const std::map &parser_params, ge::Graph &graph); graphStatus aclgrphParseONNXFromMem(const char *buffer, size_t buffer_size, - std::map &parser_params, + const std::map &parser_params, ge::Graph &graph); } // namespace ge diff --git a/metadef b/metadef index bb65bdd..f08320a 160000 --- a/metadef +++ b/metadef @@ -1 +1 @@ -Subproject commit bb65bddc85d786d59553cdd56e3cf70ce8c9ead4 +Subproject commit f08320a6d699f5b537bf66da572bf225b9cd330e diff --git a/parser/onnx/onnx_parser.cc b/parser/onnx/onnx_parser.cc index 4925f69..5e51446 100644 --- a/parser/onnx/onnx_parser.cc +++ b/parser/onnx/onnx_parser.cc @@ -40,7 +40,9 @@ namespace ge { graphStatus aclgrphParseONNX(const char *model_file, - std::map &parser_params, ge::Graph &graph) { + const std::map &parser_params, + ge::Graph &graph) { GE_CHECK_NOTNULL(model_file); GetParserContext().type = domi::ONNX; std::map options; @@ -86,7 +88,9 @@ graphStatus aclgrphParseONNX(const char *model_file, } graphStatus aclgrphParseONNXFromMem(const char *buffer, size_t buffer_size, - std::map &parser_params, ge::Graph &graph) { + const std::map &parser_params, + ge::Graph &graph) { GE_CHECK_NOTNULL(buffer); GetParserContext().type = domi::ONNX; std::map options; @@ -672,4 +676,4 @@ ge::DataType OnnxModelParser::ConvertToGeDataType(const uint32_t type) { namespace domi { REGISTER_MODEL_PARSER_CREATOR(ONNX, ge::OnnxModelParser); REGISTER_WEIGHTS_PARSER_CREATOR(ONNX, ge::OnnxWeightsParser); -} \ No newline at end of file +}