Browse Source

reshape cost computing dont create hccl group

tags/v0.6.0-beta
yao_yf 5 years ago
parent
commit
65760a2f3c
1 changed files with 2 additions and 2 deletions
  1. +2
    -2
      mindspore/ccsrc/parallel/auto_parallel/operator_costmodel.cc

+ 2
- 2
mindspore/ccsrc/parallel/auto_parallel/operator_costmodel.cc View File

@@ -465,7 +465,7 @@ double ReshapeCost::GetForwardCommCost(const std::vector<TensorInfo> &inputs, co
CheckGlobalDeviceManager();
MS_EXCEPTION_IF_NULL(g_device_manager);
RankList dev_list = g_device_manager->GetDeviceListByStageId(stage_id);
TensorRedistribution tensor_redistribution;
TensorRedistribution tensor_redistribution(false, true);
if (tensor_redistribution.Init(inputs[0].tensor_layout(), outputs[0].tensor_layout(), dev_list) == FAILED) {
MS_LOG(EXCEPTION) << "Failure: tensor_redistribution init failed.";
}
@@ -503,7 +503,7 @@ double ReshapeCost::GetForwardComputationCost(const std::vector<TensorInfo> &inp
CheckGlobalDeviceManager();
MS_EXCEPTION_IF_NULL(g_device_manager);
RankList dev_list = g_device_manager->GetDeviceListByStageId(stage_id);
TensorRedistribution tensor_redistribution;
TensorRedistribution tensor_redistribution(false, true);
if (tensor_redistribution.Init(inputs[0].tensor_layout(), outputs[0].tensor_layout(), dev_list) == FAILED) {
MS_LOG(EXCEPTION) << "Failure: tensor_redistribution init failed.";
}


Loading…
Cancel
Save