dlpack-rs 0.3.4

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

//! Auto-generated DLPack FFI bindings (rust-bindgen from `vendor/dlpack/dlpack.h`).
//!
//! Formerly the separate `dlpack-rs-sys` crate; inlined here so `dlpack-rs`
//! publishes to crates.io as a single crate (no sys crate to publish).
//! Regenerate `bindings.rs` with `--features bindgen` (needs libclang).
#![allow(
    non_upper_case_globals,
    non_camel_case_types,
    non_snake_case,
    dead_code
)]

include!("bindings.rs");

#[cfg(test)]
mod tests {
    use super::*;
    use core::mem::size_of;
    #[cfg(target_pointer_width = "64")]
    #[test]
    fn abi_layout_lp64() {
        assert_eq!(size_of::<DLTensor>(), 48);
        assert_eq!(size_of::<DLManagedTensor>(), 64);
        assert_eq!(size_of::<DLManagedTensorVersioned>(), 80);
    }
}