GetSubBlockNum(ISASI)
产 品 支 持 情 况
| 产 品 | 是 否 支 持 |
|---|---|
Ascend 950PR/Ascend 950DT | √ |
Atlas A3 训 练 系 列 产 品/Atlas A3 推 理 系 列 产 品 | √ |
Atlas A2 训 练 系 列 产 品/Atlas A2 推 理 系 列 产 品 | √ |
Atlas 200I/500 A2 推 理 产 品 | x |
Atlas 推 理 系 列 产 品 AI Core | x |
Atlas 推 理 系 列 产 品 Vector Core | x |
Atlas 训 练 系 列 产 品 | x |
Kirin X90 | √ |
Kirin 9030 | √ |
功 能 说 明
头 文 件 路 径 为:"basic_api/kernel_operator_sys_var_intf.h"。
在分 离 模 式 架 构下,获 取 当 前 配 置 中,一 个 逻 辑AI Core上Cube Core(AIC)或 者Vector Core(AIV)的 数 量。
函 数 原 型
C++
__aicore__ inline int64_t GetSubBlockNum()
参 数 说 明
无
返 回 值 说 明
不 同Kernel类 型 下(通 过设 置Kernel类 型设 置),在AIC和AIV上 调 用 该 接 口 的 返 回 值 如 下:
表 1 返 回 值 列 表
| 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 | - |
约 束 说 明
无
调 用 示 例
C++
int64_t subBlockNum = AscendC::GetSubBlockNum();