GetSubBlockNum(ISASI)

Applicability

Product

Supported

Atlas 350 Accelerator Card

Atlas A3 training product / Atlas A3 inference product

Atlas A2 training product / Atlas A2 inference product

Atlas 200I/500 A2 inference product

x

Atlas inference product AI Core

x

Atlas inference product Vector Core

x

Atlas training product

x

Function Usage

Obtains the number of AI Cube cores or AI Vector cores on an AI Core in separated mode.

Prototype

1
__aicore__ inline int64_t GetSubBlockNum()

Parameters

None

Returns

For different kernel types (Setting Kernel Type), the return values of calling this API on the AIC and AIV are as follows:

Table 1 Return values

Kernel Type

KERNEL_TYPE_AIV_ONLY

KERNEL_TYPE_AIC_ONLY

KERNEL_TYPE_MIX_AIC_1_2

KERNEL_TYPE_MIX_AIC_1_1

KERNEL_TYPE_MIX_AIC_1_0

KERNEL_TYPE_MIX_AIV_1_0

AIV

1

-

2

1

-

1

AIC

-

1

1

1

1

-

Constraints

None

Example

1
int64_t subBlockNum = AscendC::GetSubBlockNum();