SetCommEngine
【免费下载链接】asc-devkit本项目是CANN 推出的昇腾AI处理器专用的算子程序开发语言,原生支持C和C++标准规范,主要由类库和语言扩展层构成,提供多层级API,满足多维场景算子开发诉求。项目地址: https://gitcode.com/cann/asc-devkit
功能说明
设置通信任务使用的通信引擎。
函数原型
uint32_t SetCommEngine(uint8_t commEngine)参数说明
表1参数说明
| 参数名 | 输入/输出 | 描述 |
|---|---|---|
| commEngine | 输入 | 通信引擎。uint8_t类型,该参数的取值范围请参考:HcclCommConfig中的hcclOpExpansionMode参数的取值说明。 |
返回值说明
- 0表示设置成功。
- 非0表示设置失败。
约束说明
无
调用示例
static ge::graphStatus AllToAllVCustomTilingFunc(gert::TilingContext* context) { AllToAllVCustomV3TilingData* tiling = context->GetTilingData<AllToAllVCustomV3TilingData>(); const std::string groupName = "testGroup"; const std::string algConfig = "AlltoAll=level0:fullmesh;level1:pairwise"; AscendC::Mc2CcTilingConfig mc2CcTilingConfig(groupName, HCCL_CMD_ALLTOALLV, algConfig, 0); mc2CcTilingConfig.SetCommEngine(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),仅供参考