SetHcclContext

功能说明

设置通算融合算子每个通信域对应的context（消息区）地址。

函数原型

1
2

template <uint32_t index>
__aicore__ inline void SetHcclContext(__gm__ uint8_t* context)

参数说明

返回值

无

支持的型号

Atlas推理系列产品AI Core

Atlas推理系列产品Vector Core

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

约束说明

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

调用示例

1
2

// 给GROUP_0设置消息区地址
AscendC::SetHcclContext<HCCL_GROUP_ID_0>(contextGM);