AOE_MODE

Description

Sets the AOE optimization mode in online inference and training scenarios. The options are as follows:

• 1: subgraph tuning.
• 2: operator tuning.
• 4: gradient splitting tuning.

Example

export AOE_MODE=2

Restrictions

None

Applicability

Atlas training product

For the Atlas inference product, the value of AOE_MODE can only be 1 or 2.

For the Atlas A2 training product/Atlas A2 inference product, the value of AOE_MODE can only be 2 or 4.

For the Atlas A3 training product/Atlas A3 inference product, the value of AOE_MODE can only be 2.