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

HcclBarrier

产品支持情况

产品

是否支持

Atlas 350 加速卡

Atlas A3 训练系列产品/Atlas A3 推理系列产品

Atlas A2 训练系列产品/Atlas A2 推理系列产品

Atlas 200I/500 A2 推理产品

Atlas 推理系列产品

Atlas 训练系列产品

针对Atlas A2 训练系列产品/Atlas A2 推理系列产品,仅支持Atlas 800T A2 训练服务器、Atlas 900 A2 PoD 集群基础单元、Atlas 200T A2 Box16 异构子框。

功能说明

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

函数原型

1
HcclResult HcclBarrier(HcclComm comm, aclrtStream stream)

参数说明

参数名

输入/输出

描述

comm

输入

集合通信操作所在的通信域。

stream

输入

本rank所使用的stream。

返回值

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

约束说明

调用示例

1
2
3
4
5
6
7
8
9
HcclComm comm;
aclrtStream stream;
aclrtCreateStream(&stream);

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

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