From 52354ca99f756112980166267a4054d025934c57 Mon Sep 17 00:00:00 2001 From: wangxiaotian22 Date: Fri, 26 Feb 2021 16:44:15 +0800 Subject: [PATCH] support error_manager used in train --- metadef | 2 +- parser/tensorflow/tensorflow_parser.cc | 4 ++++ 2 files changed, 5 insertions(+), 1 deletion(-) diff --git a/metadef b/metadef index f982caa..3df578f 160000 --- a/metadef +++ b/metadef @@ -1 +1 @@ -Subproject commit f982caa0981b1fdcc55a8ec27b4f4de9c58d33ba +Subproject commit 3df578f6d08e51e2c4ed8a023fb8482109941665 diff --git a/parser/tensorflow/tensorflow_parser.cc b/parser/tensorflow/tensorflow_parser.cc index d2065f1..d1b57d3 100644 --- a/parser/tensorflow/tensorflow_parser.cc +++ b/parser/tensorflow/tensorflow_parser.cc @@ -818,6 +818,8 @@ Status TensorFlowModelParser::CheckOpType(const domi::tensorflow::NodeDef *node_ * @return ge::DataType */ ge::DataType TensorFlowModelParser::ConvertToGeDataType(const uint32_t type) { + ErrorManager::GetInstance().GenWorkStreamIdDefault(); + ge::DataType data_type = domi::TensorAssign::ConvertTensorflowDataType(type); return data_type; } @@ -2139,6 +2141,7 @@ Status TensorFlowWeightsParser::ParseFromMemory(const char *data, uint32_t size, Status TensorFlowWeightsParser::Parse(const char *file, ge::Graph &graph) { return SUCCESS; } Status TensorFlowModelParser::ParseProto(const google::protobuf::Message *proto, ge::ComputeGraphPtr &graph) { + ErrorManager::GetInstance().GenWorkStreamIdDefault(); PARSER_TIMESTAMP_START(ParseProto); GE_CHECK_NOTNULL(proto); GE_CHECK_NOTNULL(graph); @@ -2265,6 +2268,7 @@ Status TensorFlowModelParser::ParseProto(const google::protobuf::Message *proto, Status TensorFlowModelParser::ParseProtoWithSubgraph(const google::protobuf::Message *root_proto, domi::GetGraphCallback callback, ge::ComputeGraphPtr &root_graph) { + ErrorManager::GetInstance().GenWorkStreamIdDefault(); GE_CHECK_NOTNULL(root_proto); GE_CHECK_NOTNULL(callback); GE_CHECK_NOTNULL(root_graph);