ForceFp16CastFusionPass

Description

For the cast operator, when converting input and output data types between int32 and fp32 (either int32->fp32 or fp32->int32), set attr_keep_dtype to improve the precision.

Restrictions

For the cast operator, the fusion pattern is enabled when data types for the input and output are converted between int32 and fp32 (either int32->fp32 or fp32->int32).

Availability

Atlas A2 training product/Atlas A2 inference product

Parent topic: Graph Fusion Patterns