Browse Source

modify log level in DfGraphManager

tags/v0.2.0-alpha
chenhaozhe 5 years ago
parent
commit
0727c2e76a
3 changed files with 7 additions and 6 deletions
  1. +4
    -3
      mindspore/ccsrc/pipeline/pipeline.cc
  2. +2
    -2
      mindspore/ccsrc/transform/convert.cc
  3. +1
    -1
      mindspore/ccsrc/transform/df_graph_manager.cc

+ 4
- 3
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;
}


+ 2
- 2
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();


+ 1
- 1
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;


Loading…
Cancel
Save