Init

功能说明

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

函数原型

1

__aicore__ inline void Init(GM_ADDR context)

参数说明

返回值

无

支持的型号

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

注意事项

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

调用示例

• 示例1，默认在所有核上创建Hccl对象并调用Init接口初始化。

  Hccl hccl;
  GM_ADDR contextGM1 = GetHcclContext<HCCL_GROUP_ID_0>();  // AscendC自定义算子kernel中，通过此方式获取Hccl context
  hccl.Init(contextGM1);

• 示例2，只在Vector核上创建Hccl对象并调用Init接口初始化，后续只能在Vector核上调用Hccl其他接口。

  if ASCEND_IS_AIV { 
    Hccl hccl;
    GM_ADDR contextGM1 = GetHcclContext<HCCL_GROUP_ID_0>();  // AscendC自定义算子kernel中，通过此方式获取Hccl context
    hccl.Init(contextGM1);
    // 调用Hccl的Prepare、Commit、Wait、Finalize接口
  }

• 示例3，只在Vector的0核上创建Hccl对象并调用Init接口初始化，后续只能在Vector的0核上调用其他接口。

  if ASCEND_IS_AIV { 
    if (GetBlockIdx()==0) {
      Hccl hccl;
      GM_ADDR contextGM1 = GetHcclContext<HCCL_GROUP_ID_0>();  // AscendC自定义算子kernel中，通过此方式获取Hccl context
      hccl.Init(contextGM1);
      // 调用Hccl的Prepare、Commit、Wait、Finalize接口
    }
  }