下载
中文
注册

SetSkipBufferWindowCopy

功能说明

设置通信算法获取输入数据的位置。

函数原型

1
uint32_t SetSkipBufferWindowCopy(uint8_t skipBufferWindowCopy)

参数说明

表1 参数说明

参数名

输入/输出

描述

skipBufferWindowCopy

输入

表示通信算法获取输入数据的位置,uint8_t类型。

针对Atlas A2 训练系列产品/Atlas 800I A2 推理产品,参数取值如下:

  • 0:通信输入不放在windows中。未调用该接口设置通信输入的位置时,默认通信输入不放在windows中。其中windows为其他卡可访问的共享缓冲区。
  • 1:通信输入不放在windows中,当前该参数取值1与取值0的功能一致。
  • 2:通信输入放在windows中,仅适用于AllReduce算法、AlltoAll算法。

返回值

  • 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);