CANN/asc-devkit SetOpType接口
2026/7/15 14:51:19 网站建设 项目流程

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 = "ReduceScatter=level0:doublering"; AscendC::Mc2CcTilingConfig mc2CcTilingConfig(groupName, opType, algConfig, HCCL_REDUCE_RESERVED); mc2CcTilingConfig.SetReduceType(HCCL_REDUCE_SUM); mc2CcTilingConfig.GetTiling(tiling->mc2InitTiling); mc2CcTilingConfig.GetTiling(tiling->reduceScatterTiling); algConfig = "AllGather=level0:doublering"; mc2CcTilingConfig.SetGroupName(groupName); mc2CcTilingConfig.SetOpType(HCCL_CMD_ALLGATHER); // 设置通信任务类型 mc2CcTilingConfig.SetAlgConfig(algConfig); mc2CcTilingConfig.SetSkipLocalRankCopy(0); mc2CcTilingConfig.SetSkipBufferWindowCopy(1); mc2CcTilingConfig.GetTiling(tiling->allGatherTiling);

创作声明:本文部分内容由AI辅助生成(AIGC),仅供参考

需要专业的网站建设服务?

联系我们获取免费的网站建设咨询和方案报价,让我们帮助您实现业务目标

立即咨询