dlpark 0.7.0

dlpack Rust binding for Python
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

use snafu::prelude::*;

use crate::{ffi, utils::MemoryOrder};

pub type Result<T> = std::result::Result<T, Error>;

#[derive(Debug, Snafu)]
#[snafu(visibility(pub(crate)))]
pub enum Error {
    #[snafu(display("Data type mismatch, bits {size}, expect {expected}"))]
    DataTypeSizeMismatch { size: usize, expected: usize },

    #[snafu(display("Shape mismatch, expected {expected:?}, actual {actual:?}"))]
    ShapeMismatch {
        expected: Vec<i64>,
        actual: Vec<i64>,
    },

    #[snafu(display("non contiguous"))]
    NonContiguous { shape: Vec<i64>, strides: Vec<i64> },

    #[snafu(display("not supported memory order {order}, expected {expected}"))]
    UnsupportedMemoryOrder {
        order: MemoryOrder,
        expected: MemoryOrder,
    },

    #[snafu(display("unsupported data type {name}"))]
    UnsupportedDataType { name: String },

    #[snafu(display("unsupported device {device:?}"))]
    UnsupportedDevice { device: ffi::DeviceType },

    #[snafu(display("invalid dimensions, expected {expected}, actual {actual}"))]
    InvalidDimensions { expected: usize, actual: usize },

    #[snafu(display("invalid channels, expected {expected}, actual {actual}"))]
    InvalidChannels { expected: i64, actual: i64 },
    // Cuda {
    //     source: cudarc::
    // }
}