说明: 使用该接口时,请确保驱动固件包和CANN包都为配套的8.0.RC2版本或者配套的更高版本,否则将会引发报错,比如Bus Error等。
接口功能:完成
[object Object]计算和[object Object]通信的融合,先计算后通信。融合算子内部实现计算和通信流水并行。计算公式:
- x1指Matmul计算的左矩阵。
- x2指Matmul计算的右矩阵。
- bias指Matmul计算的偏置。
每个算子分为两段式接口,必须先调用“aclnnMatmulReduceScatterGetWorkspaceSize”接口获取计算所需workspace大小以及包含了算子计算流程的执行器,再调用“aclnnMatmulReduceScatter”接口执行计算。
[object Object]
[object Object]
- 参数说明中shape涉及的变量说明:
- m为卡数rank_size的整数倍。
- k的取值范围为[256, 65535)。
- x1/x2支持的空tensor场景,m和n可以为0,k不可为0,且需满足以下条件:
- m为0,k不为0,n不为0;
- m不为0,k不为0,n为0;
- m为0,k不为0,n为0。
- rank_size为卡数。
- x2支持转置/不转置场景,x1只支持不转置场景。
- x1、x2计算输入的数据类型要和output计算输出的数据类型一致。
- [object Object]Atlas A2 训练系列产品/Atlas A2 推理系列产品[object Object]:
- 支持2、4、8卡,并且仅支持hccs链路all mesh组网。
- 一个模型中的通算融合MC2算子,仅支持相同通信域。
- aclnnMatmulReduceScatter默认非确定性实现,支持通过aclrtCtxSetSysParamOpt开启确定性。
- [object Object]Atlas A3 训练系列产品/Atlas A3 推理系列产品[object Object]:
- 支持2、4、8、16、32卡,并且仅支持hccs链路double ring组网。
- aclnnMatmulReduceScatter默认非确定性实现,支持通过aclrtCtxSetSysParamOpt开启确定性。
- Atlas 350 加速卡:支持2、4、8、16、32、64卡,并且仅支持hccs链路all mesh组网。
- ReduceScatter集合通信数据总量不能超过16*256MB,集合通信数据总量计算方式为:m * n * sizeof(output_dtype)。由于shape不同,算子内部实现可能存在差异,实际支持的总通信量可能略小于该值。
- aclnnMatmulReduceScatter默认为确定性实现。
说明:本示例代码调用了部分HCCL集合通信库接口:HcclGetCommName、HcclCommInitAll、HcclCommDestroy, 请参考。
[object Object]Atlas A2 训练系列产品/Atlas A2 推理系列产品[object Object]、[object Object]Atlas A3 训练系列产品/Atlas A3 推理系列产品[object Object]、Atlas 350 加速卡:
[object Object]