p04896132
/
mindspore
forked from mindspore-Ecosystem/mindspore
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

modify log level in DfGraphManager

This commit is contained in:
chenhaozhe 2020-03-31 19:37:57 +08:00
parent 352c6faf85
commit 0727c2e76a
3 changed files with 7 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
mindspore/ccsrc/pipeline/pipeline.cc
View File

@ -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;
}

4
mindspore/ccsrc/transform/convert.cc
View File

@ -1100,12 +1100,12 @@ void DfGraphConvertor::UpdateDataOpDesc(const AnfNodePtr &it, const OperatorPtr
auto normal_shape_ptr = dyn_cast<abstract::Shape>(node->Shape());
vector<int> 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();

2
mindspore/ccsrc/transform/df_graph_manager.cc
View File

@ -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;