Home
last modified time | relevance | path

Searched refs:rank_list (Results 1 – 11 of 11) sorted by relevance

/third_party/mindspore/tests/ut/python/parallel/
Dtest_distribute_predict.py79 rank_list = _infer_rank_list(train_map, predict_map)["weight"]
80 assert list(rank_list[0]) == [0, 1, 2, 3, 4, 5, 6, 7]
81 assert rank_list[1] is False
88 rank_list = _infer_rank_list(train_map, predict_map)
90 assert rank_list == expect_map
97 rank_list = _infer_rank_list(train_map, predict_map)
99 assert rank_list == expect_map
106 rank_list = _infer_rank_list(train_map, predict_map)["weight"]
107 assert list(rank_list[0]) == [0, 1, 2, 3, 4, 5, 6, 7]
108 assert rank_list[1] is False
[all …]
/third_party/mindspore/tests/ut/cpp/parallel/
Ddevice_matrix_test.cc89 RankList rank_list; in TEST_F() local
92 arr.GetDevicesByTensorMap(tensor_map, &rank_list); in TEST_F()
94 ASSERT_EQ(rank_list, rank_list_except); in TEST_F()
100 RankList rank_list; in TEST_F() local
103 arr.GetDevicesByTensorMap(tensor_map, &rank_list); in TEST_F()
105 ASSERT_EQ(rank_list, rank_list_except); in TEST_F()
111 RankList rank_list; in TEST_F() local
114 arr.GetDevicesByTensorMap(tensor_map, &rank_list); in TEST_F()
116 ASSERT_EQ(rank_list, rank_list_except); in TEST_F()
/third_party/mindspore/mindspore/parallel/
D_utils.py332 rank_list = array[index].flatten()
334 ret[param_name] = (rank_list, False)
346 if len(rank_list) == 1:
347 ret[param_name] = (rank_list, True)
348 elif len(rank_list) == dev_num:
349 ret[param_name] = ([rank_list[local_rank]], True)
351 ret[param_name] = (rank_list, False)
353 ret[param_name] = (rank_list, False)
/third_party/mindspore/mindspore/ccsrc/frontend/parallel/
Ddevice_matrix.cc118 Status DeviceMatrix::GetDevicesByTensorMap(const Shape &tensor_map, RankList *rank_list) { in GetDevicesByTensorMap() argument
157 rank_list->push_back(dev_list_[loop_local_rank]); in GetDevicesByTensorMap()
Ddevice_matrix.h39 Status GetDevicesByTensorMap(const Shape &tensor_map, RankList *rank_list);
Dparameter_manager.cc386 RankList rank_list = g_device_manager->GetDeviceListInThisStage(); in IsFullySplitParameter() local
387 DeviceMatrix dev_matrix(rank, rank_list, dev_mat_shape); in IsFullySplitParameter()
/third_party/mindspore/mindspore/nn/layer/
Dnormalization.py100 self.rank_list = self.list_group(self.device_list, self.group_device_num)
101 self.rank_list_idx = len(self.rank_list)
190 if self.rank_id in self.rank_list[i]:
195 management.create_group(SYNC_BN_GROUP_NAME, self.rank_list[i])
/third_party/mindspore/mindspore/nn/optim/
Doptimizer.py611 rank_list = ()
614 rank_list = rank_list + (count,)
618 return rank_list
/third_party/mindspore/mindspore/nn/wrap/
Dcell_wrapper.py660 rank_list = [id for id in range(0, get_group_size())]
661 create_group("BroadcastWorldGroup", rank_list)
/third_party/mindspore/mindspore/ccsrc/frontend/parallel/pipeline_transformer/
Dpipeline_transformer.cc179 std::vector<int64_t> rank_list; in CreateForwardGroup() local
184 rank_list.push_back(rank_id + per_stage_rank_num_ * (i - stage_id)); 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()
/third_party/mindspore/mindspore/train/
Dserialization.py1357 rank_list = _infer_rank_list(train_strategy, predict_strategy)
1370 if param.name not in rank_list.keys():
1377 param_rank = rank_list[param.name][0]
1378 skip_merge_split = rank_list[param.name][1]