| /third_party/mindspore/mindspore/ccsrc/fl/worker/ |
| D | fl_worker.cc | 33 server_num_ = ps::PSContext::instance()->server_num(); in Run()
40 ps::PSContext::instance()->cluster_config().initial_server_num = server_num_; in Run()
42 << ", Server number:" << server_num_ << ", Scheduler ip:" << scheduler_ip_ in Run()
150 uint32_t FLWorker::server_num() const { return server_num_; } in server_num()
242 server_num_ = IntToUint(worker_node_->server_num()); in ProcessAfterScalingOut()
245 << server_num_ << ". Exit safemode."; in ProcessAfterScalingOut()
253 server_num_ = IntToUint(worker_node_->server_num()); in ProcessAfterScalingIn()
255 …ter scheduler scaling in, worker number is " << worker_num_ << ", server number is " << server_num_ in ProcessAfterScalingIn()
|
| D | fl_worker.h | 97 server_num_(0), in FLWorker()
129 uint32_t server_num_; variable
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/fl/ |
| D | fused_push_weight_kernel.h | 37 … : server_num_(0), indices_({}), weight_full_names_({}), fl_iteration_(0), total_iteration_(0) {} in FusedPushWeightKernel()
122 server_num_ = ps::PSContext::instance()->server_num(); in Init()
126 << ", server number is " << server_num_; in Init()
127 if (server_num_ == 0 || weight_full_names_.empty() || indices_.empty()) { in Init()
166 uint32_t server_num_; variable
|
| D | fused_pull_weight_kernel.h | 39 … : server_num_(0), indices_({}), weight_full_names_({}), fl_iteration_(0), total_iteration_(0) {} in FusedPullWeightKernel()
135 server_num_ = ps::PSContext::instance()->server_num(); in Init()
139 << ", server number is " << server_num_; in Init()
140 if (server_num_ == 0 || weight_full_names_.empty() || indices_.empty()) { in Init()
194 uint32_t server_num_; variable
|
| D | start_fl_job_kernel.h | 74 server_num_ = fl::worker::FLWorker::GetInstance().server_num(); in Init()
80 target_server_rank_ = rank_id_ % server_num_; in Init()
115 uint32_t server_num_; variable
|
| D | get_model_kernel.h | 104 server_num_ = fl::worker::FLWorker::GetInstance().server_num(); in Init()
110 target_server_rank_ = rank_id_ % server_num_; in Init()
153 uint32_t server_num_; variable
|
| D | update_model_kernel.h | 86 server_num_ = fl::worker::FLWorker::GetInstance().server_num(); in Init()
92 target_server_rank_ = rank_id_ % server_num_; in Init()
161 uint32_t server_num_; variable
|
| /third_party/mindspore/mindspore/ccsrc/fl/server/ |
| D | collective_ops_impl.cc | 26 server_num_ = ps::PSContext::instance()->initial_server_num(); in Initialize()
40 uint32_t rank_size = server_num_; in RingAllReduce()
142 uint32_t rank_size = server_num_; in ReduceBroadcastAllReduce()
219 uint32_t rank_size = server_num_; in AllReduce()
235 server_num_ = IntToUint(server_node_->server_num()); in ReInitForScaling()
237 << server_num_; in ReInitForScaling()
|
| D | collective_ops_impl.h | 54 CollectiveOpsImpl() : server_node_(nullptr), local_rank_(0), server_num_(0) {} in CollectiveOpsImpl()
69 uint32_t server_num_; variable
|
| D | distributed_metadata_store.h | 72 server_num_(0), in DistributedMetadataStore()
95 uint32_t server_num_; variable
|
| D | distributed_count_service.cc | 30 server_num_ = ps::PSContext::instance()->initial_server_num(); in Initialize()
180 server_num_ = IntToUint(server_node_->server_num()); in ReInitForScaling()
182 << server_num_; in ReInitForScaling()
331 for (uint32_t i = 1; i < server_num_; i++) { in TriggerFirstCountEvent()
357 for (uint32_t i = 1; i < server_num_; i++) { in TriggerLastCountEvent()
|
| D | distributed_metadata_store.cc | 29 server_num_ = ps::PSContext::instance()->initial_server_num(); in Initialize()
160 server_num_ = IntToUint(server_node_->server_num()); in ReInitForScaling()
162 << server_num_; in ReInitForScaling()
173 for (uint32_t i = 0; i < server_num_; i++) { in InitHashRing()
|
| D | server.h | 79 server_num_(0),
198 uint32_t server_num_; variable
|
| D | server.cc | 140 server_num_ = ps::PSContext::instance()->initial_server_num(); in InitServerContext()
142 if (encrypt_type == ps::kPWEncryptType && server_num_ > 1) { in InitServerContext()
143 …N) << "Only single server is supported for PW_ENCRYPT now, but got server_num is:." << server_num_; in InitServerContext()
169 … worker_num_, server_num_, task_executor_); in InitCommunicatorWithServer()
|
| D | distributed_count_service.h | 113 uint32_t server_num_; variable
|
| /third_party/mindspore/mindspore/ccsrc/ps/core/ |
| D | abstract_node.cc | 184 if (!CommUtil::ValidateRankId(node_role, rank_id, worker_num_, server_num_)) { in Send()
186 << ", the server num:" << server_num_ << ", the rank id:" << rank_id; in Send()
214 if (!CommUtil::ValidateRankId(node_role, rank_ids.at(it), worker_num_, server_num_)) { in Send()
216 … << ", the server num:" << server_num_ << ", the rank id:" << rank_ids.at(it); in Send()
248 if (!CommUtil::ValidateRankId(node_role, rank_id, worker_num_, server_num_)) { in Send()
250 << ", the server num:" << server_num_ << ", the rank id:" << rank_id; in Send()
307 if (!CommUtil::ValidateRankId(node_role, rank_ids.at(it), worker_num_, server_num_)) { in Send()
309 … << ", the server num:" << server_num_ << ", the rank id:" << rank_ids.at(it); in Send()
337 if (!CommUtil::ValidateRankId(node_role, rank_id, worker_num_, server_num_)) { in CollectiveSendAsync()
339 << ", the server num:" << server_num_ << ", the rank id:" << rank_id; in CollectiveSendAsync()
[all …]
|
| D | abstract_node.h | 50 server_num_(-1), in AbstractNode()
244 int32_t server_num_; variable
|
| /third_party/mindspore/mindspore/ccsrc/ps/ |
| D | ps_context.cc | 52 server_num_ = std::strtol(common::GetEnv(kEnvPServerNum).c_str(), nullptr, kBase); in SetPSEnable()
66 …cluster_config_ = std::make_unique<core::ClusterConfig>(worker_num_, server_num_, scheduler_host_,… in SetPSEnable()
137 uint32_t PSContext::initial_server_num() const { return server_num_; } in initial_server_num()
269 server_num_ = server_num; in set_server_num()
271 uint32_t PSContext::server_num() const { return server_num_; } in server_num()
|
| D | worker.cc | 25 server_num_ = PSContext::instance()->initial_server_num(); in Run()
193 for (int64_t i = 0; i < server_num_; i++) { in AddEmbeddingTable()
194 size_t local_row_cnt = LongToSize(Util::LocalShard(row_count, i, server_num_)); in AddEmbeddingTable()
450 if (server_num_ == 0) { in AddKeyByHashMod()
453 key_to_server_id_[key] = static_cast<int64_t>(key % server_num_); in AddKeyByHashMod()
835 partition->resize(LongToSize(server_num_)); in RoundRobinPartitioner()
870 partition->resize(LongToSize(server_num_)); in WorkerInitEmbeddingPartitioner()
932 partition->resize(LongToSize(server_num_)); in BroadcastPartitioner()
933 for (size_t i = 0; i < LongToSize(server_num_); i++) { in BroadcastPartitioner()
|
| D | ps_context.h | 197 server_num_(0), in PSContext()
236 uint32_t server_num_; variable
|
| D | worker.h | 84 Worker() : server_num_(-1), running_(false), key_cnt_(0) {} in Worker()
129 int64_t server_num_; variable
|
| /third_party/mindspore/mindspore/ccsrc/ps/core/communicator/ |
| D | tcp_communicator.h | 68 server_num_(0), in TcpCommunicator()
113 uint32_t server_num_; variable
|