Browse Source

refine log for parameter server

tags/v1.6.0
lizhenyu 4 years ago
parent
commit
2644ef3456
2 changed files with 3 additions and 2 deletions
  1. +1
    -1
      mindspore/ccsrc/ps/core/scheduler_node.cc
  2. +2
    -1
      mindspore/ccsrc/ps/util.cc

+ 1
- 1
mindspore/ccsrc/ps/core/scheduler_node.cc View File

@@ -302,7 +302,7 @@ void SchedulerNode::ProcessRegister(const std::shared_ptr<TcpServer> &server,
auto node_infos = node_manager_.nodes_info();
bool res = SendPrepareBuildingNetwork(node_infos);
if (!res) {
MS_LOG(ERROR) << "Prepare for building network failed!";
MS_LOG(WARNING) << "Prepare for building network failed!";
return;
}
MS_LOG(INFO) << "Prepare for building network success.";


+ 2
- 1
mindspore/ccsrc/ps/util.cc View File

@@ -82,7 +82,8 @@ int64_t Util::LocalShard(int64_t first_dim, int64_t rank_id, int64_t server_num)

std::map<int64_t, int64_t> Util::AllRankLocalShard(int64_t first_dim, int64_t rank_id, int64_t server_num) {
if (first_dim <= 0 || server_num <= 0 || rank_id < 0) {
MS_LOG(EXCEPTION) << "Input values are invalid.";
MS_LOG(EXCEPTION) << "Input values are invalid, first_dim: " << first_dim << ", server_num: " << server_num
<< ", rank_id: " << rank_id;
}
if (rank_id >= server_num) {
MS_LOG(EXCEPTION) << "The rank ID " << rank_id << " should be less than the number of servers " << server_num;


Loading…
Cancel
Save