Searched refs:tail_percent (Results 1 – 5 of 5) sorted by relevance
281 std::vector<double> AllreduceFusion::GenerateCostMap(int64_t fusion_times, double tail_percent) con… in GenerateCostMap()282 double offset = allreduce_graph_.max() * (1 - tail_percent) / (fusion_times - 1); in GenerateCostMap()290 cost_map.push_back(allreduce_graph_.max() * (1 - tail_percent)); in GenerateCostMap()302 auto tail_percent = CostModelContext::GetInstance()->costmodel_allreduce_fusion_tail_percent(); in SetFusionByBackwardCompTime() local303 if (tail_percent < 0 || tail_percent >= 1) { in SetFusionByBackwardCompTime()304 MS_LOG(INFO) << "'costmodel_allreduce_fusion_tail_percent' is " << tail_percent in SetFusionByBackwardCompTime()308 const auto cost_map = GenerateCostMap(fusion_times, tail_percent); in SetFusionByBackwardCompTime()
60 std::vector<double> GenerateCostMap(int64_t fusion_times, double tail_percent) const;
147 …tail_percent = cost_model_context.get_cost_model_context('costmodel_allreduce_fusion_tail_percent')148 assert tail_percent == 0.2150 …tail_percent = cost_model_context.get_cost_model_context('costmodel_allreduce_fusion_tail_percent')151 assert tail_percent == 0.1
190 void CostModelContext::set_costmodel_allreduce_fusion_tail_percent(double tail_percent) { in set_costmodel_allreduce_fusion_tail_percent() argument191 costmodel_allreduce_fusion_tail_percent_ = tail_percent; in set_costmodel_allreduce_fusion_tail_percent()
348 def set_costmodel_allreduce_fusion_tail_percent(self, tail_percent): argument361 self._context_handle.set_costmodel_allreduce_fusion_tail_percent(tail_percent)