| /third_party/mindspore/tests/ut/cpp/parallel/ |
| D | device_matrix_test.cc | 34 RankList dev_list = {0, 1, 2, 3, 4, 5}; in TEST_F()
38 std::vector<RankList> group_list; in TEST_F()
40 std::vector<RankList> group_list_expect = {{0, 3}, {0, 1, 2}}; in TEST_F()
45 RankList dev_list = {0, 1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11}; in TEST_F()
49 std::vector<RankList> group_list; in TEST_F()
51 std::vector<RankList> group_list_expect = {{5, 11}, {2, 5}, {3, 4, 5}}; in TEST_F()
56 RankList dev_list = {0, 1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 12, 13, 14, 15}; in TEST_F()
60 std::vector<RankList> group_list; in TEST_F()
62 std::vector<RankList> group_list_expect = {{5, 13}, {5}, {1, 3, 5, 7}, {4, 5}}; in TEST_F()
67 RankList dev_list; in TEST_F()
[all …]
|
| D | device_manager_test.cc | 72 RankList dev_list; in TEST_F()
73 RankList stage_map; in TEST_F()
88 RankList dev_list_0 = dm_.GetDeviceListByStageId(0); in TEST_F()
89 RankList dev_list_1 = dm_.GetDeviceListByStageId(1); in TEST_F()
93 RankList::iterator it = dev_list_0.begin(); in TEST_F()
110 RankList rlist; in TEST_F()
122 RankList dev_list; in TEST_F()
123 RankList stage_map; in TEST_F()
141 RankList dev_list_0 = dm_.GetDeviceListByStageId(0); in TEST_F()
142 RankList dev_list_1 = dm_.GetDeviceListByStageId(1); in TEST_F()
|
| D | virtual_dataset_test.cc | 40 RankList dev_list; in SetUp()
46 RankList stage_map; in SetUp()
|
| /third_party/mindspore/mindspore/ccsrc/frontend/parallel/ |
| D | device_matrix.h | 29 using RankList = std::vector<int64_t>; variable
34 DeviceMatrix(int64_t rank, RankList devices, Shape dev_shape);
37 std::vector<RankList> group_list() const { return group_list_; } in group_list()
39 Status GetDevicesByTensorMap(const Shape &tensor_map, RankList *rank_list);
40 Status GetDevicesAlongDim(const uint64_t &dim, RankList *devices);
44 RankList dev_list_;
47 std::vector<RankList> group_list_;
51 std::string ListToString(const RankList &list);
|
| D | device_manager.h | 63 …Status Init(const RankList &devices, int64_t local_device, const RankList &stage_map, const std::s…
66 RankList GetDeviceListByStageId(int64_t stage_id) const;
67 RankList GetDeviceListInThisStage() const;
70 std::vector<Device> CreateDeviceListByRankList(RankList ranks);
72 std::string GenerateGroupNameByRanks(RankList dev_ranks);
74 Group CreateGroup(const RankList &dev_ranks);
|
| D | device_manager.cc | 62 RankList devices, stage_map; in InitDevice()
108 int64_t GetListMemberByIndex(size_t index, const RankList &devices) { in GetListMemberByIndex()
144 Status DeviceManager::Init(const RankList &devices, int64_t global_device_rank, const RankList &sta… in Init()
172 RankList curr_dev_list; in Init()
204 RankList DeviceManager::GetDeviceListInThisStage() const { return GetDeviceListByStageId(stage_id_)… in GetDeviceListInThisStage()
206 RankList DeviceManager::GetDeviceListByStageId(int64_t stage_id) const { in GetDeviceListByStageId()
210 RankList res; in GetDeviceListByStageId()
223 std::vector<Device> DeviceManager::CreateDeviceListByRankList(RankList ranks) { in CreateDeviceListByRankList()
255 std::string DeviceManager::GenerateGroupNameByRanks(RankList ranks) { in GenerateGroupNameByRanks()
296 Group DeviceManager::CreateGroup(const RankList &dev_ranks) { in CreateGroup()
|
| D | device_matrix.cc | 32 DeviceMatrix::DeviceMatrix(int64_t rank, RankList dev_list, Shape dev_shape) in DeviceMatrix()
45 RankList group; in CreateGroupList()
56 Status DeviceMatrix::GetDevicesAlongDim(const uint64_t &dim, RankList *devices) { in GetDevicesAlongDim()
65 RankList group; in GetDevicesAlongDim()
66 std::vector<RankList> local_group_list; in GetDevicesAlongDim()
91 for (RankList &temp : local_group_list) { in GetDevicesAlongDim()
118 Status DeviceMatrix::GetDevicesByTensorMap(const Shape &tensor_map, RankList *rank_list) { in GetDevicesByTensorMap()
175 std::string ListToString(const RankList &list) { in ListToString()
|
| D | parameter_manager.cc | 199 static RankList GetGroupByTensorInfo(const TensorInfo &tensor_info) { in GetGroupByTensorInfo()
202 RankList stage_device_list = g_device_manager->GetDeviceListInThisStage(); in GetGroupByTensorInfo()
207 RankList group_devices; in GetGroupByTensorInfo()
258 RankList first_user_group_list = parameter_slice_info.group_ranks; in CheckParameterSplit()
264 RankList user_group_list = user_slice_info.group_ranks; in CheckParameterSplit()
386 RankList rank_list = g_device_manager->GetDeviceListInThisStage(); in IsFullySplitParameter()
388 RankList group_devices; in IsFullySplitParameter()
|
| /third_party/mindspore/tests/ut/cpp/parallel/auto_parallel/ |
| D | operator_costmodel_test.cc | 36 RankList dev_list; in SetUp()
42 RankList stage_map; in SetUp()
93 RankList dev_list; in SetUp()
99 RankList stage_map; in SetUp()
145 RankList dev_list; in SetUp()
151 RankList stage_map; in SetUp()
|
| /third_party/mindspore/tests/ut/cpp/parallel/tensor_layout/ |
| D | redistribution_operator_infer_test.cc | 31 RankList dev_list; in SetUp()
37 RankList stage_map; in SetUp()
77 RankList dev_list; in InferOperatorCheckAll()
|
| D | tensor_redistribution_test.cc | 31 RankList dev_list; in SetUp()
37 RankList stage_map; 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()
|
| D | construct_operator_test.cc | 42 RankList dev_list; in SetUp()
47 RankList stage_map; in SetUp()
70 RankList used_dev_list = g_device_manager->GetDeviceListByStageId(0); in SetUp()
|
| /third_party/mindspore/mindspore/ccsrc/frontend/parallel/tensor_layout/ |
| D | construct_operator.h | 37 Status Init(const RankList &dev_list, const Shape &dev_matrix_shape);
52 RankList dev_list_;
|
| D | tensor_redistribution.h | 49 Status Init(const TensorLayout &from, const TensorLayout &to, const RankList &dev_list);
74 RankList dev_list_;
|
| D | redistribution_operator_infer.h | 44 Status Init(const TensorLayout &tensor_layout, const Map &out_tensor_map, RankList dev_list,
70 RankList dev_list_;
|
| /third_party/mindspore/tests/ut/cpp/parallel/ops_info/ |
| D | log_softmax_info_test.cc | 41 RankList dev_list; in SetUp()
47 RankList stage_map; in SetUp()
174 RankList dev_list = log_softmax->stage_device_list(); in TEST_F()
|
| D | activation_test.cc | 43 RankList dev_list; in SetUp()
49 RankList stage_map; in SetUp()
|
| D | gelu_info_test.cc | 41 RankList dev_list; in SetUp()
47 RankList stage_map; in SetUp()
|
| D | pow_info_test.cc | 41 RankList dev_list; in SetUp()
47 RankList stage_map; in SetUp()
|
| D | generate_strategy_test.cc | 37 RankList dev_list; in SetUp()
43 RankList stage_map; in SetUp()
|
| D | tanh_info_test.cc | 41 RankList dev_list; in SetUp()
47 RankList stage_map; in SetUp()
|
| D | softmax_info_test.cc | 42 RankList dev_list; in SetUp()
48 RankList stage_map; in SetUp()
|
| D | activation_info_test.cc | 41 RankList dev_list; in SetUp()
47 RankList stage_map; in SetUp()
|
| D | onehot_info_test_axis_0.cc | 41 RankList dev_list; in SetUp()
47 RankList stage_map; in SetUp()
|
| D | softmax_entropy_loss_info_test.cc | 41 RankList dev_list; in SetUp()
47 RankList stage_map; in SetUp()
|