[object Object]

[object Object][object Object]undefined

说明： 使用该接口时，请确保驱动固件包和CANN包都为配套的8.0.RC2版本或者配套的更高版本，否则将会引发报错，比如Bus Error等。

[object Object]

• 接口功能：完成mm + reduce_scatter_base计算。

• 计算公式：

  $$output=reduceScatter(x1@x2+bias)$$

[object Object]

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

[object Object]

• 参数说明

  [object Object]
  • [object Object]Atlas A2 训练系列产品/Atlas A2 推理系列产品[object Object]、[object Object]Atlas A3 训练系列产品/Atlas A3 推理系列产品[object Object]：
    • bias：暂不支持输入为非0的场景。
  • Atlas 350 加速卡：
    • bias：支持输入为非0的场景。

• 返回值

  返回aclnnStatus状态码，具体参见。

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

  [object Object]

[object Object]

• 参数说明

  [object Object]

• 返回值：

  返回aclnnStatus状态码，具体参见。

[object Object]

• 确定性计算：

  • aclnnMatmulReduceScatter默认非确定性实现，支持通过aclrtCtxSetSysParamOpt开启确定性。

• 输入x1为2维，其shape为(m, k)，m须为卡数rank_size的整数倍。

• 输入x2必须是2维，其shape为(k, n)，轴满足mm算子入参要求，k轴相等，且k轴取值范围为[256, 65535)。

• x1/x2支持的空tensor场景，m和n可以为空，k不可为空，且需满足以下条件：

  • m为空，k不为空，n不为空；
  • m不为空，k不为空，n为空；
  • m为空，k不为空，n为空。

• x2矩阵支持转置/不转置场景，x1矩阵只支持不转置场景。

• x1、x2计算输入的数据类型要和output计算输出的数据类型一致。

• bias暂不支持输入为非0的场景。

• 输出为2维，其shape为(m/rank_size, n), rank_size为卡数。

• [object Object]Atlas A2 训练系列产品/Atlas A2 推理系列产品[object Object]：支持2、4、8卡，并且仅支持hccs链路all mesh组网。

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

  • 支持2、4、8、16、32卡，并且仅支持hccs链路double ring组网。
  • reduceScatter(x1@x2+bias)集合通信数据总量不能超过16*256MB，集合通信数据总量计算方式为：m * n * sizeof(output_dtype)。由于shape不同，算子内部实现可能存在差异，实际支持的总通信量可能略小于该值。

• Atlas 350 加速卡：支持2、4、8、16、32、64卡，并且仅支持hccs链路all mesh组网。

• [object Object]Atlas A2 训练系列产品/Atlas A2 推理系列产品[object Object]：一个模型中的通算融合MC2算子，仅支持相同通信域。

[object Object]

示例代码如下，仅供参考，具体编译和执行过程请参考。

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

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

  [object Object]