CANN/asc-devkit SetOpType接口 SetOpType功能说明设置通信任务类型。函数原型uint32_t SetOpType(uint32_t opType)参数说明表 1参数说明参数名输入/输出描述opType输入表示通信任务类型。uint32_t类型。HCCL API提供HcclCMDType枚举定义作为该参数的取值。针对Ascend 950PR/Ascend 950DT当前支持的通信任务类型为HCCL_CMD_ALLREDUCE、HCCL_CMD_ALLGATHER、HCCL_CMD_REDUCE_SCATTER、HCCL_CMD_ALLTOALL、HCCL_CMD_ALLTOALLV、HCCL_CMD_HALF_ALLTOALLV。针对Atlas A3 训练系列产品 / Atlas A3 推理系列产品当前支持的通信任务类型为HCCL_CMD_ALLREDUCE、HCCL_CMD_ALLGATHER、HCCL_CMD_REDUCE_SCATTER、HCCL_CMD_ALLTOALL、HCCL_CMD_ALLTOALLV、HCCL_CMD_BATCH_WRITE。针对Atlas A2 训练系列产品 / Atlas A2 推理系列产品当前支持的通信任务类型为HCCL_CMD_ALLREDUCE、HCCL_CMD_ALLGATHER、HCCL_CMD_REDUCE_SCATTER、HCCL_CMD_ALLTOALL、HCCL_CMD_BATCH_WRITE。返回值说明0表示设置成功。非0表示设置失败。约束说明无调用示例const char *groupName testGroup; uint32_t opType HCCL_CMD_REDUCE_SCATTER; std::string algConfig ReduceScatterlevel0:doublering; AscendC::Mc2CcTilingConfig mc2CcTilingConfig(groupName, opType, algConfig, HCCL_REDUCE_RESERVED); mc2CcTilingConfig.SetReduceType(HCCL_REDUCE_SUM); mc2CcTilingConfig.GetTiling(tiling-mc2InitTiling); mc2CcTilingConfig.GetTiling(tiling-reduceScatterTiling); algConfig AllGatherlevel0:doublering; mc2CcTilingConfig.SetGroupName(groupName); mc2CcTilingConfig.SetOpType(HCCL_CMD_ALLGATHER); // 设置通信任务类型 mc2CcTilingConfig.SetAlgConfig(algConfig); mc2CcTilingConfig.SetSkipLocalRankCopy(0); mc2CcTilingConfig.SetSkipBufferWindowCopy(1); mc2CcTilingConfig.GetTiling(tiling-allGatherTiling);创作声明:本文部分内容由AI辅助生成(AIGC),仅供参考