Home
last modified time | relevance | path

Searched refs:g_device_manager (Results 1 – 25 of 53) sorted by relevance

123

/third_party/mindspore/tests/ut/cpp/parallel/auto_parallel/
Doperator_costmodel_test.cc49 g_device_manager = std::make_shared<DeviceManager>(); in SetUp()
50 g_device_manager->Init(dev_list, local_dev, stage_map, "hccl"); in SetUp()
106 g_device_manager = std::make_shared<DeviceManager>(); in SetUp()
107 g_device_manager->Init(dev_list, local_dev, stage_map, "hccl"); in SetUp()
158 g_device_manager = std::make_shared<DeviceManager>(); in SetUp()
159 g_device_manager->Init(dev_list, local_dev, stage_map, "hccl"); in SetUp()
Dedge_costmodel_test.cc59 g_device_manager = std::make_shared<DeviceManager>(); in SetUp()
60 g_device_manager->Init(dev_list, local_dev, stage_map, "hccl"); in SetUp()
/third_party/mindspore/mindspore/ccsrc/frontend/parallel/
Ddevice_manager.cc29 DeviceManagerPtr g_device_manager = nullptr; variable
86 if (g_device_manager) { in InitDevice()
87 auto gm = g_device_manager->group_manager(); in InitDevice()
88 g_device_manager = std::make_shared<DeviceManager>(); in InitDevice()
89 g_device_manager->set_group_manager(gm); in InitDevice()
91 g_device_manager = std::make_shared<DeviceManager>(); in InitDevice()
93 if (g_device_manager->Init(devices, global_rank, stage_map, backend) == SUCCESS) { in InitDevice()
103 if (g_device_manager == nullptr) { in CheckGlobalDeviceManager()
Dparameter_manager.cc201 int64_t rank = g_device_manager->global_rank(); in GetGroupByTensorInfo()
202 RankList stage_device_list = g_device_manager->GetDeviceListInThisStage(); in GetGroupByTensorInfo()
352 auto dev_num = g_device_manager->stage_device_num(); in HandleNoUsedParameter()
385 int64_t rank = g_device_manager->global_rank(); in IsFullySplitParameter()
386 RankList rank_list = g_device_manager->GetDeviceListInThisStage(); in IsFullySplitParameter()
/third_party/mindspore/mindspore/ccsrc/frontend/parallel/auto_parallel/
Doperator_costmodel.cc109 MS_EXCEPTION_IF_NULL(g_device_manager); in GetBackwardCommCost()
110 auto total_device_num = g_device_manager->GetDeviceListByStageId(stage_id).size(); in GetBackwardCommCost()
154 MS_EXCEPTION_IF_NULL(g_device_manager); in GetBackwardComputationCost()
155 auto total_device_num = g_device_manager->GetDeviceListByStageId(stage_id).size(); in GetBackwardComputationCost()
211 MS_EXCEPTION_IF_NULL(g_device_manager); in GetBackwardCommCost()
212 auto total_device_num = g_device_manager->GetDeviceListByStageId(stage_id).size(); in GetBackwardCommCost()
277 MS_EXCEPTION_IF_NULL(g_device_manager); in GetBackwardCommCost()
278 auto total_device_num = g_device_manager->GetDeviceListByStageId(stage_id).size(); in GetBackwardCommCost()
445 MS_EXCEPTION_IF_NULL(g_device_manager); in GetBackwardCommCost()
446 auto total_device_num = g_device_manager->GetDeviceListByStageId(stage_id).size(); in GetBackwardCommCost()
[all …]
/third_party/mindspore/tests/ut/cpp/parallel/tensor_layout/
Dtensor_redistribution_test.cc44 g_device_manager = std::make_shared<DeviceManager>(); in SetUp()
45 g_device_manager->Init(dev_list, local_dev, stage_map, "hccl"); in SetUp()
87 RankList dev_list = g_device_manager->GetDeviceListByStageId(0); in TEST_F()
140 RankList dev_list = g_device_manager->GetDeviceListByStageId(0); in TEST_F()
193 RankList dev_list = g_device_manager->GetDeviceListByStageId(0); in TEST_F()
Dconstruct_operator_test.cc53 g_device_manager = std::make_shared<DeviceManager>(); in SetUp()
54 g_device_manager->Init(dev_list, local_dev, stage_map, "hccl"); in SetUp()
70 RankList used_dev_list = g_device_manager->GetDeviceListByStageId(0); in SetUp()
Dredistribution_operator_infer_test.cc44 g_device_manager = std::make_shared<DeviceManager>(); in SetUp()
45 g_device_manager->Init(dev_list, local_dev, stage_map, "hccl"); in SetUp()
/third_party/mindspore/tests/ut/cpp/parallel/
Dvirtual_dataset_test.cc53 g_device_manager = std::make_shared<DeviceManager>(); in SetUp()
54 g_device_manager->Init(dev_list, local_dev, stage_map, "hccl"); in SetUp()
/third_party/mindspore/mindspore/ccsrc/frontend/parallel/graph_util/
Dpipeline_split_utils.cc79 MS_EXCEPTION_IF_NULL(g_device_manager); in IsLastStage()
80 auto stage_num = g_device_manager->stage_num(); in IsLastStage()
81 auto stage_id = g_device_manager->stage_id(); in IsLastStage()
289 MS_EXCEPTION_IF_NULL(g_device_manager); in ReorderForForward()
293 auto stage_num = g_device_manager->stage_num(); in ReorderForForward()
294 auto stage_id = g_device_manager->stage_id(); in ReorderForForward()
305 MS_EXCEPTION_IF_NULL(g_device_manager); in ReorderForBackward()
309 auto stage_num = g_device_manager->stage_num(); in ReorderForBackward()
310 auto stage_id = g_device_manager->stage_id(); in ReorderForBackward()
585 auto stage_id = g_device_manager->stage_id(); in GetBorderNode()
/third_party/mindspore/tests/ut/cpp/parallel/ops_info/
Dactivation_test.cc56 g_device_manager = std::make_shared<DeviceManager>(); in SetUp()
57 g_device_manager->Init(dev_list, local_dev, stage_map, "hccl"); in SetUp()
Dtanh_info_test.cc54 g_device_manager = std::make_shared<DeviceManager>(); in SetUp()
55 g_device_manager->Init(dev_list, local_dev, stage_map, "hccl"); in SetUp()
Dlog_softmax_info_test.cc54 g_device_manager = std::make_shared<DeviceManager>(); in SetUp()
55 g_device_manager->Init(dev_list, local_dev, stage_map, "hccl"); in SetUp()
Dpow_info_test.cc54 g_device_manager = std::make_shared<DeviceManager>(); in SetUp()
55 g_device_manager->Init(dev_list, local_dev, stage_map, "hccl"); in SetUp()
Dgelu_info_test.cc54 g_device_manager = std::make_shared<DeviceManager>(); in SetUp()
55 g_device_manager->Init(dev_list, local_dev, stage_map, "hccl"); in SetUp()
Dgenerate_strategy_test.cc50 g_device_manager = std::make_shared<DeviceManager>(); in SetUp()
51 g_device_manager->Init(dev_list, local_dev, stage_map, "hccl"); in SetUp()
Dsoftmax_info_test.cc54 g_device_manager = std::make_shared<DeviceManager>(); in SetUp()
55 g_device_manager->Init(dev_list, local_dev, stage_map, "hccl"); in SetUp()
Dactivation_info_test.cc54 g_device_manager = std::make_shared<DeviceManager>(); in SetUp()
55 g_device_manager->Init(dev_list, local_dev, stage_map, "hccl"); in SetUp()
Donehot_info_test_axis_0.cc54 g_device_manager = std::make_shared<DeviceManager>(); in SetUp()
55 g_device_manager->Init(dev_list, local_dev, stage_map, "hccl"); in SetUp()
Dl2_normalize_info_test.cc54 g_device_manager = std::make_shared<DeviceManager>(); in SetUp()
55 g_device_manager->Init(dev_list, local_dev, stage_map, "hccl"); in SetUp()
Dsoftmax_entropy_loss_info_test.cc54 g_device_manager = std::make_shared<DeviceManager>(); in SetUp()
55 g_device_manager->Init(dev_list, local_dev, stage_map, "hccl"); in SetUp()
Dtmpidentity_test.cc54 g_device_manager = std::make_shared<DeviceManager>(); in SetUp()
55 g_device_manager->Init(dev_list, local_dev, stage_map, "hccl"); in SetUp()
Dtranspose_test.cc54 g_device_manager = std::make_shared<DeviceManager>(); in SetUp()
55 g_device_manager->Init(dev_list, local_dev, stage_map, "hccl"); in SetUp()
/third_party/mindspore/mindspore/ccsrc/frontend/parallel/pipeline_transformer/
Dpipeline_transformer.cc180 auto rank_id = g_device_manager->global_rank(); in CreateForwardGroup()
181 auto stage_id = g_device_manager->stage_id(); in CreateForwardGroup()
182 auto stage_num = g_device_manager->stage_num(); in CreateForwardGroup()
186 auto dev_list = g_device_manager->CreateDeviceListByRankList(rank_list); in CreateForwardGroup()
187 auto g = g_device_manager->CreateGroup(rank_list); in CreateForwardGroup()
189 auto g_back = g_device_manager->CreateGroup(g_back_name, dev_list); in CreateForwardGroup()
227 MS_EXCEPTION_IF_NULL(g_device_manager); in Coloring()
228 auto stage_num = g_device_manager->stage_num(); in Coloring()
821 auto stage_num = g_device_manager->stage_num(); in CutBorder()
/third_party/mindspore/mindspore/ccsrc/frontend/parallel/tensor_layout/
Dconstruct_operator.cc251 MS_EXCEPTION_IF_NULL(g_device_manager); in CreateGroupByDim()
252 int64_t rank = g_device_manager->global_rank(); in CreateGroupByDim()
264 Group g = g_device_manager->CreateGroup(group_devices); in CreateGroupByDim()

123