产品 |
是否支持 |
|---|---|
Ascend 950PR/Ascend 950DT |
√ |
√ |
|
√ |
|
x |
|
√ |
|
x |
|
x |
分离模式下,获取一个AI Core上Cube Core(AIC)或者Vector Core(AIV)的数量与AI Core数量的比例。耦合模式下,固定返回1。
1 | __aicore__ inline int64_t GetTaskRatio() |
无
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 2 | uint64_t ratio = AscendC::GetTaskRatio(); // 返回AIC和AIV的数量比例 AscendC::PRINTF("task ratio is %u", ratio); |