get_unified_buffer_size

功能说明

获取对应产品形态的Unified Buffer的buffer大小，单位为Bytes。

函数原型

get_unified_buffer_size()

参数说明

无

支持的型号

Atlas 200/300/500 推理产品

Atlas 训练系列产品

Atlas推理系列产品AI Core

Atlas推理系列产品Vector Core

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

注意事项

使用此接口前，需要先设置昇腾AI处理器的类型，若不设置，默认使用Ascend310。

返回值

返回对应产品形态的Unified Buffer的buffer大小，单位为Bytes。

调用示例

from tbe import tik
tik_dprofile = tik.Dprofile()       # 设置昇腾AI处理器的类型为Ascend310
unified_buffer_size = tik_dprofile.get_unified_buffer_size()