昇腾社区首页
中文
注册

注册融合规则

注册融合规则包括注册融合规则名称、注册融合规则类和设置融合规则是通用规则还是定制化规则。

通过REGISTER_SCOPE_FUSION_PASS向GE注册Scope融合规则,例如:
REGISTER_SCOPE_FUSION_PASS("DecodeBboxV2ScopeFusionPass", DecodeBboxV2ScopeFusionPass, false);
  • 第一个参数为融合规则名称。
  • 第二个参数为融合规则类。
  • 第三个参数为is_general。
    • 值为true表示通用规则,默认生效;
    • 值为false表示定制化规则,默认不生效,需要用户自行在图编译时通过参数enable_scope_fusion_passes配置生效。