CANN asc-devkit SetOpType接口

📅 2026/7/17 9:38:07
CANN asc-devkit SetOpType接口
SetOpType【免费下载链接】asc-devkit本项目是CANN 推出的昇腾AI处理器专用的算子程序开发语言原生支持C和C标准规范主要由类库和语言扩展层构成提供多层级API满足多维场景算子开发诉求。项目地址: https://gitcode.com/cann/asc-devkit功能说明设置通信任务类型。函数原型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);【免费下载链接】asc-devkit本项目是CANN 推出的昇腾AI处理器专用的算子程序开发语言原生支持C和C标准规范主要由类库和语言扩展层构成提供多层级API满足多维场景算子开发诉求。项目地址: https://gitcode.com/cann/asc-devkit创作声明:本文部分内容由AI辅助生成(AIGC),仅供参考