根据模型文件获取模型执行时所需的权值内存大小、工作内存大小,同步接口。
当由用户管理内存时,为确保内存不浪费,在申请工作内存、权值内存前,需要调用aclmdlQuerySize接口查询模型运行时所需工作内存、权值内存的大小。
如果模型输入数据的Shape不确定,则不能调用aclmdlQuerySize接口查询内存大小,在加载模型时,就无法由用户管理内存,因此需选择由系统管理内存的模型加载接口(例如,aclmdlLoadFromFile、aclmdlLoadFromMem)。
aclError aclmdlQuerySize(const char *fileName, size_t *workSize, size_t *weightSize)
参数名 |
输入/输出 |
说明 |
---|---|---|
fileName |
输入 |
模型文件路径的指针,路径中包含文件名。运行程序(APP)的用户需要对该路径有访问权限。 |
workSize |
输出 |
模型执行时所需的工作内存大小的指针,单位Byte。 |
weightSize |
输出 |
模型执行时所需权值内存大小的指针,单位Byte。 |
返回0表示成功,返回其它值表示失败。
接口调用流程及示例代码,参见模型加载。