Searched refs:costmodel_allreduce_fusion_algorithm (Results 1 – 6 of 6) sorted by relevance
132 cost_model_context.set_cost_model_context(costmodel_allreduce_fusion_algorithm=2)135 cost_model_context.set_cost_model_context(costmodel_allreduce_fusion_algorithm=1)188 cost_model_context.set_cost_model_context(costmodel_allreduce_fusion_algorithm=1)229 cost_model_context.set_cost_model_context(costmodel_allreduce_fusion_algorithm=1)258 cost_model_context.set_cost_model_context(costmodel_allreduce_fusion_algorithm=1)288 cost_model_context.set_cost_model_context(costmodel_allreduce_fusion_algorithm=2)
321 cost_model_context.set_cost_model_context(costmodel_allreduce_fusion_algorithm=1)492 cost_model_context.set_cost_model_context(costmodel_allreduce_fusion_algorithm=2)
99 …int64_t costmodel_allreduce_fusion_algorithm() const { return costmodel_allreduce_fusion_algorithm… in costmodel_allreduce_fusion_algorithm() function
548 costmodel_allreduce_fusion_algorithm=int, costmodel_allreduce_fusion_times=int,
395 auto algorithm = CostModelContext::GetInstance()->costmodel_allreduce_fusion_algorithm(); in ProcessAllreduceFusion()
241 …get_costmodel_allreduce_fusion_algorithm", &CostModelContext::costmodel_allreduce_fusion_algorithm, in PYBIND11_MODULE()