onednnl 0.0.1

high-level bindings to oneDNN Deep Learning library
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20

pub mod engine;
pub mod error;
pub mod graph;
pub mod memory;
pub mod primitive;
pub mod primitives;
pub mod stream;

use error::DnnlError;
pub use onednnl_sys;

pub fn set_primitive_cache_capacity(capacity: std::ffi::c_int) -> Result<(), DnnlError> {
    let status = unsafe { onednnl_sys::dnnl_set_primitive_cache_capacity(capacity) };

    if status == onednnl_sys::dnnl_status_t::dnnl_success {
        Ok(())
    } else {
        Err(status.into())
    }
}