get_unified_buffer_size

功能说明

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

此接口新开发算子不建议使用，建议使用get_soc_spec接口获取硬件相关信息。

函数原型

get_unified_buffer_size()

参数说明

无

支持的芯片型号

Atlas 200/300/500 推理产品

Atlas 训练系列产品

注意事项

使用此接口前，需要先设置昇腾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()