昇腾社区首页
中文
注册

函数:load_from_mem

产品支持情况

产品

是否支持

Atlas A3 训练系列产品 / Atlas A3 推理系列产品

Atlas A2 训练系列产品 / Atlas 800I A2 推理产品 /A200I A2 Box 异构组件

Atlas 训练系列产品

Atlas 推理系列产品

Atlas 200I/500 A2 推理产品

Atlas 200/300/500 推理产品

功能说明

从内存加载离线模型数据,由系统内部管理模型运行的内存。

系统完成模型加载后,返回的模型ID,作为后续操作时用于识别模型的标志。

函数原型

  • C函数原型
    1
    aclError aclmdlLoadFromMem(const void* model, size_t modelSize, uint32_t* modelId)
    
  • python函数
    1
    model_id, ret = acl.mdl.load_from_mem(model, model_size)
    

参数说明

参数名

说明

model

int,模型数据的内存地址对应的指针地址。

  • 应用运行在Host时,此处需申请Host上的内存。
  • 应用运行在Device时,此处需申请Device上的内存。

内存申请接口请参见内存管理

model_size

int,内存中的模型数据长度,单位Byte。

返回值说明

返回值

说明

model_id

int,系统完成模型加载后生成的模型ID对应的指针地址。

ret

int,错误码,返回0表示成功,返回其它值表示失败。

约束说明

模型加载、模型执行、模型卸载的操作必须在同一个Context下(关于Context的创建请参见acl.rt.set_deviceacl.rt.create_context)。

资源参考

接口调用流程,参见模型加载