| @@ -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(phase, convertor.GetComputeGraph()); | ||||
| } | } | ||||
| (void)DfGraphManager::GetInstance().AddGraph(init_graph, convertor.GetInitGraph()); | (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()); | (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; | return true; | ||||
| } | } | ||||
| @@ -1100,12 +1100,12 @@ void DfGraphConvertor::UpdateDataOpDesc(const AnfNodePtr &it, const OperatorPtr | |||||
| auto normal_shape_ptr = dyn_cast<abstract::Shape>(node->Shape()); | auto normal_shape_ptr = dyn_cast<abstract::Shape>(node->Shape()); | ||||
| vector<int> shape; | vector<int> shape; | ||||
| if (normal_shape_ptr == nullptr) { | 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; | return; | ||||
| } | } | ||||
| shape = normal_shape_ptr->shape(); | shape = normal_shape_ptr->shape(); | ||||
| if (node->Type() == nullptr) { | 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; | return; | ||||
| } | } | ||||
| TypeId me_type = node->Type()->type_id(); | TypeId me_type = node->Type()->type_id(); | ||||
| @@ -112,7 +112,7 @@ DfGraphWrapperPtr DfGraphManager::GetGraphByName(const std::string& name) { | |||||
| auto it = graphs_.find(name); | auto it = graphs_.find(name); | ||||
| if (it == graphs_.end()) { | if (it == graphs_.end()) { | ||||
| MS_LOG(ERROR) << "Can't found graph name: " << name; | |||||
| MS_LOG(INFO) << "Can't found graph name: " << name; | |||||
| return nullptr; | return nullptr; | ||||
| } | } | ||||
| MS_LOG(INFO) << "Return graph: " << name; | MS_LOG(INFO) << "Return graph: " << name; | ||||