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
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63

use {onednnl_sys::dnnl_status_t, std::fmt::Display};

#[derive(Debug, PartialEq)]
pub enum DnnlError {
    InvalidArguments,
    OutOfMemory,
    Unsupported,
    Unknown,
    InvalidDataType,
    InvalidGraph,
    InvalidGraphOp,
    RuntimeError,
    InvalidShape,
    LastImplReached,
    NotRequired,
    Success,
    InvalidLayout,
    NonNullViolated,
    InvalidQueryOutput,
}

impl From<dnnl_status_t::Type> for DnnlError {
    fn from(status: dnnl_status_t::Type) -> Self {
        match status {
            dnnl_status_t::dnnl_invalid_arguments => Self::InvalidArguments,
            dnnl_status_t::dnnl_out_of_memory => Self::OutOfMemory,
            dnnl_status_t::dnnl_unimplemented => Self::Unsupported,
            dnnl_status_t::dnnl_invalid_data_type => Self::InvalidDataType,
            dnnl_status_t::dnnl_invalid_graph => Self::InvalidGraph,
            dnnl_status_t::dnnl_invalid_graph_op => Self::InvalidGraphOp,
            dnnl_status_t::dnnl_runtime_error => Self::RuntimeError,
            dnnl_status_t::dnnl_invalid_shape => Self::InvalidShape,
            dnnl_status_t::dnnl_last_impl_reached => Self::LastImplReached,
            dnnl_status_t::dnnl_not_required => Self::NotRequired,
            dnnl_status_t::dnnl_success => Self::Success,
            _ => Self::Unknown,
        }
    }
}

impl std::error::Error for DnnlError {}

impl Display for DnnlError {
    fn fmt(&self, f: &mut std::fmt::Formatter<'_>) -> std::fmt::Result {
        match self {
            DnnlError::InvalidArguments => write!(f, "InvalidArguments"),
            DnnlError::OutOfMemory => write!(f, "OutOfMemory"),
            DnnlError::Unsupported => write!(f, "Unsupported"),
            DnnlError::Unknown => write!(f, "Unknown"),
            DnnlError::InvalidDataType => write!(f, "InvalidDataType"),
            DnnlError::InvalidGraph => write!(f, "InvalidGraph"),
            DnnlError::InvalidGraphOp => write!(f, "InvalidGraphOp"),
            DnnlError::RuntimeError => write!(f, "RuntimeError"),
            DnnlError::InvalidShape => write!(f, "InvalidShape"),
            DnnlError::LastImplReached => write!(f, "LastImplReached"),
            DnnlError::NotRequired => write!(f, "NotRequired"),
            DnnlError::Success => write!(f, "Success"),
            DnnlError::InvalidLayout => write!(f, "InvalidLayout"),
            DnnlError::NonNullViolated => write!(f, "NonNullViolated"),
            DnnlError::InvalidQueryOutput => write!(f, "InvalidQueryOutput"),
        }
    }
}