HcclBarrier

产品支持情况

  • Ascend 950PR/Ascend 950DT:支持
  • Atlas A3 训练系列产品/Atlas A3 推理系列产品:支持
  • Atlas A2 训练系列产品/Atlas A2 推理系列产品:支持
  • Atlas 推理系列产品:不支持
  • Atlas 训练系列产品:支持

功能说明

将指定通信域内所有rank的stream阻塞,直到所有rank都下发执行该操作为止。

函数原型

HcclResult HcclBarrier(HcclComm comm, aclrtStream stream)

参数说明

参数名 输入/输出 描述
comm 输入 集合通信操作所在的通信域。
stream 输入 本rank所使用的stream。

返回值

HcclResult:接口成功返回HCCL_SUCCESS,其他失败。

约束说明

调用示例

HcclComm comm;
aclrtStream stream;
aclrtCreateStream(&stream);

// 下发通信任务到该stream,如HcclAllReduce
// ...

// 阻塞等待所有rank均执行Barrier操作
HcclBarrier(comm, stream);