aclmdlSetExternalWeightAddress
产品支持情况
功能说明
配置存放外置权重的Device内存,调用方需在模型加载前将外置权重数据复制到该内存中。若存在多个外置权重文件,则需多次调用此接口。
模型加载过程中,计算图中的每一个FileConstant节点会保存对应外置权重文件的路径(例如/xx/../weight_hasid1),图引擎( Graph Engine ,简称GE)根据路径中的文件名(例如weight_hasid1)检索用户是否配置了外置权重的Device内存。若已配置,则直接使用该Device内存地址;否则,GE将自行申请一块Device内存,并读取外置权重文件,并将外置权重数据拷贝至该Device内存中,模型卸载时会释放该内存。

本接口需要与以下其它接口配合,实现模型加载功能:
- 调用aclmdlCreateConfigHandle接口创建模型加载的配置对象。
- (可选)调用aclmdlSetExternalWeightAddress接口配置存放外置权重的Device内存。
- 多次调用aclmdlSetConfigOpt接口设置配置对象中每个属性的值。
- 调用aclmdlLoadWithConfig接口指定模型加载时需要的配置信息,并进行模型加载。
- 模型加载成功后,调用aclmdlDestroyConfigHandle接口销毁。
函数原型
aclError aclmdlSetExternalWeightAddress(aclmdlConfigHandle *handle, const char *weightFileName, void *devPtr, size_t size)
参数说明
模型加载的配置对象的指针。需提前调用aclmdlCreateConfigHandle接口创建该对象。 |
||
|
一般对om模型文件大小有限制或模型文件加密的场景下,需单独加载权重文件,因此需在构建模型时,将权重保存在单独的文件中。例如在使用ATC工具生成om文件时,将--external_weight参数设置为1(1表示将原始网络中的Const/Constant节点的权重保存在单独的文件中)。 |
||
返回值说明
返回0表示成功,返回其他值表示失败,请参见aclError。