GetHcclContext
功能说明
获取指定Index通信域的context(消息区)地址。
函数原型
1 2 |
template <uint32_t index> __aicore__ inline __gm__ uint8_t* __gm__ GetHcclContext(void) |
参数说明
参数名 |
描述 |
---|---|
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>(); |
父主题: 通算融合试验接口