Dprofile类构造函数
功能说明
Dprofile类用来管理和配置昇腾AI处理器相关信息, 例如产品架构、产品形态、各级buffer大小等。
由于不同版本的昇腾AI处理器的内存大小和指令不一样,需要通过该函数定义昇腾AI处理器目标机,指定编程时所针对的硬件环境。
此接口新开发算子不建议使用,建议使用set_current_compile_soc_info接口设置昇腾AI处理器的类型。
函数原型
__init__(ai_core_arch=None, ai_core_version=None, ddk_version=None)
参数说明
参数名称 |
输入/输出 |
含义 |
---|---|---|
ai_core_arch |
输入 |
保留参数,对于新开发的算子,不建议配置。 |
ai_core_version |
输入 |
保留参数,对于新开发的算子,不建议配置。 |
ddk_version |
输入 |
保留参数,对于新开发的算子,不建议配置。 |
支持的芯片型号
Atlas 200/300/500 推理产品
Atlas 训练系列产品
Atlas推理系列产品AI Core
Atlas推理系列产品Vector Core
Atlas A2训练系列产品/Atlas 800I A2推理产品
注意事项
若未在Dprofile中指定产品架构和产品形态,则会用默认的产品形态“Ascend310”。
返回值
输出Dprofile类的实例。
调用示例
from tbe import tik tik_dprofile = tik.Dprofile()
父主题: 不建议使用接口