下载
EN
注册
我要评分
文档获取效率
文档正确性
内容完整性
文档易理解
在线提单
论坛求助
昇腾小AI

SetHcclContext

功能说明

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

函数原型

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

参数说明

表1 参数说明

参数名

描述

index

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

context

对应通信域的context(消息区)地址。

返回值

支持的型号

Atlas推理系列产品AI Core

Atlas推理系列产品Vector Core

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

约束说明

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

调用示例

1
2
// 给GROUP_0设置消息区地址
AscendC::SetHcclContext<HCCL_GROUP_ID_0>(contextGM);
搜索结果
找到“0”个结果

当前产品无相关内容

未找到相关内容,请尝试其他搜索词