昇腾社区首页
中文
注册

CustomPassFn

产品支持情况

产品

是否支持

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

Atlas 200I/500 A2 推理产品

Atlas 推理系列产品

Atlas 训练系列产品

Atlas 200/300/500 推理产品

功能说明

注册自定义Pass的执行函数。

函数原型

1
PassRegistrationData &CustomPassFn(const CustomPassFunc &custom_pass_fn)

参数说明

参数名

输入/输出

说明

custom_pass_fn

输入

自定义Pass的执行函数。

返回值说明

返回自身对象的引用。

约束说明

回调函数CustomPassFunc

用户自定义并实现CustomPassFunc类函数,即自定义的改图函数。

1
Status CustomPassFunc(GraphPtr &graph, CustomPassContext &custom_context)
表1 参数说明

参数名

输入/输出

说明

graph

输入

要修改的图。

custom_context

输入

维测类对象,通过此对象向框架注册维测信息。

-

输出

  • SUCCESS:成功。
  • 其他值:失败。