DropOutDoMaskFusionPass

Description

Replaces the DropOutDoMaskV3D operator with DropOutDoMask.

After fusion, the DropOutDoMaskV3D operator is replaced with the DropOutDoMask operator. The DropOutDoMask operator can be implemented through the DSL branch using the Elemwise template and supports UB fusion.

Before: After:

Restrictions

• When the DSA module is supported, the fusion patterns take effect and must be enabled.
• Fusion patterns must be enabled when the DropOutDoMask operator is used for UB fusion.
• One of the parent nodes of DropOutDoMaskV3D must be DSAGenBitMask, which is used as the mask input.

Availability

Atlas A2 training product/Atlas A2 inference product

Atlas A3 training product/Atlas A3 inference product