昇腾社区首页
中文
注册

RegisterScopeFusionPass

产品支持情况

产品

是否支持

Atlas A3 训练系列产品/Atlas A3 推理系列产品

Atlas A2 训练系列产品/Atlas 800I A2 推理产品/A200I A2 Box 异构组件

Atlas 200I/500 A2 推理产品

Atlas 推理系列产品

Atlas 训练系列产品

函数功能

注册融合算子。

函数原型

数据类型为string的接口后续版本会废弃,建议使用数据类型为非string的接口。

1
2
void RegisterScopeFusionPass(const std::string &pass_name, CreateFn create_fn, bool is_general);
void RegisterScopeFusionPass(const char_t *pass_name, CreateFn create_fn, bool is_general);

参数说明

参数名

输入/输出

描述

pass_name

输入

融合算子名称

create_fn

输入

返回融合算子对象的函数指针

is_general

输入

true:通用规则,默认生效

false:定制化规则,默认不生效

返回值

无。

约束说明

REGISTER_SCOPE_FUSION_PASS内部调用,用户不直接调用。