下载
中文
注册

Dprofile类构造函数

功能说明

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

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

此接口新开发算子不建议使用,建议使用set_current_compile_soc_info接口设置昇腾AI处理器的类型。

函数原型

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

参数说明

表1 参数说明

参数名称

输入/输出

含义

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()