Init
功能说明
Hccl客户端初始化接口。该接口默认在所有核上工作,用户也可以在调用前通过GetBlockIdx指定其在某一个核上运行。
函数原型
1
|
__aicore__ inline void Init(GM_ADDR context) |
参数说明
参数名 |
输入/输出 |
描述 |
---|---|---|
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));
父主题: Hccl