pub use rockchip_npu::{
RknpuAction,
ioctrl::{RknpuMemCreate, RknpuMemMap, RknpuSubmit},
};
#[derive(Debug, Clone, Copy, PartialEq, Eq)]
pub enum Error {
NotFound,
Busy,
InvalidData,
}
pub fn is_available() -> bool {
rdrive::get_one::<rockchip_npu::Rknpu>().is_some()
}
pub fn obj_addr_and_size(handle: u32) -> Result<(usize, usize), Error> {
with_npu(|npu| npu.get_obj_addr_and_size(handle).ok_or(Error::NotFound))
}
pub fn submit(args: &mut RknpuSubmit) -> Result<(), Error> {
with_npu(|npu| npu.submit_ioctrl(args).map_err(|_| Error::InvalidData))
}
pub fn mem_create(args: &mut RknpuMemCreate) -> Result<(), Error> {
with_npu(|npu| npu.create(args).map_err(|_| Error::InvalidData))
}
pub fn mem_map_offset(handle: u32) -> Result<u64, Error> {
with_npu(|npu| {
npu.get_phys_addr_and_size(handle)
.map(|_| (handle as u64) << 12)
.ok_or(Error::InvalidData)
})
}
pub fn action(flags: RknpuAction) -> Result<u32, Error> {
with_npu(|npu| npu.action(flags).map_err(|_| Error::InvalidData))
}
fn with_npu<F, R>(f: F) -> Result<R, Error>
where
F: FnOnce(&mut rockchip_npu::Rknpu) -> Result<R, Error>,
{
let mut npu = rdrive::get_one::<rockchip_npu::Rknpu>()
.ok_or(Error::NotFound)?
.try_lock()
.map_err(|_| Error::Busy)?;
f(&mut npu)
}