下载
EN
注册
我要评分
文档获取效率
文档正确性
内容完整性
文档易理解
在线提单
论坛求助
昇腾小AI

HcclGroup

函数功能

配置通信域名称,每个名称对应一个通信域。配置后在kernel侧调用GetHcclContext接口可获取通信域对应的context(消息区)地址。

函数原型

OpMC2Def &HcclGroup(const char *value)

OpMC2Def &HcclGroup(std::vector<const char *> value)

参数说明

参数

输入/输出

说明

value

输入

配置的通信域名称。单个通信域使用const char *,多通信域使用std::vector<const char *>。

约束说明

使用该接口前,算子需要先通过MC2接口注册该算子是通算融合算子,注册后即返回一个OpMC2Def结构。

通信域名称必须先配置为REQUIRED String类型的属性,属性名即为通信域名称。

调用示例

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("ascendxxx");
    this->MC2().HcclGroup("group"); // 配置通信域名称为group
};
OP_ADD(MC2Custom);
搜索结果
找到“0”个结果

当前产品无相关内容

未找到相关内容,请尝试其他搜索词