singe-cuda 0.1.0-alpha.3

Safe Rust wrappers for CUDA driver, runtime, NVRTC, memory, streams, modules, and graphs.
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

#[allow(unused_imports)]
use crate::error::ErrorCode;

use singe_cuda_sys::runtime;

use crate::{error::Result, try_cuda};

/// Returns the version number of the current CUDA Runtime instance.
/// The version is returned as (1000 major + 10 minor).
/// For example, CUDA 9.2 would be represented by 9020.
///
/// As of CUDA 12.0, this function no longer initializes CUDA.
/// The purpose of this API is solely to return a compile-time constant stating the CUDA Toolkit version in the above format.
///
/// Note:
///
/// * Note that this function may also return [`ErrorCode::NotInitialized`], [`ErrorCode::CallRequiresNewerDriver`] or [`ErrorCode::NoDevice`] if this call tries to initialize internal CUDA RT state.
/// * Note that as specified by [`Stream::add_callback`](crate::stream::Stream::add_callback) no CUDA function may be called from callback.
///   [`ErrorCode::NotPermitted`] may, but is not guaranteed to, be returned as a diagnostic in such case.
pub fn version() -> Result<i32> {
    let mut version: i32 = 0;
    unsafe {
        try_cuda!(runtime::cudaRuntimeGetVersion(&raw mut version))?;
    }
    Ok(version)
}

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

    #[test]
    fn it_works() {
        assert_ne!(version().unwrap(), 0);
    }
}