From 65760a2f3cba064c35b6694c23c75b4c22e36d30 Mon Sep 17 00:00:00 2001 From: yao_yf Date: Sat, 27 Jun 2020 11:42:12 +0800 Subject: [PATCH] reshape cost computing dont create hccl group --- mindspore/ccsrc/parallel/auto_parallel/operator_costmodel.cc | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/mindspore/ccsrc/parallel/auto_parallel/operator_costmodel.cc b/mindspore/ccsrc/parallel/auto_parallel/operator_costmodel.cc index cbf5bc40c8..8ebfdb7d13 100644 --- a/mindspore/ccsrc/parallel/auto_parallel/operator_costmodel.cc +++ b/mindspore/ccsrc/parallel/auto_parallel/operator_costmodel.cc @@ -465,7 +465,7 @@ double ReshapeCost::GetForwardCommCost(const std::vector &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 &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."; }