dlpark 0.8.0-alpha.1

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

use crate::ffi::{DLDevice, DLDeviceType};

impl DLDevice {
    pub const CPU: Self = Self {
        device_type: DLDeviceType::CPU,
        device_id: 0,
    };

    pub fn cuda(device_id: i32) -> Self {
        Self {
            device_type: DLDeviceType::CUDA,
            device_id,
        }
    }
}

impl Default for DLDevice {
    fn default() -> Self {
        Self {
            device_type: DLDeviceType(0),
            device_id: 0,
        }
    }
}