设置Reduce操作类型,仅对有归约操作的通信任务生效。
1 | uint32_t SetReduceType(uint32_t reduceType) |
参数名 |
输入/输出 |
描述 |
---|---|---|
reduceType |
输入 |
归约操作类型,仅对有归约操作的通信任务生效。uint32_t类型,取值详见表2参数说明。 |
无
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 | const char *groupName = "testGroup"; uint32_t opType = HCCL_CMD_REDUCE_SCATTER; std::string algConfig = "ReduceScatter=level0:fullmesh"; 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); |