HcclGroup

函数功能

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

函数原型

OpMC2Def &HcclGroup(const char *value)

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

参数说明

约束说明

使用该接口前，算子需要先通过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);