Home
last modified time | relevance | path

Searched refs:local_rank_ (Results 1 – 10 of 10) sorted by relevance

/third_party/mindspore/mindspore/ccsrc/fl/server/
Dcollective_ops_impl.cc25 local_rank_ = server_node_->rank_id(); in Initialize()
57 uint32_t send_to_rank = (local_rank_ + 1) % rank_size; in RingAllReduce()
58 uint32_t recv_from_rank = (local_rank_ - 1 + rank_size) % rank_size; in RingAllReduce()
59 …G) << "AllReduce count:" << count << ", rank_size:" << rank_size << ", local_rank_:" << local_rank_ in RingAllReduce()
70 size_t send_chunk_index = (local_rank_ - i + rank_size) % rank_size; in RingAllReduce()
75 size_t recv_chunk_index = (local_rank_ - i - 1 + rank_size) % rank_size; in RingAllReduce()
108 size_t send_chunk_index = (local_rank_ - i + 1 + rank_size) % rank_size; in RingAllReduce()
112 size_t recv_chunk_index = (local_rank_ - i + rank_size) % rank_size; in RingAllReduce()
143 …G(DEBUG) << "Reduce Broadcast AllReduce rank_size:" << rank_size << ", local_rank_:" << local_rank_ in ReduceBroadcastAllReduce()
153 if (local_rank_ == 0) { in ReduceBroadcastAllReduce()
[all …]
Ddistributed_metadata_store.cc28 local_rank_ = server_node_->rank_id(); in Initialize()
51 if (local_rank_ == stored_rank) { in RegisterMetadata()
57 MS_LOG(INFO) << "Rank " << local_rank_ << " register storage for metadata " << name; in RegisterMetadata()
71 if (local_rank_ == stored_rank) { in ResetMetadata()
77 MS_LOG(INFO) << "Rank " << local_rank_ << " reset metadata for " << name; in ResetMetadata()
92 …MS_LOG(INFO) << "Rank " << local_rank_ << " update value for " << name << " which is stored in ran… in UpdateMetadata()
93 if (local_rank_ == stored_rank) { in UpdateMetadata()
130 …MS_LOG(INFO) << "Rank " << local_rank_ << " get metadata for " << name << " which is stored in ran… in GetMetadata()
131 if (local_rank_ == stored_rank) { in GetMetadata()
159 local_rank_ = server_node_->rank_id(); in ReInitForScaling()
[all …]
Ddistributed_count_service.cc29 local_rank_ = server_node_->rank_id(); in Initialize()
58 …MS_LOG(INFO) << "Rank " << local_rank_ << " register counter for " << name << " count:" << global_… in RegisterCounter()
60 if (local_rank_ == counting_server_rank_) { in RegisterCounter()
70 …MS_LOG(INFO) << "Rank " << local_rank_ << " reinitialize counter for " << name << " count:" << glo… in ReInitCounter()
71 if (local_rank_ == counting_server_rank_) { in ReInitCounter()
84 MS_LOG(INFO) << "Rank " << local_rank_ << " reports count for " << name << " of " << id; in Count()
85 if (local_rank_ == counting_server_rank_) { in Count()
136 MS_LOG(INFO) << "Rank " << local_rank_ << " query whether count reaches threshold for " << name; in CountReachThreshold()
137 if (local_rank_ == counting_server_rank_) { in CountReachThreshold()
165 if (local_rank_ == counting_server_rank_) { in ResetCounter()
[all …]
Dcollective_ops_impl.h54 CollectiveOpsImpl() : server_node_(nullptr), local_rank_(0), server_num_(0) {} in CollectiveOpsImpl()
68 uint32_t local_rank_; variable
Ddistributed_metadata_store.h71 local_rank_(0), in DistributedMetadataStore()
94 uint32_t local_rank_; variable
Ddistributed_count_service.h85 uint32_t local_rank() { return local_rank_; } in local_rank()
112 uint32_t local_rank_; variable
/third_party/mindspore/mindspore/ccsrc/fl/server/kernel/round/
Dpush_metrics_kernel.h35 PushMetricsKernel() : local_rank_(0) {} in PushMetricsKernel()
48 uint32_t local_rank_; variable
Dpush_weight_kernel.h35 PushWeightKernel() : executor_(nullptr), local_rank_(0) {} in PushWeightKernel()
51 uint32_t local_rank_; variable
Dpush_metrics_kernel.cc25 void PushMetricsKernel::InitKernel(size_t) { local_rank_ = DistributedCountService::GetInstance().l… in InitKernel()
87 …if (!DistributedCountService::GetInstance().Count(name_, std::to_string(local_rank_), &count_reaso… in PushMetrics()
Dpush_weight_kernel.cc30 local_rank_ = DistributedCountService::GetInstance().local_rank(); in InitKernel()
112 …if (!DistributedCountService::GetInstance().Count(name_, std::to_string(local_rank_), &count_reaso… in PushWeight()