SetReduceType
功能说明
设置Reduce操作类型,仅对有规约操作的通信任务生效。
函数原型
1 | uint32_t SetReduceType(uint32_t reduceType) |
参数说明
参数名 |
输入/输出 |
描述 |
---|---|---|
reduceType |
输入 |
规约操作类型,仅对有规约操作的通信任务生效。uint32_t类型,取值详见表2参数说明。 |
返回值
- 0表示设置成功。
- 非0表示设置失败。
注意事项
无
调用示例
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 | uint32_t index = 0U; const char *groupName = context->GetAttrs()->GetAttrPointer<char>(index++); uint32_t opType = HCCL_CMD_REDUCE_SCATTER; std::string algConfig = "ReduceScatter=level0:fullmesh"; uint32_t reduceType = HCCL_REDUCE_SUM; AscendC::Mc2CcTilingConfig mc2CcTilingConfig(groupName, opType, algConfig, reduceType); mc2CcTilingConfig.GetTiling(tiling->mc2InitTiling); mc2CcTilingConfig.GetTiling(tiling->reduceScatterTiling); algConfig = "AllGather=level0:doublering"; mc2CcTilingConfig.SetGroupName(groupName); mc2CcTilingConfig.SetOpType(HCCL_CMD_ALLGATHER); mc2CcTilingConfig.SetAlgConfig(algConfig); mc2CcTilingConfig.SetReduceType(HCCL_REDUCE_RESERVED); // 设置操作类型 mc2CcTilingConfig.SetSkipLocalRankCopy(0); mc2CcTilingConfig.SetSkipBufferWindowCopy(1); mc2CcTilingConfig.GetTiling(tiling->allGatherTiling); |
父主题: Hccl Tiling