SetCheckSupport

函数功能

如果您需要在算子融合阶段进行算子参数校验，则可实现算子参数校验回调函数，并通过该接口进行注册。同时，需要将NeedCheckSupportFlag参数配置为true，则算子编译和融合阶段会调用注册的算子参数校验函数进行相关信息的校验。

若算子参数校验函数校验通过，则代表AI Core支持此算子参数，会选择AI Core上相应的算子执行；否则，会继续查询AI CPU算子库然后执行。

函数原型

OpAICoreDef &SetCheckSupport(optiling::OP_CHECK_FUNC func);

参数说明

using OP_CHECK_FUNC = ge::graphStatus (*)(const ge::Operator &op, ge::AscendString &result);

{"ret_code": "1","reason": "your reason"}

返回值说明

OpAICoreDef请参考OpAICoreDef类。

约束说明

无

调用示例

下文展示了自定义Add算子参数校验函数实现和注册的样例。

• 参数校验函数实现如下：对第一个输入参数的shape进行校验，仅支持输入x shape的第一个维度为8，否则不支持。

  static ge::graphStatus CheckSupported(const ge::Operator &op, ge::AscendString &result)
  {
      std::string resultJsonStr;
      // 仅支持第一个输入参数shape的第一个维度为8，其他shape不支持
      if (op.GetInputDesc(0).GetShape().GetDim(0) == 8) {
          resultJsonStr = R"({"ret_code": "1","reason": "x.dim[0] is 8"})";
          result = ge::AscendString(resultJsonStr.c_str());
          return ge::GRAPH_SUCCESS;
      }
      resultJsonStr = R"({"ret_code": "0","reason": "xxx"})";
      result = ge::AscendString(resultJsonStr.c_str());
      return ge::GRAPH_FAILED;
  }

• 参数校验函数注册的样例如下：

  class AddCustom : public OpDef {
  public:
      AddCustom(const char* name) : OpDef(name)
      {
          this->Input("x")
              .ParamType(REQUIRED);
          this->Input("y")
              .ParamType(REQUIRED);
          this->Output("z")
              .ParamType(REQUIRED);
          this->SetInferShape(ge::InferShape);
  
          this->AICore()
              .SetTiling(optiling::TilingFunc)
              .SetTilingParse(optiling::TilingPrepare)
              .SetOpSelectFormat(optiling::OpSelectFormat)
              .SetCheckSupport(optiling::CheckSupported);
  
          OpAICoreConfig aicConfig;
          aicConfig.DynamicCompileStaticFlag(true)
              .DynamicFormatFlag(true)
              .DynamicRankSupportFlag(true)
              .DynamicShapeSupportFlag(true)
              .NeedCheckSupportFlag(true)
              .PrecisionReduceFlag(true);
          // 注意：soc_version请替换成实际的AI处理器型号
          this->AICore().AddConfig("soc_version", aicConfig);
      }
  }；