Searched refs:group_devices (Results 1 – 5 of 5) sorted by relevance
| /third_party/mindspore/mindspore/ccsrc/frontend/parallel/ops_info/ |
| D | conv2d_info.cc | 378 RankList group_devices; in InferRankBias() local
379 if (dev_matrix.GetDevicesAlongDim(w_index_in_dev_matrix, &group_devices) != SUCCESS) { in InferRankBias()
383 if (group_devices.size() <= 1) { in InferRankBias()
384 MS_LOG(INFO) << name_ << ": The devices' size of w dimension is " << group_devices.size() in InferRankBias()
389 if (group_devices.size() != LongToSize(w_dimension_shard_num_)) { in InferRankBias()
390 MS_LOG(ERROR) << name_ << ": The devices' size of w dimension is " << group_devices.size() in InferRankBias()
395 std::vector<int64_t>::iterator it = std::find(group_devices.begin(), group_devices.end(), rank); in InferRankBias()
396 if (it == group_devices.end()) { in InferRankBias()
398 << rank << ", the device list is " << group_devices; in InferRankBias()
402 rank_bias_ = std::distance(group_devices.begin(), it); in InferRankBias()
[all …]
|
| D | operator_info.cc | 532 RankList group_devices; in CreateGroupByTensorMap() local
533 if (dev_matrix.GetDevicesByTensorMap(tensor_map, &group_devices) != SUCCESS) { in CreateGroupByTensorMap()
537 if (group_devices.size() == 1) { in CreateGroupByTensorMap()
542 Group g = g_device_manager->CreateGroup(group_devices); in CreateGroupByTensorMap()
555 RankList group_devices; in CreateGroupForOptShard() local
557 if (dev_matrix.GetDevicesByTensorMap(tensor_map, &group_devices) != SUCCESS) { in CreateGroupForOptShard()
561 if (group_devices.size() == 1) { in CreateGroupForOptShard()
568 …int64_t index = std::find(group_devices.begin(), group_devices.end(), rank) - group_devices.begin(… in CreateGroupForOptShard()
569 int64_t repeated_size = SizeToLong(group_devices.size()); in CreateGroupForOptShard()
579 group_devices.begin() + index / optimizer_weight_shard_size * optimizer_weight_shard_size, in CreateGroupForOptShard()
[all …]
|
| D | gather_v2_p_info.cc | 685 RankList group_devices; in InferGroup() local
697 if (dev_matrix.GetDevicesAlongDim(SizeToUlong(dim), &group_devices) != SUCCESS) { in InferGroup()
701 if (group_devices.size() == 1) { in InferGroup()
706 MS_LOG(INFO) << name_ << ": The group ranks is " << group_devices; in InferGroup()
707 group_ = g_device_manager->CreateGroup(group_devices); in InferGroup()
|
| /third_party/mindspore/mindspore/ccsrc/frontend/parallel/ |
| D | parameter_manager.cc | 207 RankList group_devices; in GetGroupByTensorInfo() local
208 if (dev_matrix.GetDevicesByTensorMap(tensor_map, &group_devices) != SUCCESS) { in GetGroupByTensorInfo()
212 std::sort(group_devices.begin(), group_devices.end()); in GetGroupByTensorInfo()
213 return group_devices; in GetGroupByTensorInfo()
388 RankList group_devices; in IsFullySplitParameter() local
389 if (dev_matrix.GetDevicesByTensorMap(tensor_map, &group_devices) != SUCCESS) { in IsFullySplitParameter()
394 if (group_devices.size() == 1) { in IsFullySplitParameter()
|
| /third_party/mindspore/mindspore/ccsrc/frontend/parallel/tensor_layout/ |
| D | construct_operator.cc | 254 RankList group_devices; in CreateGroupByDim() local
255 if (dev_matrix.GetDevicesAlongDim(SizeToUlong(axis), &group_devices) != SUCCESS) { in CreateGroupByDim()
259 if (group_devices.size() == 1) { in CreateGroupByDim()
264 Group g = g_device_manager->CreateGroup(group_devices); in CreateGroupByDim()
|