设置通信算法。
1
|
uint32_t SetAlgConfig(const std::string &algConfig) |
参数名 |
输入/输出 |
描述 |
---|---|---|
algConfig |
输入 |
通信算法配置。string类型,支持的最大长度为128字节。 针对 |
无
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); |