register_decompose_pass

Applicable Products

All processors

Function Usage

Registers the decorated DecomposePass subclass with the pass registry and sets op_types as a class attribute.

Prototype

1
register_decompose_pass(*, name: str, stage: PassStage, op_types: Iterable[str]) -> callable

Parameters

Parameter

Input/Output

Description

name

Input

Pass name, which is of the string type and must be unique. It cannot be the same as any registered pass name.

stage

Input

Pass execution stage, which is of the PassStage enum type.

op_types

Input

List of operator types to be decomposed. The value is an iterable object of the string type. It cannot be empty, and each element must be a non-empty string.

Returns

The class decorator function. The decorated class is registered in the pass registry, and op_types is set as a class attribute.

Restrictions

None