AI Core抽象,在with语句中实例化并用来明确针对某一AI Core类型进行建模。
1 | class Core(core_type_name) |
参数名 |
输入类型 |
说明 |
|---|---|---|
core_type_name |
string |
昇腾计算单元类型字符串,通常可以表示为“AICx”或“AIVx”,其中x为数字,即使用的AI Cube Core/ AI Vector Core的序号。 仅支持A-Za-z0-9中的一个或多个字符。 |
需在with语句下将该类初始化。
1 2 3 4 | from mskpp import Core with Core("AIC0") as aic: # AI Cube Core 0上的算子计算逻辑相关代码 ... |