文档
注册

Init

功能说明

Hccl客户端初始化接口。该接口默认在所有核上工作,用户也可以在调用前通过GetBlockIdx指定其在某一个核上运行。

函数原型

1
__aicore__ inline void Init(GM_ADDR context)

参数说明

表1 接口参数说明

参数名

输入/输出

描述

context

输入

通信上下文,对应数据结构HcclCombineOpParam,包含rankDim,rankID等信息,具体参数请参考表2

返回值

支持的型号

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

注意事项

  • 该接口只能在AI Cube核或者AI Vector核两者之一上调用

调用示例

// 假设已构造好HcclCombineOpParam对象hcclCombineOpParam
Hccl hccl;
hccl.Init(reinterpret_cast<GM_ADDR>(&hcclCombineOpParam));
搜索结果
找到“0”个结果

当前产品无相关内容

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