diff --git a/mindspore/ccsrc/pipeline/pipeline.cc b/mindspore/ccsrc/pipeline/pipeline.cc index 7b3cf06b5eba15749e9622b31d0e5e8fcfb8b27e..3c3478e89d210ca5aef34b886f861aefdb0a418c 100644 --- a/mindspore/ccsrc/pipeline/pipeline.cc +++ b/mindspore/ccsrc/pipeline/pipeline.cc @@ -1110,10 +1110,11 @@ bool ExecutorPy::AddDFGraph(const py::dict& init_params, const std::string& phas (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; } diff --git a/mindspore/ccsrc/transform/convert.cc b/mindspore/ccsrc/transform/convert.cc index c975b18812e58473acbed70e77573aa29a4c33e5..2b50d2328a297a55fc1addf8220b6a513ee50176 100755 --- a/mindspore/ccsrc/transform/convert.cc +++ b/mindspore/ccsrc/transform/convert.cc @@ -1104,12 +1104,12 @@ void DfGraphConvertor::UpdateDataOpDesc(const AnfNodePtr &it, const OperatorPtr auto normal_shape_ptr = dyn_cast(node->Shape()); vector shape; if (normal_shape_ptr == nullptr) { - MS_LOG(ERROR) << "Update data op descriptor failed! Invalid shape."; + MS_LOG(INFO) << "Invalid shape to update data op descriptor."; return; } shape = normal_shape_ptr->shape(); if (node->Type() == nullptr) { - MS_LOG(ERROR) << "Update data op descriptor failed! Invalid type."; + MS_LOG(INFO) << "Invalid type to update data op descriptor."; return; } TypeId me_type = node->Type()->type_id(); diff --git a/mindspore/ccsrc/transform/df_graph_manager.cc b/mindspore/ccsrc/transform/df_graph_manager.cc index 3339f431451cfbaa744883912bb1f2ba555fcaa5..bfe4d9f5d23a19a5e22527a906a8a593d369f829 100644 --- a/mindspore/ccsrc/transform/df_graph_manager.cc +++ b/mindspore/ccsrc/transform/df_graph_manager.cc @@ -112,7 +112,7 @@ DfGraphWrapperPtr DfGraphManager::GetGraphByName(const std::string& name) { auto it = graphs_.find(name); if (it == graphs_.end()) { - MS_LOG(ERROR) << "Can't found graph name: " << name; + MS_LOG(INFO) << "Can't found graph name: " << name; return nullptr; } MS_LOG(INFO) << "Return graph: " << name;