| @@ -10,7 +10,7 @@ endif() | |||||
| option(ENABLE_OPEN_SRC "Enable graphengine compile in opensource." FALSE) | option(ENABLE_OPEN_SRC "Enable graphengine compile in opensource." FALSE) | ||||
| if (ENABLE_OPEN_SRC) | if (ENABLE_OPEN_SRC) | ||||
| set(HI_PYTHON python3.7) | |||||
| set(HI_PYTHON python3) | |||||
| include(cmake/external_libs/protobuf_shared.cmake) | include(cmake/external_libs/protobuf_shared.cmake) | ||||
| include(cmake/external_libs/protoc.cmake) | include(cmake/external_libs/protoc.cmake) | ||||
| @@ -1 +1 @@ | |||||
| Subproject commit 220067ec17e13e5cc913e9e2c5895fcabaa5bd59 | |||||
| Subproject commit d19c9c5c92f21a0335c18681dcceed44f3a54ddc | |||||
| @@ -54,6 +54,9 @@ class CaffeModelParser : public domi::ModelParser { | |||||
| */ | */ | ||||
| Status Parse(const char *file, ge::Graph &graph) override; | Status Parse(const char *file, ge::Graph &graph) override; | ||||
| Status ParseFromMemory(const char *data, uint32_t size, ge::ComputeGraphPtr &graph) override; | Status ParseFromMemory(const char *data, uint32_t size, ge::ComputeGraphPtr &graph) override; | ||||
| virtual Status ParseFromMemory(const char *data, uint32_t size, ge::Graph &graph) { | |||||
| return domi::SUCCESS; | |||||
| } | |||||
| /** | /** | ||||
| * @ingroup domi_omg | * @ingroup domi_omg | ||||
| @@ -74,6 +74,7 @@ target_link_libraries(fmk_onnx_parser PRIVATE | |||||
| json | json | ||||
| -lrt | -lrt | ||||
| -ldl | -ldl | ||||
| error_manager | |||||
| ) | ) | ||||
| ############ install ############ | ############ install ############ | ||||
| @@ -50,6 +50,7 @@ LOCAL_SHARED_LIBRARIES := \ | |||||
| libparser_common \ | libparser_common \ | ||||
| libgraph \ | libgraph \ | ||||
| libregister \ | libregister \ | ||||
| liberror_manager \ | |||||
| LOCAL_STATIC_LIBRARIES += libmmpa | LOCAL_STATIC_LIBRARIES += libmmpa | ||||
| @@ -39,6 +39,9 @@ class OnnxModelParser : public domi::ModelParser { | |||||
| ge::DataType ConvertToGeDataType(const uint32_t type) override; | ge::DataType ConvertToGeDataType(const uint32_t type) override; | ||||
| Status ParseFromMemory(const char *data, uint32_t size, ge::ComputeGraphPtr &graph) override { return domi::SUCCESS; } | Status ParseFromMemory(const char *data, uint32_t size, ge::ComputeGraphPtr &graph) override { return domi::SUCCESS; } | ||||
| virtual Status ParseFromMemory(const char *data, uint32_t size, ge::Graph &graph) { | |||||
| return domi::SUCCESS; | |||||
| } | |||||
| Status ParseProto(const google::protobuf::Message *proto, ge::ComputeGraphPtr &graph) override { | Status ParseProto(const google::protobuf::Message *proto, ge::ComputeGraphPtr &graph) override { | ||||
| return domi::SUCCESS; | return domi::SUCCESS; | ||||
| @@ -139,8 +139,7 @@ Status TensorFlowDataParser::CheckInputShape(const std::string &name) { | |||||
| // dim i = 0, means empty tensor. | // dim i = 0, means empty tensor. | ||||
| // dim i = -1 or -2, means unknown shape. | // dim i = -1 or -2, means unknown shape. | ||||
| GE_CHK_BOOL_RET_STATUS(user_input_dims_v[i] >= kValidShapeMinValue, domi::PARAM_INVALID, | GE_CHK_BOOL_RET_STATUS(user_input_dims_v[i] >= kValidShapeMinValue, domi::PARAM_INVALID, | ||||
| "parse data node %s: shape contains placeholder ,but not designated by user", | |||||
| name.c_str()); | |||||
| "parse data node %s: shape contains placeholder ,but not designated by user", name.c_str()); | |||||
| } | } | ||||
| return SUCCESS; | return SUCCESS; | ||||
| } | } | ||||
| @@ -91,6 +91,9 @@ class TensorFlowModelParser : public domi::ModelParser { | |||||
| Status Parse(const char *file, ge::Graph &graph) override; | Status Parse(const char *file, ge::Graph &graph) override; | ||||
| Status ParseFromMemory(const char *data, uint32_t size, ge::ComputeGraphPtr &graph) override; | Status ParseFromMemory(const char *data, uint32_t size, ge::ComputeGraphPtr &graph) override; | ||||
| virtual Status ParseFromMemory(const char *data, uint32_t size, ge::Graph &graph) { | |||||
| return domi::SUCCESS; | |||||
| } | |||||
| /** | /** | ||||
| * @ingroup domi_omg | * @ingroup domi_omg | ||||