昇腾社区首页
中文
注册

REGISTER_CUSTOM_PASS

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

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

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

您可以在“${INSTALL_DIR}/include/register/register_custom_pass.h”下查看接口定义。${INSTALL_DIR}请替换为CANN软件安装后文件存储路径。若安装的Ascend-cann-toolkit软件包,以root安装举例,则安装后文件存储路径为:/usr/local/Ascend/ascend-toolkit/latest。