Stage

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/Library File

Header file: #include <ge/fusion/pass/fusion_pass_reg.h>
Library file: libge_compiler.so

Function Usage

Defines the fusion pass phase.

Note: General fusion passes cannot be registered in the kAfterAssignLogicStream phase. Modifying the graph structure is not allowed in this phase. If a pass is unexpectedly registered in this phase, it will be ignored.

Prototype

FusionPassRegistrationData &Stage(CustomPassStage stage)

Parameters

Parameter	Input/Output	Description
stage	Input	Execution phase of a custom fusion pass. For details, see CustomPassStage.

Returns

FusionPassRegistrationData object.

Restrictions

None

Parent topic: FusionPassRegistrationData