昇腾社区首页
中文
注册
开发者
下载

(beta)torch.distributed.ProcessGroupHCCL

产品支持情况

产品 是否支持
[object Object]Atlas A3 训练系列产品[object Object]
[object Object]Atlas A2 训练系列产品[object Object]
[object Object]Atlas 推理系列产品[object Object]
[object Object]Atlas 训练系列产品[object Object]

功能说明

创建一个ProcessGroupHCCL对象并返回。

函数原型

[object Object]

参数说明

  • storetorch.distributed.distributed_c10d.PrefixStore对象,可以通过构造函数构造。
  • rank:当前节点的rank序号。
  • size:全部通讯节点的数量。
  • timeout:通讯中断时间,判断节点断连,默认值为1800s。

返回值说明

ProcessGroup