Skip to main content

singe_cutensor_sys/
lib.rs

1//! Raw FFI bindings for cuTENSOR and cuTENSORMg.
2//!
3//! Prefer the safe `singe-cutensor` crate unless direct NVIDIA ABI access is required.
4
5#![allow(deprecated, warnings, unused_qualifications, clippy::all)]
6
7use num_enum::{IntoPrimitive, TryFromPrimitive};
8
9use singe_cuda_sys::driver::CUstream as cudaStream_t;
10pub use singe_cuda_sys::library_types::cudaDataType as cudaDataType_t;
11use singe_nccl_sys::ncclComm_t;
12
13#[cfg(feature = "cutensor_2_6")]
14include!("sys_20600.rs");
15
16#[cfg(all(feature = "cutensor_2_6", feature = "mg"))]
17include!("mg_20600.rs");
18
19#[cfg(all(feature = "cutensor_2_6", feature = "mp"))]
20include!("mp_20600.rs");
21
22#[cfg(test)]
23mod tests {
24    use super::*;
25
26    #[test]
27    fn it_works() {
28        let version = unsafe { cutensorGetVersion() };
29        println!("cuTENSOR version: {}", version);
30        assert_ne!(version, 0);
31    }
32}