Dprofile类构造函数

功能说明

Dprofile类用来管理和配置昇腾AI处理器相关信息, 例如产品架构、产品形态、各级buffer大小等。

由于不同版本的昇腾AI处理器的内存大小和指令不一样，需要通过该函数定义昇腾AI处理器目标机，指定编程时所针对的硬件环境。

函数原型

__init__(ai_core_arch=None, ai_core_version=None, ddk_version=None)

参数说明

支持的芯片型号

Atlas 200/300/500 推理产品

Atlas 训练系列产品

Atlas推理系列产品（Ascend 310P处理器）AI Core

Atlas推理系列产品（Ascend 310P处理器）Vector Core

Atlas A2训练系列产品/Atlas 800I A2推理产品

注意事项

若未在Dprofile中指定产品架构和产品形态，则会用默认的产品形态“Ascend310”。

返回值

输出Dprofile类的实例。

调用示例

from tbe import tik
tik_dprofile = tik.Dprofile()