diff --git a/parser/onnx/onnx_parser.cc b/parser/onnx/onnx_parser.cc index fd49ac1..6783fda 100644 --- a/parser/onnx/onnx_parser.cc +++ b/parser/onnx/onnx_parser.cc @@ -41,7 +41,7 @@ namespace ge { graphStatus PrepareBeforeParse(AclGrphParseUtil &acl_graph_parse_util, const std::map &parser_params, - ge::Graph &graph, domi::ModelParser *model_parser) { + ge::Graph &graph, std::shared_ptr model_parser) { GetParserContext().type = domi::ONNX; std::map options; options.insert(std::pair(string(ge::FRAMEWORK_TYPE), to_string(ge::ONNX))); @@ -83,7 +83,7 @@ graphStatus aclgrphParseONNX(const char *model_file, GE_CHECK_NOTNULL(model_file); // load custom plugin so and proto AclGrphParseUtil acl_graph_parse_util; - domi::ModelParser *model_parser = nullptr; + std::shared_ptr model_parser; if (PrepareBeforeParse(acl_graph_parse_util, parser_params, graph, model_parser) != ge::SUCCESS) { GELOGE(ge::FAILED, "prepare before parse failed."); @@ -113,7 +113,7 @@ graphStatus aclgrphParseONNXFromMem(const char *buffer, size_t size, GE_CHECK_NOTNULL(buffer); // load custom plugin so and proto AclGrphParseUtil acl_graph_parse_util; - domi::ModelParser *model_parser = nullptr; + std::shared_ptr model_parser; if (PrepareBeforeParse(acl_graph_parse_util, parser_params, graph, model_parser) != ge::SUCCESS) { GELOGE(ge::FAILED, "prepare before parse failed.");