下载
中文
注册
我要评分
文档获取效率
文档正确性
内容完整性
文档易理解
在线提单
论坛求助
昇腾小AI

GetCoreNumAiv

函数功能

获取当前硬件平台AI Core中Vector核数。若AI Core的架构为Cube、Vector分离架构,返回AI Core上的Vector核数;非分离架构返回AI Core的核数。

函数原型

uint32_t GetCoreNumAiv(void) const;

参数说明

参数

输入/输出

说明

-

-

-

返回值说明

针对Atlas 训练系列产品,非Cube、Vector分离架构,返回AI Core的核数

针对Atlas 推理系列产品(Ascend 310P处理器),非Cube、Vector分离架构,返回AI Core的核数

Atlas A2训练系列产品/Atlas 800I A2推理产品,Cube、Vector分离架构,返回AI Core上的Vector核数

约束说明

调用示例

ge::graphStatus TilingXXX(gert::TilingContext* context) {
    auto ascendcPlatform = platform_ascendc::PlatformAscendC(context->GetPlatformInfo());
    auto aicNum = ascendcPlatform.GetCoreNumAic();
    auto aivNum = ascendcPlatform.GetCoreNumAiv();
    // ...按照aivNum切分
    context->SetBlockDim(ascendcPlatform.CalcTschBlockDim(aivNum, aicNum, aivNum));
    return ret;
}
搜索结果
找到“0”个结果

当前产品无相关内容

未找到相关内容,请尝试其他搜索词