REG_FUSION_PASS

Applicability

Product	Supported
Atlas 350 Accelerator Card	√
Atlas A3 training product/Atlas A3 inference product	√
Atlas A2 training product/Atlas A2 inference product	√
Atlas 200I/500 A2 inference product	√
Atlas inference product	√
Atlas training product	√

Header File

#include <ge/fusion/pass/pattern_fusion_pass.h>

Function Usage

Developers can register the graph modification function with the framework. The framework calls the custom graph modification pass at the beginning of build and calls REG_FUSION_PASS to register the custom pass.

Prototype

REG_FUSION_PASS(pass_class) 

Parameters

Parameter	Input/Output	Description
pass_class	Input	Name of a custom pass.

Returns

None

Restrictions

None

Parent topic: Basic Data Structures and APIs of Graphs