|
|
@ -345,9 +345,6 @@ std::string DeviceManager::GenerateGroupNameByRanks(RankList ranks) {
|
|
|
|
// name. Otherwise, let the pointer g point to that group.
|
|
|
|
// name. Otherwise, let the pointer g point to that group.
|
|
|
|
Group DeviceManager::CreateGroup(const std::string &group_name,
|
|
|
|
Group DeviceManager::CreateGroup(const std::string &group_name,
|
|
|
|
const std::vector<mindspore::parallel::Device> &devices) {
|
|
|
|
const std::vector<mindspore::parallel::Device> &devices) {
|
|
|
|
if ((world_group() == NCCL_WORLD_GROUP) && (devices.size() != devices_.size())) {
|
|
|
|
|
|
|
|
MS_LOG(EXCEPTION) << "Do not support sub group for nccl";
|
|
|
|
|
|
|
|
}
|
|
|
|
|
|
|
|
Group g;
|
|
|
|
Group g;
|
|
|
|
(void)gm_.CreateGroup(group_name, devices, &g);
|
|
|
|
(void)gm_.CreateGroup(group_name, devices, &g);
|
|
|
|
return g;
|
|
|
|
return g;
|
|
|
|