Home
last modified time | relevance | path

Searched refs:total_worker_num (Results 1 – 3 of 3) sorted by relevance

/third_party/mindspore/mindspore/ccsrc/ps/core/
Dscheduler_node.cc573 int32_t total_worker_num = scale_worker_num + node_manager_.worker_num(); in ProcessScaleOut() local
576 MS_LOG(INFO) << "After scale out, the total worker num:" << total_worker_num in ProcessScaleOut()
579 node_manager_.set_worker_num(total_worker_num); in ProcessScaleOut()
581 node_manager_.set_total_node_num(total_worker_num + total_server_num); in ProcessScaleOut()
659 int32_t total_worker_num = node_manager_.worker_num() - scale_worker_num; in ProcessScaleIn() local
662 node_manager_.set_worker_num(total_worker_num); in ProcessScaleIn()
664 node_manager_.set_total_node_num(total_worker_num + total_server_num); in ProcessScaleIn()
Dcomm_util.h104 …ateRankId(const enum NodeRole &node_role, const uint32_t &rank_id, const int32_t &total_worker_num,
Dcomm_util.cc138 …ateRankId(const enum NodeRole &node_role, const uint32_t &rank_id, const int32_t &total_worker_num, in ValidateRankId() argument
142 } else if (node_role == NodeRole::WORKER && (rank_id > IntToUint(total_worker_num) - 1)) { in ValidateRankId()