cndrv 0.1.2

Safe Cambricon driver API.
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28

use crate::{bindings::CNmodule, impl_spore, AsRaw, CnrtcBinary, CurrentCtx};
use std::{marker::PhantomData, ptr::null_mut};

impl_spore!(Module and ModuleSpore by CNmodule);

impl CurrentCtx {
    #[inline]
    pub fn load(&self, bin: CnrtcBinary) -> Module {
        let mut module = null_mut();
        cndrv!(cnModuleLoadFatBinary(bin.as_ptr(), &mut module));
        Module(unsafe { self.wrap_raw(module) }, PhantomData)
    }
}

impl Drop for Module<'_> {
    #[inline]
    fn drop(&mut self) {
        cndrv!(cnModuleUnload(self.0.raw));
    }
}

impl AsRaw for Module<'_> {
    type Raw = CNmodule;
    #[inline]
    unsafe fn as_raw(&self) -> Self::Raw {
        self.0.raw
    }
}