CANN/hccl AllGatherV接口文档

CANN/hccl AllGatherV接口文档 HcclAllGatherV【免费下载链接】hccl集合通信库Huawei Collective Communication Library简称HCCL是基于昇腾AI处理器的高性能集合通信库为计算集群提供高性能、高可靠的通信方案项目地址: https://gitcode.com/cann/hccl产品支持情况Ascend 950PR/Ascend 950DT支持Atlas A3 训练系列产品/Atlas A3 推理系列产品支持Atlas A2 训练系列产品/Atlas A2 推理系列产品支持Atlas 推理系列产品支持Atlas 训练系列产品不支持[!NOTE]说明 针对Atlas A2 训练系列产品/Atlas A2 推理系列产品仅支持Atlas 800T A2 训练服务器、Atlas 900 A2 PoD 集群基础单元、Atlas 200T A2 Box16 异构子框。 针对Atlas 推理系列产品仅支持Atlas 300I Duo 推理卡。功能说明集合通信算子AllGatherV的操作接口将通信域内所有节点的输入按照rank id重新排序然后拼接起来再将结果发送到所有节点的输出。与AllGather算子不同的是AllGatherV算子支持通信域内不同节点的输入配置不同大小的数据量。[!NOTE]说明 针对AllGatherV操作每个节点都接收按照rank id重新排序后的数据集合即每个节点的AllGatherV输出都是一样的。函数原型HcclResult HcclAllGatherV(void *sendBuf, uint64_t sendCount, void *recvBuf, const void *recvCounts, const void *recvDispls, HcclDataType dataType, HcclComm comm, aclrtStream stream)参数说明参数名输入/输出描述sendBuf输入源数据buffer地址。sendCount输入参与AllGatherV操作的sendBuf的数据size。recvBuf输出目的数据buffer地址集合通信结果输出至此buffer中。recvCounts输出参与AllGatherV操作的每个rank在recvBuf中的数据size为uint64类型的数组。该数组的第i个元素表示需要从rank i接收的数据量且该数据量需要与rank i的sendCount值相同。recvDispls输出参与AllGatherV操作的每个rank的数据在recvBuf中的偏移量单位为dataType为uint64类型的数组。该数组的第i个元素表示从rank i接收的数据应该放置在recvBuf中的起始偏移量。dataType输入AllGatherV操作的数据类型HcclDataType类型。针对Ascend 950PR/Ascend 950DT支持数据类型int8、uint8、int16、uint16、int32、uint32、int64、uint64、float8-e5m2、float8-e4m3、float8-e8m0、hifloat8、float16、float32、float64、bfp16。针对Atlas A3 训练系列产品/Atlas A3 推理系列产品支持数据类型int8、uint8、int16、uint16、int32、uint32、int64、uint64、float16、float32、float64、bfp16。针对Atlas A2 训练系列产品/Atlas A2 推理系列产品支持数据类型int8、uint8、int16、uint16、int32、uint32、int64、uint64、float16、float32、float64、bfp16。针对Atlas 300I Duo 推理卡支持数据类型int8、uint8、int16、uint16、int32、uint32、int64、uint64、float16、float32、float64。comm输入集合通信操作所在的通信域。stream输入本rank所使用的stream。返回值HcclResult接口成功返回HCCL_SUCCESS其他失败。约束说明所有rank的recvCounts、recvDispls、dataType均应相同。针对Ascend 950PR/Ascend 950DT仅支持单Server场景仅支持通信算子展开模式为CCUCollective Communication Unit集合通信加速单元的场景。针对Atlas A3 训练系列产品/Atlas A3 推理系列产品仅支持单Server场景。针对Atlas A2 训练系列产品/Atlas A2 推理系列产品仅支持多机对称分布场景不支持非对称分布即卡数非对称的场景。针对Atlas 300I Duo 推理卡仅支持单Server场景单Server中最大支持部署2张Atlas 300I Duo 推理卡即4个NPU。【免费下载链接】hccl集合通信库Huawei Collective Communication Library简称HCCL是基于昇腾AI处理器的高性能集合通信库为计算集群提供高性能、高可靠的通信方案项目地址: https://gitcode.com/cann/hccl创作声明:本文部分内容由AI辅助生成(AIGC),仅供参考