2.4.1. Neutron v3 Architecture#
Neutron v3 consists of 1 NPU Core containing 6 CONV, 3 TENG, and 3 SDMA EUs. These physical EUs are organized into 3 basic resource groups, each containing 2 CONV, 1 TENG, and 1 SDMA.
At runtime, the virtual NPU mode can be set through AX_ENGINE_NPU_ATTR_T.eHardMode in AX_ENGINE_Init to determine how the three basic resource groups are partitioned into vNPUs. When converting a model with the Pulsar2 compiler, specify NPU1, NPU2, or NPU3 according to the required resource scale.
2.4.1.1. Runtime modes and npu_mode#
2.4.1.1.1. AXEngine runtime modes#
AXEngine runtime mode |
Compute resources |
|---|---|
|
A 1 + 1 + 1 partition that divides the three basic resource groups into three independent vNPUs. Each vNPU contains 2 |
|
A 2 + 1 or 1 + 2 partition. The large vNPU uses two basic resource groups, and the small vNPU uses one. |
|
A partition of 3. Multiple vNPUs are disabled, and all three basic resource groups are used as one complete NPU resource. |
2.4.1.1.2. Offline npu_mode and hardware resources#
Chip platform |
|
Hardware compute resources |
|---|---|---|
|
|
One Neutron v3 basic resource group containing 2 |
|
|
Two Neutron v3 basic resource groups containing 4 |
|
|
Three Neutron v3 basic resource groups containing 6 |
|
|
One Neutron v3 basic resource group containing 2 |
|
|
Two Neutron v3 basic resource groups containing 4 |
|
|
Three Neutron v3 basic resource groups containing 6 |