开发者
下载
[object Object]

[object Object][object Object]undefined
[object Object]
  • 接口功能:完成通信和加法的融合。
  • 计算公式gatherOut=AllGather(a0,a1)gatherOut=AllGather(a0, a1) c[i]=gatherOut[i]+b[i]c[i]=gatherOut[i] + b[i]
[object Object]

每个算子分为,必须先调用“aclnnAllGatherAddGetWorkspaceSize”接口获取计算所需workspace大小以及包含了算子计算流程的执行器,再调用“aclnnAllGatherAdd”接口执行计算。

[object Object]
[object Object]
[object Object]
  • 参数说明:

    [object Object]
  • 返回值:

    aclnnStatus:返回状态码,具体参见

    第一段接口完成入参校验,出现以下场景时报错:

    [object Object]
[object Object]
  • 参数说明:

    [object Object]
  • 返回值:

    返回aclnnStatus状态码,具体参见

[object Object]
  • 确定性计算:allGatherAdd算子默认确定性实现。
  • 当前该示例算子仅支持固定shape:a(240, 256),b(240 * 2, 256),和固定rank_size = 2。
  • 所有输入不支持空tensor场景,取值范围在[-5,5]之间。
[object Object]

示例代码如下,仅供参考,具体编译和执行过程请参考本算子README文件调用说明中的

说明:本示例代码调用了部分HCCL集合通信库接口:HcclGetCommName、HcclCommInitAll、HcclCommDestroy,请参考

  • [object Object]Atlas A2 训练系列产品/Atlas A2 推理系列产品[object Object]、[object Object]Atlas A3 训练系列产品/Atlas A3 推理系列产品[object Object]:

    [object Object]