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上的算子计算逻辑相关代码 ...