singe-cutensor-sys 0.1.0-alpha.3

Low-level FFI bindings for the NVIDIA cuTENSOR tensor algebra 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

#![allow(
    warnings,
    deprecated,
    unused_qualifications,
    clippy::all,
    clippy::pedantic
)]

use num_enum::{IntoPrimitive, TryFromPrimitive};

pub use singe_cuda_sys::library_types::cudaDataType_t;
use singe_cuda_sys::{driver::CUstream as cudaStream_t, library_types::cudaDataType};

#[cfg(feature = "cutensor_2_6")]
include!("sys_20600.rs");

#[cfg(test)]
mod tests {
    use super::*;

    #[test]
    fn it_works() {
        let version = unsafe { cutensorGetVersion() };
        println!("cuTENSOR version: {}", version);
        assert_ne!(version, 0);
    }
}