下载
中文
注册

函数:query_size_from_mem

C函数原型

aclError aclmdlQuerySizeFromMem(const void *model, size_t modelSize, size_t *workSize, size_t *weightSize)

Python函数

work_size, weight_size, ret = acl.mdl.query_size_from_mem(model, model_size)

函数功能

根据内存中的模型数据获取模型执行时所需的权值内存大小、内存大小,同步接口。

输入说明

model:int,需要获取内存信息的模型数据对应的指针地址。

model_size:int,模型数据长度,单位Byte。

返回值说明

work_size:int,模型执行时所需的工作内存的大小,单位Byte。

weight_size:int,模型执行时所需权值内存的大小,单位Byte。

ret:int,错误码。

  • 返回0表示成功。
  • 返回其它值表示失败。

约束说明

执行和权重内存为Device内存,而且需要用户申请和释放。