Home
last modified time | relevance | path

Searched refs:costmodel_allreduce_fusion_allreduce_bandwidth (Results 1 – 6 of 6) sorted by relevance

/third_party/mindspore/mindspore/ccsrc/frontend/parallel/
Dcostmodel_context.h116 double costmodel_allreduce_fusion_allreduce_bandwidth() const { in costmodel_allreduce_fusion_allreduce_bandwidth() function
/third_party/mindspore/tests/ut/python/parallel/
Dtest_allreduce_fusion.py169 cost_model_context.set_cost_model_context(costmodel_allreduce_fusion_allreduce_bandwidth=0.2)
291 … cost_model_context.set_cost_model_context(costmodel_allreduce_fusion_allreduce_bandwidth=0.000001)
Dtest_auto_parallel_resnet.py495 … cost_model_context.set_cost_model_context(costmodel_allreduce_fusion_allreduce_bandwidth=0.000001)
/third_party/mindspore/mindspore/parallel/
D_cost_model_context.py551 costmodel_allreduce_fusion_allreduce_bandwidth=float,
/third_party/mindspore/mindspore/ccsrc/frontend/parallel/allreduce_fusion/
Dallreduce_fusion.cc336 …reduce_bandwidth_ = CostModelContext::GetInstance()->costmodel_allreduce_fusion_allreduce_bandwidt… in GetSetFusionByBackwardCompAndAllreduceTimeParams()
/third_party/mindspore/mindspore/ccsrc/pipeline/jit/
Dinit.cc265 &CostModelContext::costmodel_allreduce_fusion_allreduce_bandwidth, in PYBIND11_MODULE()