diff --git a/mindspore/ccsrc/pipeline/pipeline_ge.cc b/mindspore/ccsrc/pipeline/pipeline_ge.cc index abf55bb2d8..2f68935591 100644 --- a/mindspore/ccsrc/pipeline/pipeline_ge.cc +++ b/mindspore/ccsrc/pipeline/pipeline_ge.cc @@ -227,10 +227,12 @@ bool AddDFGraph(const std::map& info, const py::di (void)DfGraphManager::GetInstance().AddGraph(phase, convertor.GetComputeGraph()); } (void)DfGraphManager::GetInstance().AddGraph(init_graph, convertor.GetInitGraph()); - (void)DfGraphManager::GetInstance().AddGraph(checkpoint_name, convertor.GetSaveCheckpointGraph()); (void)DfGraphManager::GetInstance().AddGraph(BROADCAST_GRAPH_NAME, convertor.GetBroadcastGraph()); - DfGraphManager::GetInstance().SetAnfGraph(checkpoint_name, anf_graph); + Status ret = DfGraphManager::GetInstance().AddGraph(checkpoint_name, convertor.GetSaveCheckpointGraph()); + if (ret == Status::SUCCESS) { + DfGraphManager::GetInstance().SetAnfGraph(checkpoint_name, anf_graph); + } return true; } @@ -389,8 +391,7 @@ std::shared_ptr DoExecGraph(const FuncGraphPtr& graph, const std::ve const std::string& phase) { std::vector ge_tensors = TransformUtil::ConvertInputTensors(inputs, kOpFormat_NCHW); if (ge_tensors.size() != inputs.size()) { - MS_LOG(ERROR) << "Args convert to ge tensor error"; - return nullptr; + MS_LOG(EXCEPTION) << "Convert me args to ge tensor error."; } std::vector ge_outputs; @@ -401,8 +402,7 @@ std::shared_ptr DoExecGraph(const FuncGraphPtr& graph, const std::ve auto graph_runner = DfGraphManager::GetInstance().GetGraphRunner(); if (graph_runner == nullptr) { - MS_LOG(ERROR) << "Can not found GraphRunner"; - return nullptr; + MS_LOG(EXCEPTION) << "Can not found GraphRunner."; } { @@ -419,7 +419,7 @@ std::shared_ptr DoExecGraph(const FuncGraphPtr& graph, const std::ve std::vector me_outputs = TransformUtil::ConvertGeTensors(ge_outputs); if (me_outputs.size() != ge_outputs.size()) { - MS_LOG(ERROR) << "Convert output Ge tensor to Me tensor failed"; + MS_LOG(WARNING) << "Convert output Ge tensor to Me tensor failed"; } py::tuple outputs(me_outputs.size());