下载
中文
注册

GetHcclContext

功能说明

获取指定Index通信域的context(消息区)地址。

函数原型

1
2
template <uint32_t index>
__aicore__ inline __gm__ uint8_t* __gm__ GetHcclContext(void)

参数说明

表1 模板参数说明

参数名

描述

index

模板参数,用来表示要设置的通信域ID,当前只支持2个通信域,index只能为0/1。

返回值

指定通信域的context(消息区)地址。

支持的型号

Atlas推理系列产品AI Core

Atlas推理系列产品Vector Core

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

约束说明

当前最多只支持2个通信域。

调用示例

1
GM_ADDR contextGM = AscendC::GetHcclContext<HCCL_GROUP_ID_0>();