昇腾社区首页
中文
注册
开发者
下载

REGISTER_CUSTOM_PASS

产品支持情况

产品

是否支持

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

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

Atlas 200I/500 A2 推理产品

Atlas 推理系列产品

Atlas 训练系列产品

头文件/库文件

  • 头文件:#include <register/register_custom_pass.h>
  • 库文件:libregister.so

功能说明

开发人员可以选择将改图函数注册到框架中,由框架在编译最开始调用自定义改图Pass,调用REGISTER_CUSTOM_PASS进行自定义Pass注册。

函数原型

1
REGISTER_CUSTOM_PASS(name) 

参数说明

参数名

输入/输出

描述

name

输入

自定义Pass的名称。

返回值说明

约束说明

调用时以REGISTER_CUSTOM_PASS开始,以“.”连接CustomPassFn等接口。例如:

1
2
#include "register/register_custom_pass.h"
REGISTER_CUSTOM_PASS("pass_name").CustomPassFn(CustomPassFunc);