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。
父主题: ge命名空间