CANN/asc-devkit HCCL通信步长设置

SetStepSize【免费下载链接】asc-devkit本项目是CANN 推出的昇腾AI处理器专用的算子程序开发语言原生支持C和C标准规范主要由类库和语言扩展层构成提供多层级API满足多维场景算子开发诉求。项目地址: https://gitcode.com/cann/asc-devkit功能说明设置细粒度通信时通信算法的步长即设置细粒度通信时一次子通信任务执行或准备执行的通信算法的步骤数。例如图1使用pairwise算法的AlltoAllV通信步骤示意图中该细粒度通信场景下AlltoAllV通信任务的通信步长为1。函数原型uint32_t SetStepSize(uint8_t stepSize)参数说明表1参数说明参数名输入/输出描述stepSize输入设置细粒度通信时每次通信的步长。0表示当前非细粒度通信。返回值说明0表示设置成功。非0表示设置失败。约束说明目前仅支持Atlas A3 训练系列产品/Atlas A3 推理系列产品。调用示例static ge::graphStatus AllToAllVCustomTilingFunc(gert::TilingContext* context) { AllToAllVCustomV3TilingData* tiling context-GetTilingDataAllToAllVCustomV3TilingData(); const std::string groupName testGroup; const std::string algConfig AlltoAlllevel0:fullmesh;level1:pairwise; AscendC::Mc2CcTilingConfig mc2CcTilingConfig(groupName, HCCL_CMD_ALLTOALLV, algConfig, 0); mc2CcTilingConfig.SetStepSize(1U); mc2CcTilingConfig.GetTiling(tiling-mc2InitTiling); mc2CcTilingConfig.GetTiling(tiling-mc2CcTiling); return ge::GRAPH_SUCCESS; }【免费下载链接】asc-devkit本项目是CANN 推出的昇腾AI处理器专用的算子程序开发语言原生支持C和C标准规范主要由类库和语言扩展层构成提供多层级API满足多维场景算子开发诉求。项目地址: https://gitcode.com/cann/asc-devkit创作声明：本文部分内容由AI辅助生成（AIGC），仅供参考