harmony 鸿蒙INnrtDevice

  • 2023-06-24
  • 浏览 (732)

INnrtDevice

概述

定义了与设备相关的接口，实现设备管理和模型编译等操作。

当有多个设备注册时，需要保证设备名称和设备商名称的组合全局唯一。

Since:

3.2

Version:

2.0

相关模块:

NNRt

汇总

Public 成员函数

名称 描述
GetDeviceName ([out] String name) 获取设备名称 
GetVendorName ([out] String name) 获取设备商名称 
GetDeviceType ([out] enum DeviceType deviceType) 获取设备类型 
GetDeviceStatus ([out] enum DeviceStatus status) 获取设备当前状态 
GetSupportedOperation ([in] struct Model model, [out] boolean[] ops) 查询设备对指定模型的算子支持程度 
IsFloat16PrecisionSupported ([out] boolean isSupported) 查询设备是否支持以Float16精度运算Float32的模型。 
IsPerformanceModeSupported ([out] boolean isSupported) 查询设备是否支持性能偏好设置，性能偏好的定义可以参考PerformanceMode。 
IsPrioritySupported ([out] boolean isSupported) 查询设备是否支持任务优先级设置，优先级的定义可以参考Priority。 
IsDynamicInputSupported ([out] boolean isSupported) 查询设备是否支持变尺寸输入，变尺寸输入意味着同一个模型的不同次运算输入的形状可以不一样。 
PrepareModel ([in] struct Model model, [in] struct ModelConfig config, [out] IPreparedModel preparedModel) 编译模型 
IsModelCacheSupported ([out] boolean isSupported) 查询是否支持模型缓存功能 
PrepareModelFromModelCache ([in] struct SharedBuffer[] modelCache, [in] struct ModelConfig config, [out] IPreparedModel preparedModel) 加载模型缓存，该模型缓存是通过ExportModelCache接口导出的。 
PrepareOfflineModel ([in] struct SharedBuffer[] modelCache, [in] struct ModelConfig config, [out] IPreparedModel preparedModel) 加载离线模型文件的缓存，该离线模型是由推理框架传入NNRt并由NNRt解析得到的。 
AllocateBuffer ([in] unsigned int length, [out] struct SharedBuffer buffer) 申请设备共享内存，以文件描述符的形式返回，共享内存主要用于推理输入输出数据的快速传递。 
ReleaseBuffer ([in] struct SharedBuffer buffer) 释放共享内存。 

成员函数说明

AllocateBuffer()

INnrtDevice::AllocateBuffer ([in] unsigned int length, [out] struct SharedBuffer buffer )

描述: 申请设备共享内存，以文件描述符的形式返回，共享内存主要用于推理输入输出数据的快速传递。

参数:

名称 描述
length 申请共享内存的大小，单位是字节。 
buffer 共享内存的信息，包含共享内存的文件描述符和空间大小，SharedBuffer定义请查看SharedBuffer。

返回:

返回0表示成功

返回非0表示失败，负数为HDF标准错误码，正数为NNRt定义的专用错误码（请查看NNRT_ReturnCode）

GetDeviceName()

INnrtDevice::GetDeviceName ([out] String name)

描述: 获取设备名称

参数:

名称 描述
name 设备名称

返回:

返回0表示成功

返回非0表示失败，负数为HDF标准错误码，正数为NNRt定义的专用错误码（请查看NNRT_ReturnCode）

GetDeviceStatus()

INnrtDevice::GetDeviceStatus ([out] enum DeviceStatus status)

描述: 获取设备当前状态

参数:

名称 描述
status 设备当前状态，DeviceStatus枚举定义了可选的设备状态，详细定义请查看DeviceStatus。

返回:

返回0表示成功

返回非0表示失败，负数为HDF标准错误码，正数为NNRt定义的专用错误码（请查看NNRT_ReturnCode）

GetDeviceType()

INnrtDevice::GetDeviceType ([out] enum DeviceType deviceType)

描述: 获取设备类型

参数:

名称 描述
deviceType 设备类型，DeviceType枚举定义了可选的设备类型，详细定义请查看DeviceType。

返回:

返回0表示成功

返回非0表示失败，负数为HDF标准错误码，正数为NNRt定义的专用错误码（请查看NNRT_ReturnCode）

GetSupportedOperation()

INnrtDevice::GetSupportedOperation ([in] struct Model model, [out] boolean[] ops )

描述: 查询设备对指定模型的算子支持程度

参数:

名称 描述
model AI模型，模型结构由Model定义，详细定义请查看Model。 
ops 算子是否支持列表，算子支持列表的顺序与在model中的顺序要一致。

返回:

返回0表示成功

返回非0表示失败，负数为HDF标准错误码，正数为NNRt定义的专用错误码（请查看NNRT_ReturnCode）

GetVendorName()

INnrtDevice::GetVendorName ([out] String name)

描述: 获取设备商名称

参数:

名称 描述
name 设备商名称

返回:

返回0表示成功

返回非0表示失败，负数为HDF标准错误码，正数为NNRt定义的专用错误码（请查看NNRT_ReturnCode）

IsDynamicInputSupported()

INnrtDevice::IsDynamicInputSupported ([out] boolean isSupported)

描述: 查询设备是否支持变尺寸输入，变尺寸输入意味着同一个模型的不同次运算输入的形状可以不一样。

如果支持变尺寸输入，模型输入Tensor的形状上用-1标记该维度是否可变。

参数:

名称 描述
isSupported 是否支持变尺寸输入。

返回:

返回0表示成功

返回非0表示失败，负数为HDF标准错误码，正数为NNRt定义的专用错误码（请查看NNRT_ReturnCode）

IsFloat16PrecisionSupported()

INnrtDevice::IsFloat16PrecisionSupported ([out] boolean isSupported)

描述: 查询设备是否支持以Float16精度运算Float32的模型。

参数:

名称 描述
isSupported 是否支持Float16精度。

返回:

返回0表示成功

返回非0表示失败，负数为HDF标准错误码，正数为NNRt定义的专用错误码（请查看NNRT_ReturnCode）

IsModelCacheSupported()

INnrtDevice::IsModelCacheSupported ([out] boolean isSupported)

描述: 查询是否支持模型缓存功能

若支持，则需要实现PrepareModelFromModelCache和ExportModelCache两个接口。

参数:

名称 描述
isSupported 是否支持模型缓存。

返回:

返回0表示成功

返回非0表示失败，负数为HDF标准错误码，正数为NNRt定义的专用错误码（请查看NNRT_ReturnCode）

IsPerformanceModeSupported()

INnrtDevice::IsPerformanceModeSupported ([out] boolean isSupported)

描述: 查询设备是否支持性能偏好设置，性能偏好的定义可以参考PerformanceMode。

参数:

名称 描述
isSupported 是否支持性能偏好设置。

返回:

返回0表示成功

返回非0表示失败，负数为HDF标准错误码，正数为NNRt定义的专用错误码（请查看NNRT_ReturnCode）

IsPrioritySupported()

INnrtDevice::IsPrioritySupported ([out] boolean isSupported)

描述: 查询设备是否支持任务优先级设置，优先级的定义可以参考Priority。

参数:

名称 描述
isSupported 是否支持性能偏好设置。

返回:

返回0表示成功

返回非0表示失败，负数为HDF标准错误码，正数为NNRt定义的专用错误码（请查看NNRT_ReturnCode）

PrepareModel()

INnrtDevice::PrepareModel ([in] struct Model model, [in] struct ModelConfig config, [out] IPreparedModel preparedModel )

描述: 编译模型

如果是变尺寸输入模型，则模型输入的维度信息中至少有一个是-1。

参数:

名称 描述
model 需要编译的模型，Model定义请查看Model。 
config 编译模型的配置，ModelConfig定义请查看ModelConfig。 
preparedModel 编译好的模型对象，用于后续的运算，IPreparedModel定义请查看IPreparedModel。

返回:

返回0表示成功

返回非0表示失败，负数为HDF标准错误码，正数为NNRt定义的专用错误码（请查看NNRT_ReturnCode）

PrepareModelFromModelCache()

INnrtDevice::PrepareModelFromModelCache ([in] struct SharedBuffer[] modelCache, [in] struct ModelConfig config, [out] IPreparedModel preparedModel )

描述: 加载模型缓存，该模型缓存是通过ExportModelCache接口导出的。

参数:

名称 描述
modelCache 模型缓存文件的数组，数组顺序与导出时的数组顺序一致，数组元素类型请查看SharedBuffer定义SharedBuffer。 
config 加载模型缓存的配置，配置参数的详细定义请参考ModelConfig。 
preparedModel 加载缓存得到的模型对象，用于后续的运算，IPreparedModel定义请查看IPreparedModel。

返回:

返回0表示成功

返回非0表示失败，负数为HDF标准错误码，正数为NNRt定义的专用错误码（请查看NNRT_ReturnCode）

PrepareOfflineModel()

INnrtDevice::PrepareOfflineModel ([in] struct SharedBuffer[] modelCache, [in] struct ModelConfig config, [out] IPreparedModel preparedModel )

描述: 加载离线模型文件的缓存，该离线模型是由推理框架传入NNRt并由NNRt解析得到的。

参数:

名称 描述
modelCache 离线模型文件缓存的数组，元素顺序与用户传入的离线模型格式有关，元素类型请查看SharedBuffer定义SharedBuffer。 
config 加载离线模型文件缓存的配置，配置参数的详细定义请参考ModelConfig。 
preparedModel 加载离线模型文件缓存得到的模型对象，用于后续的运算，IPreparedModel定义请查看IPreparedModel。

返回:

返回0表示成功

返回非0表示失败，负数为HDF标准错误码，正数为NNRt定义的专用错误码（请查看NNRT_ReturnCode）

ReleaseBuffer()

INnrtDevice::ReleaseBuffer ([in] struct SharedBuffer buffer)

描述: 释放共享内存。

参数:

名称 描述
buffer 共享内存的信息，包含共享内存的文件描述符和空间大小，SharedBuffer定义请查看SharedBuffer。

返回:

返回0表示成功

返回非0表示失败，负数为HDF标准错误码，正数为NNRt定义的专用错误码（请查看NNRT_ReturnCode）

你可能感兴趣的鸿蒙文章

harmony 鸿蒙HDI 参数参考

harmony 鸿蒙ActRecognitionEvent

harmony 鸿蒙Activation

harmony 鸿蒙ActivityRecognitionTypes.idl

harmony 鸿蒙AddFusion

harmony 鸿蒙Alignment

harmony 鸿蒙AllocInfo

harmony 鸿蒙ApduSimIORequestInfo

harmony 鸿蒙ArgMaxFusion

harmony 鸿蒙Audio

0  赞
  • 所属分类： 后端技术
  • 本文标签：
  • 版权声明： 原创文章如转载，请注明本文链接: https://m.seaxiang.com/blog/d1c68864072f4139ba2fe3a623f100b0
热门推荐
标签云
Linux Java Spring Spring Boot docker 大数据 前端 生活 maven golang 软件 spring cloud fintech mac vscode 其他 区块链 以太坊 V神 开发者 技术 股票 git greenplum 分布式事务 chrome mysql flink kafka 设计模式 原理 鸿蒙 seo
最新文章
本文目录