REGISTER_CUSTOM_PASS

You can register the graph modification APIs with the framework. Call REGISTER_CUSTOM_PASS to register the custom pass, which is followed by CustomPassFn and other APIs, with a period (.) between two APIs. For example:

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

You can view the API definition in the ${INSTALL_DIR}/include/register/register_custom_pass.h directory. Replace ${INSTALL_DIR} with the actual CANN component directory. If the Ascend-CANN-Toolkit package is installed as the root user, the CANN component directory is /usr/local/Ascend/ascend-toolkit/latest.