产品 |
是否支持 |
|---|---|
Ascend 950PR/Ascend 950DT |
√ |
√ |
|
√ |
|
x |
|
x |
|
x |
|
x |
分离模式下，获取一个AI Core上Cube Core（AIC）或者Vector Core（AIV）的数量。
1 | __aicore__ inline int64_t GetSubBlockNum() |
无
不同Kernel类型下（通过设置Kernel类型设置），在AIC和AIV上调用该接口的返回值如下：
Kernel类型 |
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 |
- |
无
1 | int64_t subBlockNum = AscendC::GetSubBlockNum(); |