dlpack-rs 0.3.3

Portable Rust bindings for the DLPack protocol (FFI + safe wrappers + optional pyo3 glue)
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

//! Re-export of auto-generated DLPack FFI bindings from the inlined `sys` module.
//! The bindgen-generated types replace the former hand-written structs.
#[allow(unused_imports)]
pub use crate::sys::*;

// ── Stable public aliases ──────────────────────────────────────────────────────
// These insulate consumers from bindgen naming (DLDeviceType::kDLCPU, etc.) and
// match the struct field types so comparisons compile without a cast at call sites.
//
// DLDevice::device_type  → DLDeviceType::Type = c_uint = u32
// DLDataType::code       → u8
// DLManagedTensorVersioned::flags → u64

/// CPU device type constant — directly comparable to `DLDevice::device_type` (u32).
pub const K_DL_CPU: u32 = DLDeviceType::kDLCPU;

/// CUDA GPU device type constant — directly comparable to `DLDevice::device_type` (u32).
pub const K_DL_CUDA: u32 = DLDeviceType::kDLCUDA;

/// Unsigned-integer dtype code — directly comparable to `DLDataType::code` (u8).
pub const K_DL_UINT: u8 = DLDataTypeCode::kDLUInt as u8;

/// Signed-integer dtype code — directly comparable to `DLDataType::code` (u8).
pub const K_DL_INT: u8 = DLDataTypeCode::kDLInt as u8;

/// IEEE floating-point dtype code — directly comparable to `DLDataType::code` (u8).
pub const K_DL_FLOAT: u8 = DLDataTypeCode::kDLFloat as u8;

/// Boolean dtype code — directly comparable to `DLDataType::code` (u8).
pub const K_DL_BOOL: u8 = DLDataTypeCode::kDLBool as u8;

/// Read-only bitmask for `DLManagedTensorVersioned::flags` (u64).
pub const DLPACK_FLAG_BITMASK_READ_ONLY: u64 = crate::sys::DLPACK_FLAG_BITMASK_READ_ONLY as u64;

/// Is-copied bitmask for `DLManagedTensorVersioned::flags` (u64).
pub const DLPACK_FLAG_BITMASK_IS_COPIED: u64 = crate::sys::DLPACK_FLAG_BITMASK_IS_COPIED as u64;