SetHcclContext
产品支持情况
- Ascend 950PR/Ascend 950DT:不支持
- Atlas A3 训练系列产品/Atlas A3 推理系列产品:支持
- Atlas A2 训练系列产品/Atlas A2 推理系列产品:支持
- Atlas 200I/500 A2 推理产品:不支持
- Atlas 推理系列产品AI Core:支持
- Atlas 推理系列产品Vector Core:支持
- Atlas 训练系列产品:不支持
功能说明
设置通算融合算子每个通信域对应的context(消息区)地址。
函数原型
Text
template <uint32_t index>
__aicore__ inline void SetHcclContext(__gm__ uint8_t* context)
参数说明
表1 参数说明
| 参数名 | 描述 |
|---|---|
| index | 模板参数,用来表示要设置的通信域ID,当前只支持2个通信域,index只能为0/1。 |
| context | 对应通信域的context(消息区)地址。 |
返回值说明
无
约束说明
当前最多只支持2个通信域。
调用示例
Text
// 给GROUP_0设置消息区地址
AscendC::SetHcclContext<0>(contextGM);