HcclAlltoAll

函数原型

HcclResult HcclAlltoAll(const void *sendBuf, uint64_t sendCount, HcclDataType sendType, const void *recvBuf, uint64_t recvCount, HcclDataType recvType, HcclComm comm, aclrtStream stream)

功能说明

集合通信alltoall操作接口。向通信域内所有rank发送相同数据量的数据,并从所有rank接收相同数据量的数据。

参数说明

参数名

输入/输出

描述

sendBuf

输入

源数据buffer地址。

sendCount

输入

表示向每个rank发送的数据量。

sendType

输入

发送数据的数据类型。

针对Atlas 训练系列产品,支持数据类型:int8, uint8, int16, uint16, int32, uint32, int64, uint64, float16, float32, float64。

针对Atlas A2 训练系列产品,支持数据类型:int8, uint8, int16, uint16, int32, uint32, int64, uint64, float16, float32, float64,bfp16。

recvBuf

输出

目的数据buffer地址,集合通信结果输出至此buffer中。

recvCount

输入

表示从每个rank接收的数据量。

recvType

输入

接收数据的数据类型

针对Atlas 训练系列产品,支持数据类型:int8, uint8, int16, uint16, int32, uint32, int64, uint64, float16, float32, float64。

针对Atlas A2 训练系列产品,支持数据类型:int8, uint8, int16, uint16, int32, uint32, int64, uint64, float16, float32, float64,bfp16。

comm

输入

集合通信操作所在的通信域。

stream

输入

本rank所使用的stream。

返回值

HcclResult:接口成功返回HCCL_SUCCESS。其他失败。

约束说明