CANN/asc-devkit HCCL服务端类型配置

HcclServerType

【免费下载链接】asc-devkit本项目是CANN 推出的昇腾AI处理器专用的算子程序开发语言，原生支持C和C++标准规范，主要由类库和语言扩展层构成，提供多层级API，满足多维场景算子开发诉求。项目地址: https://gitcode.com/cann/asc-devkit

功能说明

配置HCCL的服务端类型。

函数原型

void HcclServerType(enum HcclServerType type, const char *soc=nullptr)

参数说明

namespace ops{ enum HcclServerType : uint32_t { AICPU = 0, // AI CPU服务端 AICORE, // AI Core服务端 CCU, // CCU服务端，仅在AI处理器包含CCU单元时支持 MAX // 预留参数，不支持使用 }; }

约束说明

• 使用该接口前，算子需要先通过MC2接口注册该算子是通算融合算子，注册后即返回一个OpMC2Def结构。
• 同时为特定AI处理器型号和所有AI处理器型号配置服务端类型时，特定AI处理器型号配置的优先级更高。

调用示例

class MC2Custom : public OpDef { public: MC2Custom(const char* name) : OpDef(name) { this->Input("x").ParamType(REQUIRED).DataType({ge::DT_FLOAT}).Format({ge::FORMAT_ND}); this->Input("y").ParamType(REQUIRED).DataType({ge::DT_FLOAT}).Format({ge::FORMAT_ND}); this->Output("z").ParamType(REQUIRED).DataType({ge::DT_FLOAT}).Format({ge::FORMAT_ND}); this->Attr("group").AttrType(REQUIRED).String(); this->AICore().AddConfig("ascendxxx1"); this->AICore().AddConfig("ascendxxx2"); this->MC2().HcclGroup("group"); // 配置通信域名称为group this->MC2().HcclServerType(HcclServerType::AICPU, "ascendxxx1"); // 配置ascendxxx1型号的通信模式为AI CPU this->MC2().HcclServerType(HcclServerType::AICORE); // 配置其他型号即ascendxxx2的通信模式为AI Core } }; OP_ADD(MC2Custom);

【免费下载链接】asc-devkit本项目是CANN 推出的昇腾AI处理器专用的算子程序开发语言，原生支持C和C++标准规范，主要由类库和语言扩展层构成，提供多层级API，满足多维场景算子开发诉求。项目地址: https://gitcode.com/cann/asc-devkit