每个算子分为,必须先调用“aclnnAllGatherAddGetWorkspaceSize”接口获取计算所需workspace大小以及包含了算子计算流程的执行器,再调用“aclnnAllGatherAdd”接口执行计算。
[object Object]
[object Object]
- 确定性计算:allGatherAdd算子默认确定性实现。
- 当前该示例算子仅支持固定shape:a(240, 256),b(240 * 2, 256),和固定rank_size = 2。
- 所有输入不支持空tensor场景,取值范围在[-5,5]之间。
示例代码如下,仅供参考,具体编译和执行过程请参考本算子README文件调用说明中的。
说明:本示例代码调用了部分HCCL集合通信库接口:HcclGetCommName、HcclCommInitAll、HcclCommDestroy,请参考。
[object Object]Atlas A2 训练系列产品/Atlas A2 推理系列产品[object Object]、[object Object]Atlas A3 训练系列产品/Atlas A3 推理系列产品[object Object]:
[object Object]