比较与torch.distributed.new_group的功能差异
torch.distributed.new_group
torch.distributed.new_group(
ranks=None,
timeout=datetime.timedelta(0, 1800),
backend=None
)
更多内容详见torch.distributed.new_group。
mindspore.communication.create_group
mindspore.communication.create_group(group, rank_ids)
使用方式
PyTorch：该接口传入待构建通信域rank列表，指定backend创建指定的通信域，并返回创建的通信域。
MindSpore：该接口传入group名字，以及待构建通信域rank列表，创建一个以传入的group名字为key的通信域，不返回任何值。
分类 |
子类 |
PyTorch |
MindSpore |
差异 |
|---|---|---|---|---|
参数 |
参数1 |
- |
group |
PyTorch无此参数，MindSpore：group名字 |
参数2 |
ranks |
rank_ids |
功能一致，参数名有差异 |
|
参数3 |
timeout |
- |
PyTorch：超时阈值，MindSpore无此参数且需在调用该接口之前配置相应环境变量 |
|
参数4 |
backend |
- |
PyTorch：集合通信后端，MindSpore无此参数且需在调用该接口之前配置相应环境变量 |