Core

函数功能

AI Core抽象,在with语句中实例化并使用来明确针对某一AI Core类型进行建模。

函数原型

class Core(core_type_name)

参数说明

参数名

输入类型

说明

core_type_name

string

昇腾计算单元类型字符串,通常可以表示为“AICx”“AIVx”,其中x为数字,即使用的AI Core/ AI Vector Core的序号。

约束说明

需在with语句下将该类初始化。

使用示例

from mskpp import Core
with Core("AIC0") as aic:
    # AI Core 0上的算子计算逻辑相关代码
    ...